Searched defs:Tile1D (Results 1 – 2 of 2) sorted by relevance
55 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles) { in Tile1D() function
36 HloSharding HloSharding::Tile1D(const Shape& input_shape, int64 num_tiles, in Tile1D() function in xla::HloSharding