Searched refs:HloCreationUtilsTest (Results 1 – 1 of 1) sorted by relevance
28 class HloCreationUtilsTest : public HloTestBase { class46 TEST_F(HloCreationUtilsTest, CollapseFirst1Dim) { in TEST_F() argument65 TEST_F(HloCreationUtilsTest, CollapseFirst2Dims) { in TEST_F() argument88 TEST_F(HloCreationUtilsTest, Prepend1DegenerateDim) { in TEST_F() argument107 TEST_F(HloCreationUtilsTest, Prepend2DegenerateDims) { in TEST_F() argument126 TEST_F(HloCreationUtilsTest, Prepend2DegenerateDimsToScalar) { in TEST_F() argument145 TEST_F(HloCreationUtilsTest, ExpandFirstDimInto3Dims) { in TEST_F() argument166 TEST_F(HloCreationUtilsTest, PadVectorWithZeros) { in TEST_F() argument186 TEST_F(HloCreationUtilsTest, BroadcastZeros_S32) { in TEST_F() argument205 TEST_F(HloCreationUtilsTest, BroadcastZeros_F32) { in TEST_F() argument