experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA] Add more 1D SelectAndScatter tests.

Browse Source 
PiperOrigin-RevId: 253637442
This commit is contained in:
A. Unique TensorFlower 2019-06-17 12:35:07 -07:00 committed by TensorFlower Gardener
parent a6825e00a2
commit bfa58c30dd
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/xla/tests/select_and_scatter_test.cc
View File

@ -84,7 +84,7 @@ XLA_TEST_P(SelectAndScatterTest, ParamTest) {
GetParam().window_strides, GetParam().padding_type, source, GetParam().window_strides, GetParam().padding_type, source,
ConstantR0<float>(&builder_, 0.0f), add_f32_); ConstantR0<float>(&builder_, 0.0f), add_f32_);
ComputeAndCompare(&builder_, {}, ErrorSpec(1e-4)); ComputeAndCompare(&builder_, {}, ErrorSpec(2e-4));
} }
INSTANTIATE_TEST_CASE_P( INSTANTIATE_TEST_CASE_P(
@ -201,8 +201,11 @@ INSTANTIATE_TEST_CASE_P(
SelectAndScatterTestParam{ SelectAndScatterTestParam{
{7, 8, 256}, {4, 8, 256}, Padding::kSame, {2, 1, 1}, {2, 1, 1}}, {7, 8, 256}, {4, 8, 256}, Padding::kSame, {2, 1, 1}, {2, 1, 1}},
SelectAndScatterTestParam{{1104}, {551}, Padding::kValid, {3}, {2}}, 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{ SelectAndScatterTestParam{
{1300}, {1171}, Padding::kValid, {130}, {1}})); {3000}, {401}, Padding::kValid, {2600}, {1}}));
// Test for F32 1D array, with a zero-element input. // Test for F32 1D array, with a zero-element input.
XLA_TEST_F(SelectAndScatterTest, R1S0F32) { XLA_TEST_F(SelectAndScatterTest, R1S0F32) {