diff --git a/tinynn/converter/operators/torch/aten.py b/tinynn/converter/operators/torch/aten.py index eb9c6859..7e6e9956 100644 --- 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)