Searched refs:kPrimReduceMean (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | bce_with_logits_loss_fusion.cc | 51 …reduce_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceMean->name())), new_cno… in AddReduceNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bce_with_logits_loss_fission.cc | 53 …reduce_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimReduceMean->name())), new_cno… in AddReduceNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 40 Register(prim::kPrimReduceMean->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | reduce_eliminate.h | 46 …if (IsPrimitiveCNode(node, prim::kPrimReduceMean) || IsPrimitiveCNode(node, prim::kPrimReduceAll) … in operator()
|
| D | branch_culling.cc | 64 … {prim::kPrimReduceMean, {2}}, in InConvertWhiteList()
87 {prim::kPrimReduceSum, {2}}, {prim::kPrimReduceMean, {2}}, in InConvertWhiteList()
|
| D | less_batch_normalization.cc | 161 …{prim::kPrimReduceMean, prim::kPrimRelu6, prim::kPrimTupleGetItem, prim::kPrimBatchNorm, prim::kPr…
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | reduce_ops_declare.cc | 93 REG_ADPT_DESC(ReduceMeanD, prim::kPrimReduceMean->name(), ADPT_DESC(ReduceMeanD))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reduce_cpu_kernel.cc | 79 } else if (kernel_name_ == prim::kPrimReduceMean->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 | graph_kernel_expander.cc | 81 {kGPUDevice, OpLevel_1, prim::kPrimReduceMean}, in GetExpandOps()
|
| D | decrease_compute_precision.cc | 216 …std::vector<PrimitivePtr> cast_aware_list = {prim::kPrimReduceSum, prim::kPrimReduceMean, prim::kP… in IsCastUnAware()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 222 {prim::kPrimReduceMean, {InferImplReduceFunc, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 461 auto prim = prim::kPrimReduceMean; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 782 auto name = prim::kPrimReduceMean->name(); in ExportPrimReduce()
1578 if (node->IsApply(prim::kPrimReduceMean) || node->IsApply(prim::kPrimReduceSum)) { in ExportCNode()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 472 inline const PrimitivePtr kPrimReduceMean = std::make_shared<Primitive>("ReduceMean"); variable
|