From 1fd562e0a86515d04885bee63aac4e760cc9cd32 Mon Sep 17 00:00:00 2001 From: peterjc123 Date: Mon, 8 Apr 2024 15:14:35 +0800 Subject: [PATCH] [converter] expand strides for conv ops (#289) --- tinynn/converter/operators/torch/aten.py | 3 +++ 1 file changed, 3 insertions(+) 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)