Searched defs:sharding2 (Results 1 – 2 of 2) sorted by relevance
167 HloSharding sharding2 = HloSharding::Replicate(); in TEST_F() local173 HloSharding sharding2 = HloSharding::AssignDevice(1); in TEST_F() local179 HloSharding sharding2 = HloSharding::AssignDevice(2); in TEST_F() local185 HloSharding sharding2 = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F() local194 HloSharding sharding2 = HloSharding::Tuple(shape_tree); in TEST_F() local202 HloSharding sharding2 = HloSharding::Tuple(shape_tree); in TEST_F() local216 HloSharding sharding2 = HloSharding::Tuple(shape_tree2); in TEST_F() local230 HloSharding sharding2 = HloSharding::Tuple(shape_tree2); in TEST_F() local
59 const HloSharding& sharding2) { in ShardingMatches()