| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | confusion_softmax_grad_rule.h | 30 reduce_sum_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimReduceSum->name()));
|
| D | softmax_grad_ext_fusion.h | 32 sum_var_ = std::make_shared<Var>(std::make_shared<Primitive>(prim::kPrimReduceSum->name())); in PatternProcessPass()
|
| D | square_sum_fusion.cc | 96 return VectorRef({prim::kPrimReduceSum, VectorRef({prim::kPrimSquare, X})}); in DefinePattern()
|
| D | confusion_mul_grad_fusion.cc | 122 VectorRef reduce_sum({prim::kPrimReduceSum, mul1}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | axis_normalizer.cc | 27 …std::vector<PrimitivePtr> node_with_axis = {prim::kPrimReduceSum, prim::kPrimReduceMax, prim::kPri… in IsReduce()
|
| D | add_atomic_clean.h | 50 PrimitivePtr target_type_{prim::kPrimReduceSum};
|
| D | raise_reduction_precision.cc | 31 …return IsPrimitiveCNode(node, prim::kPrimReduceSum) && AnfAlgo::GetInputDeviceDataType(node, 0) ==… in IsFp16ReduceSum()
|
| D | add_stitch_atomic_clean_gpu.cc | 117 …etNodeAttr<std::string>(n, kAttrStitch) == "atomic" && IsPrimitiveCNode(n, prim::kPrimReduceSum)) { in IsStitchWithAtomic()
|
| D | add_atomic_clean.cc | 41 if (!IsPrimitiveCNode(node, prim::kPrimReduceSum)) { in GetUniqReduceAxes()
78 if (IsPrimitiveCNode(n, prim::kPrimReduceSum)) { in HaveReduceInPredecessors()
|
| D | decrease_transfer_precision.cc | 52 if (AnfAlgo::GetCNodeName(input_node) == prim::kPrimReduceSum->name()) { in IsPreNodeReduce()
|
| D | decrease_compute_precision.cc | 216 …std::vector<PrimitivePtr> cast_aware_list = {prim::kPrimReduceSum, prim::kPrimReduceMean, prim::kP… in IsCastUnAware()
|
| D | graph_kernel_cluster.cc | 49 {kAllTarget, OpLevel_1, prim::kPrimReduceSum}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | bce_with_logits_loss_fusion.cc | 49 …reduce_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), new_cnod… in AddReduceNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bce_with_logits_loss_fission.cc | 51 …reduce_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), new_cnod… in AddReduceNode()
|
| D | dynamic_gru_v2_grad_fission.cc | 415 …dePtr> reducesum_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), in CreateDwReduceSumDNode()
432 …dePtr> reducesum_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), in CreateDbReduceSumDNode()
|
| D | dynamic_rnn_grad_fission_v2.cc | 455 …ePtr> reduce_sum_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), in CreateDwReduceSum()
505 …ePtr> reduce_sum_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceSum->name())), in CreateDbReduceSum()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | value_based_eliminate.cc | 74 if (IsPrimitiveCNode(node, prim::kPrimReduceSum) || IsPrimitiveCNode(node, prim::kPrimSqueeze)) { in IsCNodePositive()
|
| D | reduce_eliminate.h | 47 … IsPrimitiveCNode(node, prim::kPrimReduceSum) || IsPrimitiveCNode(node, prim::kPrimReduceMax) || in operator()
|
| D | branch_culling.cc | 63 {prim::kPrimReduceSum, {2}}, in InConvertWhiteList()
87 {prim::kPrimReduceSum, {2}}, {prim::kPrimReduceMean, {2}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 46 const auto kReduceSum = prim::kPrimReduceSum->name(); in GetDependsFormMap()
221 {prim::kPrimReduceSum, {ops::ReduceSumInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 39 Register(prim::kPrimReduceSum->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 69 REG_ADPT_DESC(ReduceSumD, prim::kPrimReduceSum->name(), ADPT_DESC(ReduceSumD))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_cpu_kernel.cc | 76 } else if (kernel_name_ == prim::kPrimReduceSum->name()) { in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 101 …{prim::kPrimReduceMean, prim::kPrimReduceAll, prim::kPrimReduceSum, prim::kPrimReduceMax, prim::kP… in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_graph_kernel.cc | 317 if (!IsPrimitiveCNode(cnode, prim::kPrimReduceSum)) { in UpdateEquivFormat()
|