Searched refs:HloShardingTest (Results 1 – 1 of 1) sorted by relevance
39 class HloShardingTest : public HloTestBase {}; class41 TEST_F(HloShardingTest, Replicate) { in TEST_F() argument56 TEST_F(HloShardingTest, DevicePlacement) { in TEST_F() argument78 TEST_F(HloShardingTest, Tile) { in TEST_F() argument119 TEST_F(HloShardingTest, EmptySingleTuple) { in TEST_F() argument125 TEST_F(HloShardingTest, NestedTuple) { in TEST_F() argument157 TEST_F(HloShardingTest, Hash) { in TEST_F() argument235 TEST_F(HloShardingTest, ToStringReplicatedTest) { in TEST_F() argument240 TEST_F(HloShardingTest, ToStringAssignDeviceTest) { in TEST_F() argument245 TEST_F(HloShardingTest, ToStringTiledTest) { in TEST_F() argument[all …]