Searched refs:shape_tree2 (Results 1 – 1 of 1) sorted by relevance
211 ShapeTree<HloSharding> shape_tree2( in TEST_F() local214 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()216 HloSharding sharding2 = HloSharding::Tuple(shape_tree2); in TEST_F()225 ShapeTree<HloSharding> shape_tree2( in TEST_F() local228 *shape_tree2.mutable_element({0}) = HloSharding::AssignDevice(0); in TEST_F()230 HloSharding sharding2 = HloSharding::Tuple(shape_tree2); in TEST_F()