Searched defs:NumTiles (Results 1 – 1 of 1) sorted by relevance
598 int64 HloSharding::NumTiles() const { in NumTiles() function in xla::HloSharding610 int64 HloSharding::NumTiles(absl::Span<const int64> dims) const { in NumTiles() function in xla::HloSharding