Searched defs: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() argument240 XLA_TEST_F(BroadcastSimpleTest, BooleanAnd2DTo3D_Pred) { in XLA_TEST_F() argument266 XLA_TEST_F(BroadcastSimpleTest, ZeroElement_1DTo2D) { in XLA_TEST_F() argument282 XLA_TEST_F(BroadcastSimpleTest, InDimensionAndDegenerateBroadcasting) { in XLA_TEST_F() argument379 XLA_TEST_F(BroadcastSimpleTest, Add3DTo3DDegenerate_1_2) { in XLA_TEST_F() argument[all …]
156 class BroadcastSimpleTest(test.TestCase): class