Searched refs:ALL_GATHER (Results 1 – 5 of 5) sorted by relevance
285 auto all_gather_output = gen_g.PushBack({gen_g.NewOpInst(ALL_GATHER), expandim_output}); in ComputeReplaceGraph()319 auto all_gather_output = gen_g.PushBack({gen_g.NewOpInst(ALL_GATHER), expandim_output}); in ComputeReplaceGraph()
243 constexpr char ALL_GATHER[] = "AllGather"; variable
366 prim->name() == ALL_GATHER) { in AddCommOpFusionType()400 OperatorName operator_name = ALL_GATHER; in CreateAllGatherOp()
160 op_ = std::make_pair(ALL_GATHER, args); in AllGatherOP()
60 static const std::set<std::string> COMMUNICATION_OPS = {ALL_REDUCE, ALL_GATHER, ALL_TO_ALL, REDUCE_…1571 op_name = ALL_GATHER; in ApplyParallelOptOnParam()