Searched defs:TileShape (Results 1 – 1 of 1) sorted by relevance
567 Shape HloSharding::TileShape(const Shape& shape) const { in TileShape() function in xla::HloSharding579 Shape HloSharding::TileShape(const Shape& shape, int64 device) const { in TileShape() function in xla::HloSharding