Searched refs:kPrimScalarMul (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | grad_implementations_test.cc | 36 FuncGraphPtr fg = ad::g_k_prims.KPrimitive(nullptr, NewValueNode(kPrimScalarMul), nullptr); in TEST_F()
39 auto fg1 = ad::g_k_prims.KPrimitive(nullptr, NewValueNode(kPrimScalarMul), nullptr); in TEST_F()
|
| D | ops_test.cc | 49 ASSERT_EQ(prim->name(), kPrimScalarMul->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 56 … {"__mul__", prim::kPrimScalarMul}, // P.scalar_mul in GetMethodMap()
78 … {"__mul__", prim::kPrimScalarMul}, // P.scalar_mul, in GetMethodMap()
100 … {"__mul__", prim::kPrimScalarMul}, // P.scalar_mul, in GetMethodMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 37 …MATCH_REPLACE(node, PBinOperation(prim::kPrimScalarMul, x, one_scalar_, true), x); // Sc… in operator ()()
40 …MATCH_REPLACE(node, PBinOperation(prim::kPrimScalarMul, x, zero_scalar_, true), zero_scalar_.NewVa… in operator ()()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ad/ |
| D | ad_test.cc | 100 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarMul), resourcePtr); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 59 … {prim::kPrimScalarAdd, prim::kPrimScalarMul, prim::kPrimAdd, in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 61 …emwiseMap = {{"__add__", kPrimScalarAdd}, {"__sub__", kPrimScalarSub}, {"__mul__", kPrimScalarMul},
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 112 inline const PrimitivePtr kPrimScalarMul = std::make_shared<Primitive>(kScalarMul); variable
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 1438 {prim::kPrimScalarMul, {prim::ScalarMul, true, nullptr, true}}, in GetUniformPrimitiveToImplMap()
|