Searched defs:tile_shape (Results 1 – 4 of 4) sorted by relevance
41 OpSharding Tile(const Shape& tile_shape, in Tile()55 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles) { in Tile1D()
646 ShapeProto tile_shape = 2; field
1187 const int64 tile_shape = CeilOfRatio(user.shape().dimensions(cdim), in GetShardingFromUser() local
314 const Shape tile_shape = sharding.TileShape(source_shape); in ReshapeSharding() local