Home
last modified time | relevance | path

Searched refs:ALL_REDUCE (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dget_parallel_info.cc60 auto allreduce_prim_list = FindPrimtive(graph, ALL_REDUCE); in GetAllreduceFusion()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dunique_info.cc147 AnfNodePtr reduce_op = gen_g.PushBack({gen_g.NewOpInst(ALL_REDUCE, attrs), mul}); in ComputeReplaceGraph()
Dops_utils.h238 constexpr char ALL_REDUCE[] = "AllReduce"; variable
Dgather_v2_p_info.cc797 reduce_op = gen_g.PushBack({gen_g.NewOpInst(ALL_REDUCE, attrs), mul}); in ComputeReplaceGraph()
Doperator_info.cc325 OperatorName operator_name = ALL_REDUCE; in CreateAllReduceOp()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc60 static const std::set<std::string> COMMUNICATION_OPS = {ALL_REDUCE, ALL_GATHER, ALL_TO_ALL, REDUCE_…