Searched refs:CustomCallTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_test.cc | 53 class CustomCallTest : public ClientLibraryTestBase {}; class
62 TEST_F(CustomCallTest, IsInvoked) { in TEST_F() argument
72 TEST_F(CustomCallTest, UnknownTarget) { in TEST_F() argument
88 TEST_F(CustomCallTest, Memcpy) { in TEST_F() argument
105 TEST_F(CustomCallTest, Opaque) { in TEST_F() argument
138 TEST_F(CustomCallTest, SubBuffers) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 75 class CustomCallTest : public HloTestBase { class
81 XLA_TEST_F(CustomCallTest, CustomCallR0F32Add2) { in XLA_TEST_F() argument
96 XLA_TEST_F(CustomCallTest, CustomCallR2F32Reduce) { in XLA_TEST_F() argument
117 XLA_TEST_F(CustomCallTest, UsedInOtherComputations) { in XLA_TEST_F() argument
141 XLA_TEST_F(CustomCallTest, InputAndOutputLayoutDiffer) { in XLA_TEST_F() argument
163 XLA_TEST_F(CustomCallTest, LayoutConstrained) { in XLA_TEST_F() argument
190 XLA_TEST_F(CustomCallTest, TupleOutput) { in XLA_TEST_F() argument
|