| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_function.h | 37 AbstractFunctionPtr GetUnique() override { return shared_from_base<AbstractFuncAtom>(); } in MS_DECLARE_PARENT()
38 AbstractFunctionPtr Join(const AbstractFunctionPtr &other) final;
48 AbstractFuncUnion(const AbstractFunctionPtr &first, const AbstractFunctionPtr &second);
54 AbstractFunctionPtr GetUnique() override { in GetUnique()
56 AbstractFunctionPtr result; in GetUnique()
59 bool IsSuperSet(const AbstractFunctionPtr &other);
60 AbstractFunctionPtr Join(const AbstractFunctionPtr &other) final;
64 AbstractFunctionPtr Copy() const override { in Copy()
66 AbstractFunctionPtr result; in Copy()
90 …AbstractFunctionPtr Copy() const override { return std::make_shared<PrimitiveAbstractClosure>(prim… in Copy()
[all …]
|
| D | abstract_function.cc | 25 AbstractFunctionPtr AbstractFunction::MakeAbstractFunction(const AbstractFuncAtomPtrList &func_list… in MakeAbstractFunction()
32 AbstractFunctionPtr AbstractFuncAtom::Join(const AbstractFunctionPtr &other) { in Join()
57 AbstractFuncUnion::AbstractFuncUnion(const AbstractFunctionPtr &first, const AbstractFunctionPtr &s… in AbstractFuncUnion()
80 bool AbstractFuncUnion::IsSuperSet(const AbstractFunctionPtr &other) { in IsSuperSet()
94 AbstractFunctionPtr AbstractFuncUnion::Join(const AbstractFunctionPtr &other) { in Join()
|
| D | abstract_value.h | 198 using AbstractFunctionPtr = std::shared_ptr<AbstractFunction>; variable
211 virtual AbstractFunctionPtr GetUnique() = 0;
219 virtual AbstractFunctionPtr Copy() const = 0;
222 virtual AbstractFunctionPtr Join(const AbstractFunctionPtr &other) = 0;
228 static AbstractFunctionPtr MakeAbstractFunction(const AbstractFuncAtomPtrList &func_list);
234 using AbstractFunctionPtrList = std::vector<AbstractFunctionPtr>;
|
| D | utils.cc | 179 AbstractFunctionPtr f_spec = dyn_cast<AbstractFunction>(spec); in SensitivityTransform()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | static_analysis.h | 199 std::size_t operator()(const std::pair<AbstractFunctionPtr, AbstractBasePtrList> &p) const { in operator()
200 auto h1 = std::hash<AbstractFunctionPtr>{}(p.first); in operator()
248 EvaluatorPtr GetEvaluatorFor(const AbstractFunctionPtr &fn);
255 EvalResultPtr Execute(const AbstractFunctionPtr &fn, const AbstractBasePtrList &args_spec_list);
296 …std::unordered_map<AbstractFunctionPtr, EvaluatorPtr, AbstractFunctionHasher, AbstractFunctionEqua…
297 …std::unordered_map<std::pair<AbstractFunctionPtr, AbstractBasePtrList>, EvaluatorPtr, PartialAppHa…
312 EvaluatorPtr _GetEvaluatorFor(const AbstractFunctionPtr &fn);
|
| D | stack_frame.h | 126 const AbstractFunctionPtr &graph_func);
128 const AbstractFunctionPtr &graph_func);
|
| D | program_specialize.h | 131 … const AbstractFunctionPtr &func, const AbstractBasePtrList &args,
135 SpecializeStatusCode FindUniqueArgvals(const AbstractFunctionPtr &fn, const EvaluatorPtr &eval,
|
| D | stack_frame.cc | 42 const AbstractFunctionPtr &graph_func) { in GetParentContext()
63 const AbstractFunctionPtr &graph_func) { in DoJump()
|
| D | program_specialize.cc | 429 AbstractFunctionPtr real_a = dyn_cast<AbstractFunction>(abs); in BuildSpecializedNode()
432 AbstractFunctionPtr func = real_a->GetUnique(); in BuildSpecializedNode()
464 const AbstractFunctionPtr &func, in BuildSpecializedNodeInner()
698 auto func_abs = func->abstract()->cast<AbstractFunctionPtr>(); in ProcessCNode()
745 bool IsPolyFunc(const AbstractFunctionPtr &func, const AbstractBasePtrList &argvals) { in IsPolyFunc()
766 SpecializeStatusCode FuncGraphSpecializer::FindUniqueArgvals(const AbstractFunctionPtr &func, const… in FindUniqueArgvals()
845 AbstractFunctionPtr abs = dyn_cast<AbstractFunction>(ival); in BuildPossibleValueNode()
|
| D | static_analysis.cc | 286 AbstractFunctionPtr func = dyn_cast<AbstractFunction>(possible_func); in EvalCNode()
313 EvalResultPtr AnalysisEngine::Execute(const AbstractFunctionPtr &func, const AbstractBasePtrList &a… in Execute()
466 AbstractFunctionPtr func_orig = func->fn(); in _GetEvaluatorFor()
481 AbstractFunctionPtr func_orig = func->fn(); in _GetEvaluatorFor()
499 EvaluatorPtr AnalysisEngine::_GetEvaluatorFor(const AbstractFunctionPtr &func) { in _GetEvaluatorFor()
526 EvaluatorPtr AnalysisEngine::GetEvaluatorFor(const AbstractFunctionPtr &func) { in GetEvaluatorFor()
543 AbstractFunctionPtr func_generic = func->Copy(); in GetEvaluatorFor()
|
| D | evaluator.h | 328 JEvaluator(const EvaluatorPtr &evaluator, const AbstractFunctionPtr &orig_func) in JEvaluator()
355 AbstractFunctionPtr orig_func_;
|
| D | evaluator.cc | 508 AbstractFunctionPtr bprop = in Run()
|
| D | prim.cc | 173 auto fn = args_spec_list[0]->cast<AbstractFunctionPtr>(); in Run()
821 AbstractFunctionPtr abs_func = dyn_cast<abstract::AbstractFunction>(abs_ptr); in StaticGetterInferred()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | data_test.cc | 95 AbstractFunctionPtr f1 = std::make_shared<PrimitiveAbstractClosure>(nullptr, nullptr); in TEST_F()
107 AbstractFunctionPtr f1 = std::make_shared<PrimitiveAbstractClosure>(nullptr, nullptr); in TEST_F()
148 AbstractFunctionPtr f1 = in TEST_F()
186 AbstractFunctionPtr f1 = in TEST_F()
|
| D | prim_test.cc | 290 AbstractFunctionPtr res_J_1 = dyn_cast<AbstractFunction>(res_J->elements()[1]); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | cse.cc | 34 using mindspore::abstract::AbstractFunctionPtr;
63 auto new_fg_abs = node_abs->cast<abstract::AbstractFunctionPtr>()->Copy(); in AbsOf()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.cc | 42 using mindspore::abstract::AbstractFunctionPtr;
43 using PrimTypePair = std::pair<PrimitivePtr, AbstractFunctionPtr>;
421 …const std::function<std::shared_ptr<FuncGraph>(const PrimitivePtr, const AbstractFunctionPtr)> &ge… in TraverseGraphMap()
463 auto get_prim_graph = [&prim_graphs](const PrimitivePtr &prim, const AbstractFunctionPtr &type) { in WrapPrimitives()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph_extends.cc | 33 using mindspore::abstract::AbstractFunctionPtr;
38 AbstractFunctionPtr FuncGraph::abstract() { in abstract()
|
| D | func_graph.h | 98 using AbstractFunctionPtr = std::shared_ptr<AbstractFunction>; variable
164 abstract::AbstractFunctionPtr abstract();
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.cc | 293 AbstractFunctionPtr fn = CheckArg<AbstractFunction>(primitive->name(), args_spec_list, 0); in InferImplListMap()
321 AbstractFunctionPtr fn = CheckArg<AbstractFunction>(op_name, args_spec_list, 0); in InferImplListReduce()
596 AbstractFunctionPtr x = dyn_cast<AbstractFunction>(args_spec_list[0]); in InferImplJ()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 54 using mindspore::abstract::AbstractFunctionPtr;
665 AbstractFunctionPtr fn = dyn_cast<AbstractFunction>(args_spec_list[0]); in GenerateFuncGraph()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | infer_mindir.cc | 496 abstract::AbstractFunctionPtr func = dyn_cast<abstract::AbstractFunction>(possible_func); in InferCNode()
|