Searched refs:kPrimAllReduce (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 104 auto all_reduce_pat = PPrimitive(prim::kPrimAllReduce, x); in operator ()()
|
| D | special_op_eliminate.h | 639 auto pattern = PPrimitive(prim::kPrimAllReduce, x); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 146 if (IsPrimitiveCNode(cnode, prim::kPrimAllReduce)) { in GetGeNodeName()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 117 …ubstitution(std::make_shared<AllReduceConstElim>(), "reduce_all_const_elim", prim::kPrimAllReduce); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 232 {prim::kPrimAllReduce, {InferImplAllReduce, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pass.cc | 626 if (IsPrimitiveCNode(cnode, prim::kPrimAllReduce) && cnode->inputs().size() > 1 && in RemoveValueNodeDuplicationsPass()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 420 inline const PrimitivePtr kPrimAllReduce = std::make_shared<Primitive>("AllReduce"); variable
|