Fixed roi_align golden mismatch error
Signed-off-by: Chen Xin <jack.chen@verisilicon.com>
This commit is contained in:
parent
25c38d45e1
commit
8867c8de35
|
|
@ -102,5 +102,5 @@ TEST(RoiAlign, shape_4_2_1_1_float32) {
|
|||
|
||||
std::vector<float> output(num_rois * out_height * out_width * depth);
|
||||
EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data()));
|
||||
EXPECT_EQ(golden, output);
|
||||
ArraysMatch(golden, output, 1e-5f);
|
||||
}
|
||||
Loading…
Reference in New Issue