Searched refs:kPrimReduceMax (Results 1 – 9 of 9) sorted by relevance
| /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 | graph_kernel_cluster.cc | 90 {kGPUDevice, OpLevel_1, prim::kPrimReduceMax}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 37 Register(prim::kPrimReduceMax->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reduce_eliminate.h | 47 … IsPrimitiveCNode(node, prim::kPrimReduceSum) || IsPrimitiveCNode(node, prim::kPrimReduceMax) || in operator()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 109 REG_ADPT_DESC(ReduceMaxD, prim::kPrimReduceMax->name(), ADPT_DESC(ReduceMaxD))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_cpu_kernel.cc | 70 if (kernel_name_ == prim::kPrimReduceMax->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/core/abstract/ |
| D | primitive_infer_map.cc | 225 {prim::kPrimReduceMax, {InferImplReduceFunc, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 476 inline const PrimitivePtr kPrimReduceMax = std::make_shared<Primitive>("ReduceMax"); variable
|