Searched refs:kPrimReduceAll (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 53 Register(prim::kPrimReduceAll->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reduce_eliminate.h | 46 …f (IsPrimitiveCNode(node, prim::kPrimReduceMean) || IsPrimitiveCNode(node, prim::kPrimReduceAll) || in operator()
|
| D | branch_culling.cc | 65 {prim::kPrimReduceAll, {2}}, in InConvertWhiteList()
88 {prim::kPrimReduceAll, {2}}, {prim::kPrimCast, {2}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 85 REG_ADPT_DESC(ReduceAllD, prim::kPrimReduceAll->name(), ADPT_DESC(ReduceAllD))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_cpu_kernel.cc | 60 if (kernel_name_ == prim::kPrimReduceAll->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/backend/optimizer/graph_kernel/ |
| D | decrease_compute_precision.cc | 216 …rimitivePtr> cast_aware_list = {prim::kPrimReduceSum, prim::kPrimReduceMean, prim::kPrimReduceAll}; in IsCastUnAware()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 223 {prim::kPrimReduceAll, {InferImplReduceFunc, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 474 inline const PrimitivePtr kPrimReduceAll = std::make_shared<Primitive>("ReduceAll"); variable
|