Searched refs:shape_tree1 (Results 1 – 1 of 1) sorted by relevance
207 ShapeTree<HloSharding> shape_tree1( in TEST_F() local210 *shape_tree1.mutable_element({0}) = HloSharding::Replicate(); in TEST_F()215 HloSharding sharding1 = HloSharding::Tuple(shape_tree1); in TEST_F()221 ShapeTree<HloSharding> shape_tree1( in TEST_F() local224 *shape_tree1.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()229 HloSharding sharding1 = HloSharding::Tuple(shape_tree1); in TEST_F()