[converter] expand strides for conv ops (#289)

alibaba · Apr 8, 2024 · 1fd562e · 1fd562e
1 parent d8b2a34
commit 1fd562e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tinynn/converter/operators/torch/aten.py b/tinynn/converter/operators/torch/aten.py
@@ -1729,6 +1729,9 @@ def parse(self, node, attrs, args, graph_converter):
         inputs = [self.find_or_create_input(i, graph_converter) for i in range(end_index)]
         outputs = self.to_tfl_tensors(self.output_names, self.output_tensors)
 
+        if len(stride) != len(padding) and len(stride) == 1:
+            stride = stride * len(padding)
+
         if transpose == 0:
             graph_converter.add_operator(
                 tfl.GenericConvOperator(inputs, outputs, stride, padding, dilation, output_padding, groups)