Searched refs:AbstractFuncAtomPtr (Results 1 – 8 of 8) sorted by relevance
/third_party/mindspore/mindspore/core/abstract/ |
D | abstract_function.h | 39 void Visit(std::function<void(const AbstractFuncAtomPtr &)>) const final; 61 void Visit(std::function<void(const AbstractFuncAtomPtr &)>) const final; 188 PartialAbstractClosure(const AbstractFuncAtomPtr &fn, const AbstractBasePtrList &args_spec_list, 208 AbstractFuncAtomPtr fn_; 218 explicit JTransformedAbstractClosure(const AbstractFuncAtomPtr &fn) : fn_(fn) {} in JTransformedAbstractClosure() 222 AbstractFuncAtomPtr fn() { return fn_; } in MS_DECLARE_PARENT() 230 AbstractFuncAtomPtr fn_; 292 …PyInterpretAbstractClosure(const AbstractFuncAtomPtr &fn, const AbstractBasePtrList &args_spec_lis… 312 AbstractFuncAtomPtr fn_;
|
D | abstract_function.cc | 49 void AbstractFuncAtom::Visit(std::function<void(const AbstractFuncAtomPtr &)> visit_func) const { in Visit() 59 …auto build_func_list = [&new_func_list](const AbstractFuncAtomPtr &func) { new_func_list.push_back… in AbstractFuncUnion() 83 auto build_in_list = [this, &is_in_list](const AbstractFuncAtomPtr &func) { in IsSuperSet() 111 void AbstractFuncUnion::Visit(std::function<void(const AbstractFuncAtomPtr &)> visit_func) const { in Visit() 112 for (const AbstractFuncAtomPtr &poss : func_list_) { in Visit()
|
D | prim_statement.cc | 106 AbstractFuncAtomPtrList func_list{b->cast<AbstractFuncAtomPtr>()}; in InferImplSwitchLayer()
|
D | abstract_value.h | 200 using AbstractFuncAtomPtr = std::shared_ptr<AbstractFuncAtom>; variable 201 using AbstractFuncAtomPtrList = std::vector<AbstractFuncAtomPtr>; 224 virtual void Visit(std::function<void(const AbstractFuncAtomPtr &)>) const = 0;
|
/third_party/mindspore/mindspore/core/load_mindir/ |
D | infer_mindir.cc | 50 void EvalAbstractFunction(const abstract::AbstractFuncAtomPtr &abstractFunc, const CNodePtr &node, 229 auto real_func = (*args)[0]->cast<abstract::AbstractFuncAtomPtr>(); in EvalPartialPrimitive() 244 auto real_func = func->cast<abstract::AbstractFuncAtomPtr>(); in EvalPartialPrimitive() 272 auto build_fuction = [&abstractFuncList](const abstract::AbstractFuncAtomPtr &poss) { in EvalPartialAbastract() 444 void MindIREngine::EvalAbstractFunction(const abstract::AbstractFuncAtomPtr &func, const CNodePtr &… in EvalAbstractFunction() 502 auto build_fuction = [&abstractFuncList](const abstract::AbstractFuncAtomPtr &poss) { in InferCNode()
|
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
D | static_analysis.cc | 302 auto build_evaluator = [this, &evaluators, &cnode](const AbstractFuncAtomPtr &poss) { in EvalCNode() 320 auto build_evaluator = [this, &infs](const AbstractFuncAtomPtr &poss) { in Execute()
|
D | prim.cc | 1387 auto build_partial = [args, cnode, &partial_funcs_list](const AbstractFuncAtomPtr &atom_func) { in Run()
|
/third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
D | ops_front_infer_function.cc | 602 auto build_jv = [&jv](const AbstractFuncAtomPtr &func) { in InferImplJ()
|