Searched refs:InfeedTest (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 35 class InfeedTest : public ClientLibraryTestBase { class
54 TEST_F(InfeedTest, SingleInfeedR0Bool) { in TEST_F() argument
58 TEST_F(InfeedTest, SingleInfeedR1U32) { in TEST_F() argument
62 TEST_F(InfeedTest, SingleInfeedR2F32) { in TEST_F() argument
66 TEST_F(InfeedTest, SingleInfeedR3F32) { in TEST_F() argument
72 TEST_F(InfeedTest, SingleInfeedR3F32DifferentLayout) { in TEST_F() argument
87 TEST_F(InfeedTest, SingleInfeedR4S32) { in TEST_F() argument
94 TEST_F(InfeedTest, LargeInfeed) { in TEST_F() argument
100 TEST_F(InfeedTest, SingleInfeedTuple) { in TEST_F() argument
106 TEST_F(InfeedTest, SingleInfeedEmptyTuple) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 39 class InfeedTest : public ClientLibraryTestBase { class
60 TEST_F(InfeedTest, SingleInfeedR0Bool) { in TEST_F() argument
64 TEST_F(InfeedTest, SingleInfeedR1U32) { in TEST_F() argument
68 TEST_F(InfeedTest, SingleInfeedR2F32) { in TEST_F() argument
72 TEST_F(InfeedTest, SingleInfeedR3F32) { in TEST_F() argument
78 TEST_F(InfeedTest, SingleInfeedR3F32DifferentLayout) { in TEST_F() argument
93 TEST_F(InfeedTest, SingleInfeedR4S32) { in TEST_F() argument
99 TEST_F(InfeedTest, SingleInfeedTuple) { in TEST_F() argument
105 TEST_F(InfeedTest, SingleInfeedEmptyTuple) { in TEST_F() argument
119 TEST_F(InfeedTest, DISABLED_SingleInfeedInWhile) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_infeed_test.py | 29 class InfeedTest(test.TestCase): class
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_execute_test.cc | 897 XLA_TEST_F(LocalClientExecuteTest, DISABLED_ON_INTERPRETER(InfeedTest)) { in XLA_TEST_F() argument
|