colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed roi_align golden mismatch error 

Signed-off-by: Chen Xin <jack.chen@verisilicon.com>
This commit is contained in:
Chen Xin 2022-11-08 16:56:35 +08:00 committed by Sven
parent 25c38d45e1
commit 8867c8de35
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/tim/vx/ops/roi_align_test.cc
View File

@ -102,5 +102,5 @@ TEST(RoiAlign, shape_4_2_1_1_float32) {
std::vector<float> output(num_rois * out_height * out_width * depth); std::vector<float> output(num_rois * out_height * out_width * depth);
EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data())); EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data()));
EXPECT_EQ(golden, output); ArraysMatch(golden, output, 1e-5f);
} }