Searched defs:shard_ordinal (Results 1 – 2 of 2) sorted by relevance
600 HloInstruction* shard_ordinal, SpmdBuilder* b) const { in Calculate()689 HloInstruction* OffsetCalculation::Calculate(HloInstruction* shard_ordinal, in Calculate()
876 [&](int64_t shard_ordinal) { in ReshardAsWindowedInput()886 for (int64_t shard_ordinal = 0; shard_ordinal < shard_count; in ReshardAsWindowedInput() local904 for (int64_t shard_ordinal = 1; shard_ordinal < shard_count; in ReshardAsWindowedInput() local