Searched refs:kPrimNeg (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | prelu_fusion.cc | 29 VectorRef x_pattern({prim::kPrimRelu, VectorRef({prim::kPrimNeg, x_})}); in DefinePattern()
30 VectorRef mul_pattern({prim::kPrimMul, VectorRef({prim::kPrimNeg, weight_}), x_pattern}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | neg.h | 33 Neg() : PrimitiveC(prim::kPrimNeg->name()) { InitIOName({"x"}, {"y"}); } in Neg()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 215 {prim::kPrimNeg, {ops::NegInfer, nullptr, false}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | reorder_ops.cc | 36 prim::kPrimSqueeze, prim::kPrimTile, prim::kPrimNeg, prim::kPrimRelu, in IsTypeInsensitive()
|
| D | graph_kernel_cluster.cc | 45 {kAllTarget, OpLevel_0, prim::kPrimNeg}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.cc | 313 {prim::kPrimNeg->name(), Neg<T>}, in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.cc | 434 REG_ADPT_DESC(Neg, prim::kPrimNeg->name(), ADPT_DESC(Neg))
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 491 auto prim = prim::kPrimNeg; in TEST_F()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | pattern_matcher.h | 743 if (calcu_type == prim::kPrimNeg) {
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 479 inline const PrimitivePtr kPrimNeg = std::make_shared<Primitive>(kNeg); variable
|