Home
last modified time | relevance | path

Searched refs:PrimitiveAbstractClosure (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dabstract_function.h74 class MS_CORE_API PrimitiveAbstractClosure : public AbstractFuncAtom {
79 …explicit PrimitiveAbstractClosure(const PrimitivePtr &prim, const AnfNodePtr &tracking_id = nullpt…
81 ~PrimitiveAbstractClosure() override = default;
82 MS_DECLARE_PARENT(PrimitiveAbstractClosure, AbstractFuncAtom) in MS_DECLARE_PARENT() argument
90 …AbstractFunctionPtr Copy() const override { return std::make_shared<PrimitiveAbstractClosure>(prim… in Copy()
105 using PrimitiveAbstractClosurePtr = std::shared_ptr<PrimitiveAbstractClosure>;
Dabstract_function.cc137 bool PrimitiveAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
138 if (!other.isa<PrimitiveAbstractClosure>()) { in operator ==()
141 auto other_prim = static_cast<const PrimitiveAbstractClosure *>(&other); in operator ==()
146 std::size_t PrimitiveAbstractClosure::hash() const { in hash()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Ddata_test.cc95 AbstractFunctionPtr f1 = std::make_shared<PrimitiveAbstractClosure>(nullptr, nullptr); in TEST_F()
107 AbstractFunctionPtr f1 = std::make_shared<PrimitiveAbstractClosure>(nullptr, nullptr); in TEST_F()
Dprim_test.cc367 std::make_shared<PrimitiveAbstractClosure>(prim::kPrimScalarAdd), fn_args_list); in TEST_F()
/third_party/mindspore/mindspore/core/ir/
Dprimitive.cc60 …return std::make_shared<abstract::PrimitiveAbstractClosure>(shared_from_base<Primitive>(), nullptr… in ToAbstract()
Dfunc_graph_extends.cc35 using mindspore::abstract::PrimitiveAbstractClosure;
75 auto f = std::make_shared<PrimitiveAbstractClosure>(prim::kPrimReturn, input0); in set_output()
/third_party/mindspore/tests/ut/cpp/operator/
Dcomposite_test.cc253 auto fn_arg= std::make_shared<abstract::PrimitiveAbstractClosure>(prim::kPrimMakeTuple); in TEST_F()
280 auto fn_arg = std::make_shared<abstract::PrimitiveAbstractClosure>(prim::kPrimMakeTuple); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc42 auto func_ptr = std::make_shared<abstract::PrimitiveAbstractClosure>(prim::kPrimMakeRecord); in ToAbstract()
69 auto func_ptr = std::make_shared<abstract::PrimitiveAbstractClosure>(prim::kPrimCreateInstance); in ToAbstract()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprogram_specialize.cc490 auto prim_func = dyn_cast<PrimitiveAbstractClosure>(func); in BuildSpecializedNodeInner()
747 if (func->isa<PrimitiveAbstractClosure>() && argvals.empty()) { in IsPolyFunc()
852 if (abs->isa<PrimitiveAbstractClosure>()) { in BuildPossibleValueNode()
853 auto real_fn = dyn_cast<PrimitiveAbstractClosure>(abs); in BuildPossibleValueNode()
Dstatic_analysis.cc424 EvaluatorPtr AnalysisEngine::_GetEvaluatorFor(const std::shared_ptr<PrimitiveAbstractClosure> &func… in _GetEvaluatorFor()
501 if (func->isa<PrimitiveAbstractClosure>()) { in _GetEvaluatorFor()
502 return _GetEvaluatorFor(func->cast<std::shared_ptr<PrimitiveAbstractClosure>>()); in _GetEvaluatorFor()
988 auto prim_func = std::make_shared<abstract::PrimitiveAbstractClosure>(primitive, anf_node); in MakeAbstractClosure()
Dstatic_analysis.h263 EvaluatorPtr _GetEvaluatorFor(const std::shared_ptr<PrimitiveAbstractClosure> &fn);
Dprim.cc831 auto prim_func = dyn_cast<abstract::PrimitiveAbstractClosure>(abs_func); in StaticGetterInferred()
1366 if (func->isa<PrimitiveAbstractClosure>()) { in Run()
1367 auto prim_func = dyn_cast<PrimitiveAbstractClosure>(func); in Run()
/third_party/mindspore/mindspore/core/load_mindir/
Dinfer_mindir.cc420 result = std::make_shared<abstract::PrimitiveAbstractClosure>(prim, node); in InferValueNode()
447 if (func->isa<abstract::PrimitiveAbstractClosure>()) { in EvalAbstractFunction()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dcse.cc57 if (node_abs->isa<abstract::PrimitiveAbstractClosure>()) { in AbsOf()