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