Searched refs:kPrimAddN (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | merge_addn.h | 42 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(node); in operator()
75 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(first_input); in Visit()
95 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(last_input); in Visit()
152 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(node); in operator()
|
| D | arithmetic_simplify.cc | 107 auto addn_pat = PPrimitive(prim::kPrimAddN, admktup_pat); in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | addn_fission.cc | 28 …std::vector<AnfNodePtr> new_addn_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimAddN->… in CreateNewAddn()
45 return VectorRef({prim::kPrimAddN, Xs}); in DefinePattern()
61 …r<AnfNodePtr> base_addn_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimAddN->name()))}; in Process()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 71 …stitution(std::make_shared<AdjustAllReduceMulAdd>(), "adjust_all_reduce_mul_add", prim::kPrimAddN); in OptimizeIRPassLib()
168 merge_addn_ = MakeSubstitution(std::make_shared<MergeAddN>(), "merge_addn", prim::kPrimAddN); in OptimizeIRPassLib()
169 …ilter_ = MakeSubstitution(std::make_shared<AddNZeroFilter>(), "addn_zero_filter", prim::kPrimAddN); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | replace_addn_fusion.cc | 25 VectorRef addn = VectorRef({prim::kPrimAddN, A, B}); in DefinePattern()
|
| D | apply_momentum_weight_fusion.cc | 30 VectorRef({prim::kPrimAddN, VectorRef({prim::kPrimMul, load_para, weight_decay_}), gradient_}); in DefinePattern()
|
| D | apply_momentum_weight_scale_fusion.cc | 89 …VectorRef({prim::kPrimAddN, VectorRef({prim::kPrimMul, load_para, weight_decay_}), cast_gradient_}… in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | matmul_eltwise_fusion_pass.cc | 56 AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimAddN)) { in MatchSingleFusionPattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | mul_addn_fusion.cc | 53 VectorRef addn({prim::kPrimAddN, mul, Y}); in DefinePattern()
|
| D | confusion_mul_grad_fusion.cc | 88 if (addn == nullptr || AnfAlgo::GetCNodeName(addn) != prim::kPrimAddN->name()) { in QuitFusion()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 66 {prim::kPrimAddN->name(), {{kAllPositions}, {0}}},
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | addn.cc | 101 REGISTER_PRIMITIVE_EVAL_IMPL(AddN, prim::kPrimAddN, AddNInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/ |
| D | pass_registry_test.cc | 174 if (!opt::CheckPrimitiveType(node, prim::kPrimAddN)) { in Execute()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 53 {kAllTarget, OpLevel_0, prim::kPrimAddN}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 212 inline const PrimitivePtr kPrimAddN = std::make_shared<Primitive>("AddN"); variable
|