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 } })); } }