Searched defs:BroadcastSimpleTest (Results 1 – 2 of 2) sorted by relevance
35 class BroadcastSimpleTest : public ClientLibraryTestBase { class106 XLA_TEST_F(BroadcastSimpleTest, ScalarNoOpBroadcast) { in XLA_TEST_F() argument112 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x3) { in XLA_TEST_F() argument119 XLA_TEST_F(BroadcastSimpleTest, ScalarParamTo2D_2x3) { in XLA_TEST_F() argument132 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_2x0) { in XLA_TEST_F() argument139 XLA_TEST_F(BroadcastSimpleTest, ScalarTo2D_0x2) { in XLA_TEST_F() argument238 XLA_TEST_F(BroadcastSimpleTest, BooleanAnd2DTo3D_Pred) { in XLA_TEST_F() argument264 XLA_TEST_F(BroadcastSimpleTest, ZeroElement_1DTo2D) { in XLA_TEST_F() argument280 XLA_TEST_F(BroadcastSimpleTest, InDimensionAndDegenerateBroadcasting) { in XLA_TEST_F() argument377 XLA_TEST_F(BroadcastSimpleTest, Add3DTo3DDegenerate_1_2) { in XLA_TEST_F() argument[all …]
151 class BroadcastSimpleTest(test.TestCase): class