Lines Matching refs:U32
43 Shape tile_shape = ShapeUtil::MakeShape(U32, {4});
53 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}),
69 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}),
72 sharding.Validate(ShapeUtil::MakeShape(U32, {4}), /*num_devices=*/5));
75 sharding.GetAsShapeTree(ShapeUtil::MakeShape(U32, {4}));
83 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
92 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
95 EXPECT_IS_NOT_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4, 6}),
101 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
111 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
120 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
200 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
203 HloSharding sharding2 = HloSharding::Tile(ShapeUtil::MakeShape(U32, {2, 3}),
209 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
212 HloSharding sharding2 = HloSharding::Tile(ShapeUtil::MakeShape(U32, {2, 3}),
218 Shape tile_shape = ShapeUtil::MakeShape(U32, {2, 3});
221 HloSharding sharding2 = HloSharding::Tile(ShapeUtil::MakeShape(U32, {2, 3}),
