Searched refs:SetupCommonCollectiveOpAttributes (Results 1 – 1 of 1) sorted by relevance
1035 Status SetupCommonCollectiveOpAttributes(OpT op, const HloInstruction* instr, in SetupCommonCollectiveOpAttributes() function1053 SetupCommonCollectiveOpAttributes(all_to_all_op, instr, builder_)); in EmitAllToAllOp()1067 SetupCommonCollectiveOpAttributes(all_gather_op, instr, builder_)); in EmitAllGatherOp()1081 SetupCommonCollectiveOpAttributes(all_reduce_op, instr, builder_)); in EmitAllReduceOp()