Searched refs:kPrimScalarAdd (Results 1 – 16 of 16) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/pipeline/ |
| D | resource_test.cc | 66 ASSERT_TRUE(value == Any(prim::kPrimScalarAdd)); in TEST_F()
68 ASSERT_TRUE(value == Any(prim::kPrimScalarAdd)); in TEST_F()
70 ASSERT_TRUE(value == Any(prim::kPrimScalarAdd)); in TEST_F()
72 ASSERT_TRUE(value == Any(prim::kPrimScalarAdd)); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | data_test.cc | 74 AbstractBasePtr abs_f2 = FromValue(prim::kPrimScalarAdd, false); in TEST_F()
77 …_EQ(*func_tuple_built, ValueTuple(std::vector<ValuePtr>{prim::kPrimReturn, prim::kPrimScalarAdd})); in TEST_F()
82 …RT_EQ(*func_list_built, ValueList(std::vector<ValuePtr>{prim::kPrimReturn, prim::kPrimScalarAdd})); in TEST_F()
87 …ple_built, ValueTuple(std::vector<ValuePtr>{std::make_shared<Int64Imm>(1), prim::kPrimScalarAdd})); in TEST_F()
|
| D | static_analysis_test.cc | 51 prim::kPrimScalarAdd, StandardPrimitiveImplReg{InferImplScalarAddStub, nullptr, true}); in InitPrimitiveScalarAddEvaluatorStub()
76 PrimEvaluatorConstructorsStub[prim::kPrimScalarAdd] = InitPrimitiveScalarAddEvaluatorStub(); in InitPrimEvaluatorConstructorsStub()
232 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in SetUp()
|
| D | prim_test.cc | 256 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in TEST_F()
358 PrimitivePtr add = prim::kPrimScalarAdd; in TEST_F()
367 std::make_shared<PrimitiveAbstractClosure>(prim::kPrimScalarAdd), fn_args_list); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 54 … {"__add__", prim::kPrimScalarAdd}, // P.scalar_add in GetMethodMap()
76 … {"__add__", prim::kPrimScalarAdd}, // P.scalar_add, in GetMethodMap()
98 … {"__add__", prim::kPrimScalarAdd}, // P.scalar_add, in GetMethodMap()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | manager_test.cc | 317 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in MakeNestedGraph2()
351 auto prim_scalar_add = prim::kPrimScalarAdd; in TEST_F()
357 auto prim_scalar_add = prim::kPrimScalarAdd; in TEST_F()
471 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in TEST_F()
477 inputs.push_back(NewValueNode(prim::kPrimScalarAdd)); in TEST_F()
512 auto prim_scalar_add = prim::kPrimScalarAdd; in TEST_F()
|
| D | anf_test.cc | 54 auto primitive = prim::kPrimScalarAdd; in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ |
| D | clean_test.cc | 74 …AnfNodePtr cnode_66 = me_graph->NewCNode({NewValueNode(prim::kPrimScalarAdd), cnode_57, cnode_67}); in SetUp()
189 auto apply11 = func_graph->NewCNode({NewValueNode(prim::kPrimScalarAdd), apply22, apply33}); in TEST_F()
|
| D | opt_test.cc | 81 …= MakeSubstitution(std::make_shared<irpass::ArithmeticSimplify>(), "elim_Z", prim::kPrimScalarAdd); in SetUp()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ad/ |
| D | ad_test.cc | 95 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarAdd), resourcePtr); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 35 …MATCH_REPLACE(node, PBinOperation(prim::kPrimScalarAdd, x, zero_scalar_, true), x); // Sc… in operator ()()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 59 … {prim::kPrimScalarAdd, prim::kPrimScalarMul, prim::kPrimAdd, in OptimizeIRPassLib()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 39 ASSERT_EQ(prim->name(), kPrimScalarAdd->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 61 ElemwiseMap kElemwiseMap = {{"__add__", kPrimScalarAdd}, {"__sub__", kPrimScalarSub}, {"__mul__", k…
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 110 inline const PrimitivePtr kPrimScalarAdd = std::make_shared<Primitive>(kScalarAdd); variable
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.cc | 1436 {prim::kPrimScalarAdd, {prim::ScalarAdd, true, nullptr, true}}, in GetUniformPrimitiveToImplMap()
|