Searched refs:collective_permute_done (Results 1 – 2 of 2) sorted by relevance
235 HloInstruction* collective_permute_done);
754 HloInstruction* collective_permute_done) { in UpdateCollectivePermuteDoneValueSet() argument755 CHECK_EQ(collective_permute_done->opcode(), in UpdateCollectivePermuteDoneValueSet()760 GetValueSet(collective_permute_done->operand(0), {1}); in UpdateCollectivePermuteDoneValueSet()761 HloValueSet& value_set = GetValueSet(collective_permute_done); in UpdateCollectivePermuteDoneValueSet()