From 4841074de0f019c1eb95789cc29b0531bec86a91 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=D0=A1=D0=B5=D1=80=D0=B3=D0=B5=D0=B9=20=D0=91=D0=B0=D1=80?= =?utf8?q?=D0=B0=D0=BD=D0=BD=D0=B8=D0=BA=D0=BE=D0=B2/AI=20Tools=20Lab=20/S?= =?utf8?q?RR/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 4 Sep 2019 18:36:31 +0300 Subject: [PATCH] [mir2loco] Change type of strides argument (#7178) Change the type of `strides` argument of `Conv2D` and `DepthwiseConv2D` operations to `vector`. Signed-off-by: Sergei Barannikov --- compiler/mir2loco/src/mir2loco.test.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/compiler/mir2loco/src/mir2loco.test.cpp b/compiler/mir2loco/src/mir2loco.test.cpp index 223fdd3..139f4db 100644 --- a/compiler/mir2loco/src/mir2loco.test.cpp +++ b/compiler/mir2loco/src/mir2loco.test.cpp @@ -368,7 +368,7 @@ TEST_F(TestTransformer_mir2loco, Conv2D_Test) auto mir_tensor = mir::TensorVariant(mir::DataType::FLOAT32, shape, (const void *)data); auto *constant = mir_graph.create(mir_tensor)->getOutput(0); auto *conv = mir_graph - .create(input, constant, mir::Shape{2, 3}, + .create(input, constant, std::vector{2, 3}, std::vector{5, 9}, std::vector{7, 4}) ->getOutput(0); @@ -495,9 +495,9 @@ TEST_F(TestTransformer_mir2loco, DepthwiseConv2D_Test) auto mir_tensor = mir::TensorVariant(mir::DataType::FLOAT32, shape, (const void *)data); auto *constant = mir_graph.create(mir_tensor)->getOutput(0); auto *conv = mir_graph - .create(input, constant, mir::Shape{2, 3}, - std::vector{5, 9}, - std::vector{7, 4}) + .create( + input, constant, std::vector{2, 3}, + std::vector{5, 9}, std::vector{7, 4}) ->getOutput(0); mir_graph.create(conv); input->setName("x"); -- 2.7.4