Searched refs:collective_permute_op (Results 1 – 1 of 1) sorted by relevance
3079 auto collective_permute_op = in EmitCollectivePermuteFromMlir() local3081 if (collective_permute_op.channel_id()) in EmitCollectivePermuteFromMlir()3086 ConvertNx2Attribute(collective_permute_op.source_target_pairs())); in EmitCollectivePermuteFromMlir()3090 GetAllocationSliceForMlir(collective_permute_op.operand())); in EmitCollectivePermuteFromMlir()3093 GetAllocationSliceForMlir(collective_permute_op.output())); in EmitCollectivePermuteFromMlir()