Home
last modified time | relevance | path

Searched refs:kPrimReturn (Results 1 – 25 of 59) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dstatic_analysis_test.cc67 prim::kPrimReturn, StandardPrimitiveImplReg{InferImplReturnStub, nullptr, true}); in InitPrimitiveReturnEvaluatorStub()
75 PrimEvaluatorConstructorsStub[prim::kPrimReturn] = InitPrimitiveReturnEvaluatorStub(); in InitPrimEvaluatorConstructorsStub()
106 auto prim_return = prim::kPrimReturn; in GenerateFromTypes()
144 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeFuncGraph()
194 auto prim_return = prim::kPrimReturn; in SetUp()
237 inputs.push_back(NewValueNode(prim::kPrimReturn)); in SetUp()
341 auto prim_return = prim::kPrimReturn; in SetUp()
Ddata_test.cc64 AbstractBasePtr abs_f1 = FromValue(prim::kPrimReturn, false); in TEST_F()
66 ASSERT_EQ(abs_f1_built, prim::kPrimReturn); in TEST_F()
77 …ASSERT_EQ(*func_tuple_built, ValueTuple(std::vector<ValuePtr>{prim::kPrimReturn, prim::kPrimScalar… in TEST_F()
82 …ASSERT_EQ(*func_list_built, ValueList(std::vector<ValuePtr>{prim::kPrimReturn, prim::kPrimScalarAd… in TEST_F()
/third_party/mindspore/tests/ut/cpp/ir/
Dmanager_test.cc263 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeFuncGraph()
284 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph()
290 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph()
322 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph2()
334 inputs.push_back(NewValueNode(prim::kPrimReturn)); in MakeNestedGraph2()
368 inputs.push_back(NewValueNode(prim::kPrimReturn)); in TEST_F()
451 inputs.push_back(NewValueNode(prim::kPrimReturn)); in TEST_F()
464 inputs.push_back(NewValueNode(prim::kPrimReturn)); in TEST_F()
484 inputs.push_back(NewValueNode(prim::kPrimReturn)); in TEST_F()
/third_party/mindspore/tests/ut/cpp/optimizer/
Dclean_test.cc77 AnfNodePtr valuenode_return = NewValueNode(prim::kPrimReturn); in SetUp()
152 auto cons_return = NewValueNode(prim::kPrimReturn); in TEST_F()
191 auto cons_return = NewValueNode(prim::kPrimReturn); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dbatchmatmul_fusedmuladd_fusion_pass.cc49 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dconv_bnreduce_fusion_pass.cc50 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dconv2dbackprop_eltwise_fusion_pass.cc53 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dmatmul_confusiontranspose_fusion_pass.cc52 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dbnupdate_eltwise_fusion_pass.cc53 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dmatmul_eltwise_fusion_pass.cc49 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dconv_single_in_fusion_pass.cc63 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Deltwise_fusion_pass.cc62 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dconv_double_in_fusion_pass.cc59 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dmulti_output_fusion_pass.cc67 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Ddepthwiseconv_eltwise_fusion_pass.cc58 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dbnupdate_eltwise_eltwise_fusion_pass.cc64 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dstridedread_conv_stridedwrite_fusion_pass.cc70 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dsegment_eltwise_fusion_pass.cc73 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dconv2dbackprop_eltwise_eltwise_fusion_pass.cc75 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
Dreduce_eltwise_fusion_pass.cc76 AnfAlgo::CheckPrimitiveType(node, prim::kPrimReturn)) { in MatchSingleFusionPattern()
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/
Dmodel_parser_registry_test.cc58 bool is_return = opt::CheckPrimitiveType(*iter, prim::kPrimReturn); in TEST_F()
/third_party/mindspore/tests/ut/cpp/common/
Dbackend_common_test.cc39 !IsPrimitiveCNode(node, prim::kPrimReturn)) { in GetCNodeList()
/third_party/mindspore/mindspore/ccsrc/vm/
Dtransform.cc47 std::vector<PrimitivePtr> nonlinear_ops = {prim::kPrimReturn, prim::kPrimPartial, prim::kPrimSwitch,
50 std::vector<PrimitivePtr> control_ops = {prim::kPrimReturn, prim::kPrimPartial, prim::kPrimSwitch, …
54 …static const std::vector<PrimitivePtr> ms_nonlinear_ops = {prim::kPrimReturn, prim::kPrimPartial, in GetMsNonlinearOps()
191 if (IsPrimitive(fn, prim::kPrimReturn)) { in InterpretNode()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_extends.cc61 std::vector<AnfNodePtr> params({NewValueNode(prim::kPrimReturn), value}); in set_output()
75 auto f = std::make_shared<PrimitiveAbstractClosure>(prim::kPrimReturn, input0); in set_output()
/third_party/mindspore/mindspore/lite/tools/optimizer/common/
Dformat_utils.cc193 CheckPrimitiveType(cnode, prim::kPrimReturn); in IsSpecialType()

123