Searched refs:collective_permute_start (Results 1 – 2 of 2) sorted by relevance
233 HloInstruction* collective_permute_start);
737 HloInstruction* collective_permute_start) { in UpdateCollectivePermuteStartValueSet() argument738 CHECK_EQ(collective_permute_start->opcode(), in UpdateCollectivePermuteStartValueSet()744 GetValueSet(collective_permute_start->operand(0)); in UpdateCollectivePermuteStartValueSet()745 HloValueSet& value_set = GetValueSet(collective_permute_start, {0}); in UpdateCollectivePermuteStartValueSet()