Fix matmul

kendryte · Jan 14, 2019 · 3cdb4ed · 3cdb4ed
1 parent 5ad8fc5
commit 3cdb4ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 } }));
         }
     }