[mir_onnx] Change type of strides argument (#7173)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Wed, 4 Sep 2019 15:34:13 +0000 (18:34 +0300)
committerAlexander Efimov/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Wed, 4 Sep 2019 15:34:13 +0000 (18:34 +0300)
Change the type of `strides` argument of `Conv2D` and `DepthwiseConv2D` operations to `vector<int32_t>`.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
compiler/mir-onnx-importer/Op/Conv.cpp

index ea29c3f..aba6fe6 100644 (file)
@@ -110,10 +110,9 @@ void ConvNodeConverter::convertV1(const onnx::NodeProto &onnx_node, ConverterCon
     // TODO handle properly kernel with layer multiplier
     auto transposed_tensor = mir::transposeTensor<0, 1, 3, 2>(kernel_tensor);
     kernel = createOp<mir::ops::ConstantOp>(graph, transposed_tensor)->getOutput(0);
-    result =
-        createOp<mir::ops::DepthwiseConv2DOp>(graph, input, kernel, mir::Shape(strides),
-                                              padding_before, padding_after, mir::DataFormat::NCHW)
-            ->getOutput(0);
+    result = createOp<mir::ops::DepthwiseConv2DOp>(graph, input, kernel, strides, padding_before,
+                                                   padding_after, mir::DataFormat::NCHW)
+                 ->getOutput(0);
   }
   else
   {
@@ -122,7 +121,7 @@ void ConvNodeConverter::convertV1(const onnx::NodeProto &onnx_node, ConverterCon
       kernel_tensor = fixGroupedKernel(group, kernel_tensor);
     kernel_tensor = mir::transposeTensor<3, 0, 1, 2>(kernel_tensor);
     kernel = createOp<mir::ops::ConstantOp>(graph, kernel_tensor)->getOutput(0);
-    result = createOp<mir::ops::Conv2DOp>(graph, input, kernel, mir::Shape(strides), padding_before,
+    result = createOp<mir::ops::Conv2DOp>(graph, input, kernel, strides, padding_before,
                                           padding_after, mir::DataFormat::NCHW)
                  ->getOutput(0);
   }