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_t num_tiles) { in Tile1D()
695 ShapeProto tile_shape = 2; field
1363 const int64_t tile_shape = CeilOfRatio( in GetShardingFromUser() local
372 const Shape tile_shape = sharding.TileShape(source_shape); in ReshapeSharding() local