From 3cdb4edd0e7fd100974332c64fd7202c3b0771c3 Mon Sep 17 00:00:00 2001 From: sunnycase Date: Mon, 14 Jan 2019 17:59:12 +0800 Subject: [PATCH] Fix matmul --- src/NnCase.Converter/Model/Layers/K210/K210AddPadding.cs | 1 + 1 file changed, 1 insertion(+) diff --git a/src/NnCase.Converter/Model/Layers/K210/K210AddPadding.cs b/src/NnCase.Converter/Model/Layers/K210/K210AddPadding.cs index 0caf4d2e0a..acded86aab 100644 --- a/src/NnCase.Converter/Model/Layers/K210/K210AddPadding.cs +++ b/src/NnCase.Converter/Model/Layers/K210/K210AddPadding.cs @@ -26,6 +26,7 @@ protected override void OnPlanning(GraphPlanContext context) var graph = context.TFGraph; var input = context.TFOutputs[Input.Connection.From]; + input = graph.Reshape(input, graph.Const(new[] { Input.Dimensions[0], 1, 1, Input.Dimensions[1] })); context.TFOutputs[Output] = graph.Pad(input, graph.Const(new[,] { { 0, 0 }, { 0, 3 }, { 0, 3 }, { 0, 0 } })); } }