Searched defs:shard_ordinal (Results 1 – 2 of 2) sorted by relevance
683 HloInstruction* shard_ordinal, SpmdBuilder* b) const { in Calculate()772 HloInstruction* OffsetCalculation::Calculate(HloInstruction* shard_ordinal, in Calculate()
649 [&](int64 shard_ordinal) { in ReshardAsWindowedInput()659 for (int64 shard_ordinal = 0; shard_ordinal < shard_count; in ReshardAsWindowedInput() local676 for (int64 shard_ordinal = 1; shard_ordinal < shard_count; in ReshardAsWindowedInput() local