Searched refs:tiled_sharding (Results 1 – 1 of 1) sorted by relevance
133 HloSharding tiled_sharding = HloSharding::Tile(Array<int64>({{0, 1}})); in TEST_F() local138 *proto.add_tuple_shardings() = tiled_sharding.ToProto(); in TEST_F()146 EXPECT_EQ(shape_tree.element({2}), tiled_sharding); in TEST_F()