From 8867c8de35aaa39a94835fdb58aace88346353b1 Mon Sep 17 00:00:00 2001 From: Chen Xin Date: Tue, 8 Nov 2022 16:56:35 +0800 Subject: [PATCH] Fixed roi_align golden mismatch error Signed-off-by: Chen Xin --- src/tim/vx/ops/roi_align_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/vx/ops/roi_align_test.cc b/src/tim/vx/ops/roi_align_test.cc index bd21542..ad45608 100644 --- a/src/tim/vx/ops/roi_align_test.cc +++ b/src/tim/vx/ops/roi_align_test.cc @@ -102,5 +102,5 @@ TEST(RoiAlign, shape_4_2_1_1_float32) { std::vector output(num_rois * out_height * out_width * depth); EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data())); - EXPECT_EQ(golden, output); + ArraysMatch(golden, output, 1e-5f); } \ No newline at end of file