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