Searched refs:BroadcastSimpleTest (Results 1 – 2 of 2) sorted by relevance
35 class BroadcastSimpleTest : public ClientLibraryTestBase { class108 XLA_TEST_F(BroadcastSimpleTest, ScalarNoOpBroadcast) { in XLA_TEST_F() argument114 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x3) { in XLA_TEST_F() argument121 XLA_TEST_F(BroadcastSimpleTest, ScalarParamTo2D_2x3) { in XLA_TEST_F() argument134 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x0) { in XLA_TEST_F() argument141 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_0x2) { in XLA_TEST_F() argument148 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D) {162 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D_WithDimsUsual) {175 XLA_TEST_F(BroadcastSimpleTest, 1DTo2D_WithDimsTranspose) {188 XLA_TEST_F(BroadcastSimpleTest, 2DTo3D_WithDims) {[all …]
156 class BroadcastSimpleTest(test.TestCase): class