diff --git a/tensorflow/compiler/xla/tests/select_and_scatter_test.cc b/tensorflow/compiler/xla/tests/select_and_scatter_test.cc index 4b3283b5cd7..a1307e20371 100644 --- a/tensorflow/compiler/xla/tests/select_and_scatter_test.cc +++ b/tensorflow/compiler/xla/tests/select_and_scatter_test.cc @@ -84,7 +84,7 @@ XLA_TEST_P(SelectAndScatterTest, ParamTest) { GetParam().window_strides, GetParam().padding_type, source, ConstantR0(&builder_, 0.0f), add_f32_); - ComputeAndCompare(&builder_, {}, ErrorSpec(1e-4)); + ComputeAndCompare(&builder_, {}, ErrorSpec(2e-4)); } INSTANTIATE_TEST_CASE_P( @@ -201,8 +201,11 @@ INSTANTIATE_TEST_CASE_P( SelectAndScatterTestParam{ {7, 8, 256}, {4, 8, 256}, Padding::kSame, {2, 1, 1}, {2, 1, 1}}, SelectAndScatterTestParam{{1104}, {551}, Padding::kValid, {3}, {2}}, + SelectAndScatterTestParam{{1300}, {1171}, Padding::kValid, {130}, {1}}, + SelectAndScatterTestParam{{3000}, {1701}, Padding::kValid, {1300}, {1}}, + SelectAndScatterTestParam{{6500}, {5}, Padding::kValid, {1300}, {1300}}, SelectAndScatterTestParam{ - {1300}, {1171}, Padding::kValid, {130}, {1}})); + {3000}, {401}, Padding::kValid, {2600}, {1}})); // Test for F32 1D array, with a zero-element input. XLA_TEST_F(SelectAndScatterTest, R1S0F32) {