From ddcb00c11b172ed00a575959b8e14e42c082ce75 Mon Sep 17 00:00:00 2001 From: chxin66 <57057788+chxin66@users.noreply.github.com> Date: Sat, 20 May 2023 16:56:00 +0800 Subject: [PATCH] Fixed bug for pad test (#588) Type: Bug fix Signed-off-by: Chen Xin Co-authored-by: Chen Xin --- src/tim/vx/ops/pad_test.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/tim/vx/ops/pad_test.cc b/src/tim/vx/ops/pad_test.cc index a440138..1d24431 100644 --- a/src/tim/vx/ops/pad_test.cc +++ b/src/tim/vx/ops/pad_test.cc @@ -45,7 +45,7 @@ TEST(Pad, constant) { auto output_tensor = graph->CreateTensor(output_spec); std::vector input_data = { - 0, 1, 2, 3, 4, 5, 6, + 0, 1, 2, 3, 4, 5, }; std::vector golden = { @@ -168,11 +168,11 @@ TEST(Pad, reflect) { auto output_tensor = graph->CreateTensor(output_spec); std::vector input_data = { - 0, 1, 2, 3, 4, 5, 6, + 0, 1, 2, 3, 4, 5, }; std::vector golden = { - 0, 0, 1, 2, 2, 0, 0, 1, 2, 2, 3, 3, 4, 5, 5, 3, 3, 4, 5, 5, + 4, 3, 4, 5, 4, 1, 0, 1, 2, 1, 4, 3, 4, 5, 4, 1, 0, 1, 2, 1 }; EXPECT_TRUE( @@ -180,7 +180,7 @@ TEST(Pad, reflect) { std::vector front = {1, 1}; std::vector back = {1, 1}; auto op = graph->CreateOperation( - front, back, 0, tim::vx::ops::Pad::PAD_MODE_EDGE); + front, back, 0, tim::vx::ops::Pad::PAD_MODE_REFLECT); (*op).BindInput(input_tensor).BindOutput(output_tensor); EXPECT_TRUE(graph->Compile()); @@ -206,7 +206,7 @@ TEST(Pad, edge) { auto output_tensor = graph->CreateTensor(output_spec); std::vector input_data = { - 0, 1, 2, 3, 4, 5, 6, + 0, 1, 2, 3, 4, 5, }; std::vector golden = {0, 0, 1, 2, 2, 0, 0, 1, 2, 2,