[mir_tflite] Change type of strides argument (#7176)

Change the type of `strides` argument of `Conv2D` and `DepthwiseConv2D` operations to `vector<int32_t>`.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>

diff --git a/compiler/mir-tflite-importer/tflite_op_creator.cpp b/compiler/mir-tflite-importer/tflite_op_creator.cpp
index 3f4232b..ae5bd94 100644 (file)
--- a/compiler/mir-tflite-importer/tflite_op_creator.cpp
+++ b/compiler/mir-tflite-importer/tflite_op_creator.cpp
@@ -130,8 +130,7 @@ TFLiteOpCreator::convertConv2D(const Conv2DOptions *opts,
                    padding_after);
 
   auto result =
-      createOp<ops::Conv2DOp>(input, kernel, Shape(strides), padding_before, padding_after)
-          ->getOutput(0);
+      createOp<ops::Conv2DOp>(input, kernel, strides, padding_before, padding_after)->getOutput(0);
   result = createOp<ops::AddOp>(result, bias)->getOutput(0);
   return {addFusedActivation(result, opts->fused_activation_function())};
 }
@@ -159,7 +158,7 @@ TFLiteOpCreator::convertDepthwiseConv2D(const DepthwiseConv2DOptions *opts,
                    padding_after);
 
   auto result =
-      createOp<ops::DepthwiseConv2DOp>(input, kernel, Shape(strides), padding_before, padding_after)
+      createOp<ops::DepthwiseConv2DOp>(input, kernel, strides, padding_before, padding_after)
           ->getOutput(0);
   result = createOp<ops::AddOp>(result, bias)->getOutput(0);
   return {addFusedActivation(result, opts->fused_activation_function())};
@@ -265,7 +264,7 @@ TFLiteOpCreator::convertTransposeConv(const ::tflite::TransposeConvOptions *opts
   auto kernel = inputs.at(1);
   auto input = inputs.at(2);
 
-  Shape strides{opts->stride_h(), opts->stride_w()};
+  const std::vector<std::int32_t> strides{opts->stride_h(), opts->stride_w()};
   Shape output_shape(convertIntTensorToVector<int32_t>(output_shape_tensor));
 
   // OHWI -> HWOI