Skip to content

Commit fa5a3f9

Browse files
DavidNormanjhseu
authored andcommitted
[XLA] Break the Slice test into different tests for each data type (tensorflow#12354)
* Break the Slice test into different tests for each data type * Changes after review
1 parent fc9009f commit fa5a3f9

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

tensorflow/compiler/xla/tests/slice_test.cc

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,12 +193,27 @@ class SliceR1Test : public ClientLibraryTestBase,
193193
}
194194
};
195195

196-
XLA_TEST_P(SliceR1Test, DoIt) {
196+
XLA_TEST_P(SliceR1Test, DoIt_F32) {
197197
Run<float>(GetParam());
198+
}
199+
200+
XLA_TEST_P(SliceR1Test, DoIt_F64) {
198201
Run<double>(GetParam());
202+
}
203+
204+
XLA_TEST_P(SliceR1Test, DoIt_U32) {
199205
Run<uint32>(GetParam());
206+
}
207+
208+
XLA_TEST_P(SliceR1Test, DoIt_S32) {
200209
Run<int32>(GetParam());
210+
}
211+
212+
XLA_TEST_P(SliceR1Test, DoIt_U64) {
201213
Run<uint64>(GetParam());
214+
}
215+
216+
XLA_TEST_P(SliceR1Test, DoIt_S64) {
202217
Run<int64>(GetParam());
203218
}
204219

0 commit comments

Comments
 (0)