| /third_party/mindspore/mindspore/core/abstract/ |
| D | abstract_function.h | 31 class MS_CORE_API AbstractFuncAtom : public AbstractFunction {
35 MS_DECLARE_PARENT(AbstractFuncAtom, AbstractFunction) in MS_DECLARE_PARENT() argument
40 bool operator==(const AbstractFunction &other) const override;
45 class MS_CORE_API AbstractFuncUnion : public AbstractFunction {
50 MS_DECLARE_PARENT(AbstractFuncUnion, AbstractFunction)
62 bool operator==(const AbstractFunction &other) const override;
92 bool operator==(const AbstractFunction &other) const override;
132 bool operator==(const AbstractFunction &other) const override;
170 bool operator==(const AbstractFunction &other) const override;
202 bool operator==(const AbstractFunction &other) const override;
[all …]
|
| D | abstract_function.cc | 25 AbstractFunctionPtr AbstractFunction::MakeAbstractFunction(const AbstractFuncAtomPtrList &func_list… in MakeAbstractFunction()
53 bool AbstractFuncAtom::operator==(const AbstractFunction &other) const { return this == &other; } in operator ==()
95 auto this_func = shared_from_base<AbstractFunction>(); in Join()
117 bool AbstractFuncUnion::operator==(const AbstractFunction &other) const { in operator ==()
137 bool PrimitiveAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
156 bool FuncGraphAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
183 bool MetaFuncGraphAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
205 bool PartialAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
237 bool JTransformedAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
251 bool VirtualAbstractClosure::operator==(const AbstractFunction &other) const { in operator ==()
[all …]
|
| D | abstract_value.h | 197 class AbstractFunction; variable
198 using AbstractFunctionPtr = std::shared_ptr<AbstractFunction>;
203 class MS_CORE_API AbstractFunction : public AbstractBase {
205 AbstractFunction() = default;
206 ~AbstractFunction() override = default;
207 MS_DECLARE_PARENT(AbstractFunction, AbstractBase)
217 return const_cast<AbstractFunction *>(this)->shared_from_base<AbstractFunction>(); in Broaden()
226 virtual bool operator==(const AbstractFunction &other) const = 0;
|
| D | abstract_value.cc | 247 AbstractBasePtr AbstractFunction::Join(const AbstractBasePtr &other) { in Join()
249 auto other_func = dyn_cast<AbstractFunction>(other); in Join()
256 bool AbstractFunction::operator==(const AbstractBase &other) const { in operator ==()
257 if (!other.isa<AbstractFunction>()) { in operator ==()
260 const auto &other_func = static_cast<const AbstractFunction &>(other); in operator ==()
|
| D | utils.cc | 179 AbstractFunctionPtr f_spec = dyn_cast<AbstractFunction>(spec); in SensitivityTransform()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | validator.cc | 33 using mindspore::abstract::AbstractFunction;
118 …ptrBase->isa<AbstractType>() || ptrBase->isa<AbstractFunction>() || ptrBase->isa<AbstractTuple>() … in ValidateAbstract()
|
| D | action.cc | 216 …a<ValueNode>() || (prev_inferred != nullptr && prev_inferred->isa<abstract::AbstractFunction>())) { in AbstractAnalyze()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | cell.cc | 26 using mindspore::abstract::AbstractFunction;
|
| D | func_graph.h | 97 class AbstractFunction; variable
98 using AbstractFunctionPtr = std::shared_ptr<AbstractFunction>;
|
| D | func_graph_extends.cc | 32 using mindspore::abstract::AbstractFunction;
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | unpack_call.cc | 35 using mindspore::abstract::AbstractFunction;
|
| D | composite.cc | 53 using mindspore::abstract::AbstractFunction;
665 AbstractFunctionPtr fn = dyn_cast<AbstractFunction>(args_spec_list[0]); in GenerateFuncGraph()
|
| /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()
608 return AbstractFunction::MakeAbstractFunction(jv); in InferImplJ()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | stack_frame.cc | 130 auto res = DoJump(engine, cnode, dyn_cast<AbstractFunction>(maybe_func)); in Jump()
|
| D | program_specialize.cc | 429 AbstractFunctionPtr real_a = dyn_cast<AbstractFunction>(abs); in BuildSpecializedNode()
697 if (func->abstract()->isa<AbstractFunction>() && !func->abstract()->isa<AbstractFuncUnion>()) { in ProcessCNode()
845 AbstractFunctionPtr abs = dyn_cast<AbstractFunction>(ival); in BuildPossibleValueNode()
|
| D | static_analysis.cc | 286 AbstractFunctionPtr func = dyn_cast<AbstractFunction>(possible_func); in EvalCNode()
749 if (abstract->isa<AbstractFunction>()) { in NeedWaitForBranches()
755 [](const AbstractBasePtr &item) { return item->isa<AbstractFunction>(); })) { in NeedWaitForBranches()
|
| D | prim.cc | 488 } else if (abs_base->isa<AbstractFunction>()) { in ConvertAbstractToPython()
821 AbstractFunctionPtr abs_func = dyn_cast<abstract::AbstractFunction>(abs_ptr); in StaticGetterInferred()
1364 auto func = CheckArg<AbstractFunction>("partial", args_spec_list, 0); in Run()
1393 auto ret = AbstractFunction::MakeAbstractFunction(partial_funcs_list); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | cse.cc | 33 using mindspore::abstract::AbstractFunction;
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | transform.cc | 41 using mindspore::abstract::AbstractFunction;
449 … dyn_cast<AbstractFunction>(const_primitive_node->abstract())); in TraverseGraphMap()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | prim_test.cc | 290 AbstractFunctionPtr res_J_1 = dyn_cast<AbstractFunction>(res_J->elements()[1]); in TEST_F()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | infer_mindir.cc | 496 abstract::AbstractFunctionPtr func = dyn_cast<abstract::AbstractFunction>(possible_func); in InferCNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | incorporate_getitem.h | 298 abs_tuple[1]->isa<abstract::AbstractFunction>()) { in ShouldTransform()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 822 if (cnode->abstract()->isa<abstract::AbstractFunction>()) { in ProcessNodeRetFunc()
|