Searched refs:kPrimReduceMin (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | reduce_min_fission.cc | 28 …std::vector<AnfNodePtr> inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceMin->n… in CreateReduceMin()
102 return VectorRef({prim::kPrimReduceMin, X}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | axis_normalizer.cc | 27 …r<PrimitivePtr> node_with_axis = {prim::kPrimReduceSum, prim::kPrimReduceMax, prim::kPrimReduceMin, in IsReduce()
|
| D | graph_kernel_cluster.cc | 91 {kGPUDevice, OpLevel_1, prim::kPrimReduceMin}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 38 Register(prim::kPrimReduceMin->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reduce_eliminate.h | 48 IsPrimitiveCNode(node, prim::kPrimReduceMin)) { in operator()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 101 REG_ADPT_DESC(ReduceMinD, prim::kPrimReduceMin->name(), ADPT_DESC(ReduceMinD))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_cpu_kernel.cc | 73 } else if (kernel_name_ == prim::kPrimReduceMin->name()) { in InitKernel()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 101 …duceMean, prim::kPrimReduceAll, prim::kPrimReduceSum, prim::kPrimReduceMax, prim::kPrimReduceMin}); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 226 {prim::kPrimReduceMin, {InferImplReduceFunc, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 477 inline const PrimitivePtr kPrimReduceMin = std::make_shared<Primitive>("ReduceMin"); variable
|