| /third_party/mindspore/mindspore/core/abstract/ |
| D | infer_functions.h | 28 AbstractBasePtr InferImplReturn(const AnalysisEnginePtr &, const PrimitivePtr &,
30 AbstractBasePtr InferImplSwitch(const AnalysisEnginePtr &, const PrimitivePtr &,
32 AbstractBasePtr InferImplSwitchLayer(const AnalysisEnginePtr &, const PrimitivePtr &,
34 AbstractBasePtr InferImplIs_(const AnalysisEnginePtr &, const PrimitivePtr &,
36 AbstractBasePtr InferImplIsNot(const AnalysisEnginePtr &, const PrimitivePtr &,
38 AbstractBasePtr InferImplInDict(const AnalysisEnginePtr &, const PrimitivePtr &,
40 AbstractBasePtr InferImplNotInDict(const AnalysisEnginePtr &, const PrimitivePtr &,
42 AbstractBasePtr InferImplIsConstant(const AnalysisEnginePtr &, const PrimitivePtr &,
44 AbstractBasePtr InferImplPooling(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
46 AbstractBasePtr InferImplPoolingGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
[all …]
|
| D | prim_others.cc | 36 AbstractBasePtr InferImplIdentity(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplIdentity() argument
43 AbstractBasePtr InferImplEnvGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvGetItem() argument
76 AbstractBasePtr InferImplEnvSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvSetItem() argument
94 AbstractBasePtr InferImplEnvAdd(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvAdd() argument
101 AbstractBasePtr InferImplMakeRefKey(const AnalysisEnginePtr &, const PrimitivePtr &prim, const Abst… in InferImplMakeRefKey() argument
115 AbstractBasePtr InferImplMakeRef(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplMakeRef() argument
126 AbstractBasePtr InferImplGetRefKey(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplGetRefKey() argument
141 AbstractBasePtr InferImplGetRefValue(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplGetRefValue() argument
157 AbstractBasePtr InferImplStateSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplStateSetItem() argument
170 AbstractBasePtr InferImplDepend(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplDepend() argument
[all …]
|
| D | prim_structures.cc | 25 AbstractBasePtr InferImplMakeTuple(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplMakeTuple() argument
30 AbstractBasePtr InferImplMakeList(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplMakeList() argument
35 AbstractBasePtr InferImplMakeDict(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeDict() argument
65 AbstractBasePtr InferImplMakeKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeKwarg() argument
81 AbstractBasePtr InferImplExtractKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplExtractKwarg() argument
103 AbstractBasePtr InferImplMakeSlice(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeSlice() argument
217 AbstractBasePtr InferImplTupleGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleGetItem() argument
222 AbstractBasePtr InferImplListGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplListGetItem() argument
227 AbstractBasePtr InferImplTupleSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleSetItem() argument
232 AbstractBasePtr InferImplListSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplListSetItem() argument
[all …]
|
| D | prim_statement.cc | 25 AbstractBasePtr InferImplReturn(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplReturn() argument
37 AbstractBasePtr InferImplSwitch(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplSwitch() argument
69 AbstractBasePtr InferImplSwitchLayer(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplSwitchLayer() argument
126 AbstractBasePtr InferImplIs_(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplIs_() argument
142 AbstractBasePtr InferImplIsNot(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplIsNot() argument
176 AbstractBasePtr InferImplInDict(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplInDict() argument
183 AbstractBasePtr InferImplNotInDict(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplNotInDict() argument
190 AbstractBasePtr InferImplIsConstant(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplIsConstant() argument
|
| D | prim_arrays.cc | 29 AbstractBasePtr InferImplScalarToArray(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplScalarToArray() argument
38 AbstractBasePtr InferImplArrayToScalar(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplArrayToScalar() argument
52 AbstractBasePtr InferImplBroadCastShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBroadCastShape() argument
92 AbstractBasePtr InferImplStack(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplStack() argument
134 AbstractBasePtr InferImplUnique(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUnique() argument
179 AbstractBasePtr InferImplPadAndShift(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplPadAndShift() argument
201 AbstractBasePtr InferImplUniqueGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUniqueGrad() argument
226 AbstractBasePtr InferImplUnsortedSegmentSum(const AnalysisEnginePtr &, const PrimitivePtr &primitiv… in InferImplUnsortedSegmentSum() argument
282 AbstractBasePtr InferImplUnsortedSegmentMax(const AnalysisEnginePtr &, const PrimitivePtr &primitiv… in InferImplUnsortedSegmentMax() argument
333 AbstractBasePtr InferImplUnsortedSegmentMin(const AnalysisEnginePtr &, const PrimitivePtr &primitiv… in InferImplUnsortedSegmentMin() argument
[all …]
|
| D | prim_nn.cc | 41 AbstractBasePtr InferImplPooling(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplPooling() argument
107 AbstractBasePtr InferImplPoolingGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplPoolingGrad() argument
127 AbstractBasePtr InferImplBatchNorm(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBatchNorm() argument
185 AbstractBasePtr InferImplFusedSparseAdam(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplFusedSparseAdam() argument
243 AbstractBasePtr InferImplConv2D(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplConv2D() argument
344 AbstractBasePtr InferImplBiasAdd(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBiasAdd() argument
376 AbstractBasePtr InferImplBiasAddGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBiasAddGrad() argument
399 AbstractBasePtr InferImplHSigmoid(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplHSigmoid() argument
409 AbstractBasePtr InferImplHSigmoidGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplHSigmoidGrad() argument
422 AbstractBasePtr InferImplBpropCut(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBpropCut() argument
[all …]
|
| D | prim_maths.cc | 25 AbstractBasePtr InferImplMinOrMaxGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMinOrMaxGrad() argument
44 AbstractBasePtr InferImplSqrt(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplSqrt() argument
54 AbstractBasePtr InferImplSqrtGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplSqrtGrad() argument
126 AbstractBasePtr InferImplReduceFunc(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplReduceFunc() argument
162 AbstractBasePtr InferImplBinaryBase(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBinaryBase() argument
210 AbstractBasePtr InferImplMinimum(const AnalysisEnginePtr &engine_ptr, const PrimitivePtr &primitive, in InferImplMinimum()
215 AbstractBasePtr InferImplDivNoNan(const AnalysisEnginePtr &engine_ptr, const PrimitivePtr &primitiv… in InferImplDivNoNan()
220 AbstractBasePtr InferImplLinSpace(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplLinSpace() argument
265 AbstractBasePtr InferImplMatMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMatMul() argument
332 AbstractBasePtr InferImplBatchMatMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBatchMatMul() argument
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.h | 22 AbstractBasePtr InferImplTypeof(const AnalysisEnginePtr &, const PrimitivePtr &,
24 AbstractBasePtr InferImplHasType(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
26 AbstractBasePtr InferImplBroadcastGradientArgs(const AnalysisEnginePtr &, const PrimitivePtr &primi…
28 AbstractBasePtr InferImplListMap(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
30 AbstractBasePtr InferImplListReduce(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
32 AbstractBasePtr InferImplTupleReversed(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
34 AbstractBasePtr InferImplReduceShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
36 AbstractBasePtr InferImplTupleDiv(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
38 AbstractBasePtr InferImplTuple2Array(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
40 AbstractBasePtr InferImplShapeMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
[all …]
|
| D | ops_front_infer_function.cc | 169 AbstractBasePtr InferImplTypeof(const AnalysisEnginePtr &, const PrimitivePtr &, in InferImplTypeof() argument
182 AbstractBasePtr InferImplHasType(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplHasType() argument
250 AbstractBasePtr InferImplBroadcastGradientArgs(const AnalysisEnginePtr &, const PrimitivePtr &primi… in InferImplBroadcastGradientArgs() argument
286 AbstractBasePtr InferImplListMap(const AnalysisEnginePtr &engine, const PrimitivePtr &primitive, in InferImplListMap()
314 AbstractBasePtr InferImplListReduce(const AnalysisEnginePtr &engine, const PrimitivePtr &primitive, in InferImplListReduce()
335 AbstractBasePtr InferImplTupleReversed(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleReversed() argument
349 AbstractBasePtr InferImplReduceShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplReduceShape() argument
389 AbstractBasePtr InferImplTupleDiv(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleDiv() argument
440 AbstractBasePtr InferImplTuple2Array(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTuple2Array() argument
456 AbstractBasePtr InferImplShapeMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplShapeMul() argument
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | evaluator.h | 52 virtual EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list,
55 virtual EvalResultPtr Eval(AnalysisEnginePtr engine, const AbstractBasePtrList &args_spec_list,
57 virtual EvalResultPtr SingleRun(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list,
113 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) final… in Eval() argument
123 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list,
125 …virtual EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args_sp…
133 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list,
136 …virtual EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args_sp…
145 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list,
158 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
[all …]
|
| D | prim.h | 39 EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args) override;
45 EvalResultPtr EvalPyCheckPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args);
46 EvalResultPtr RunPyInferValue(const AnalysisEnginePtr &engine, const AbstractBasePtr &abs_base,
60 EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args) override;
73 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &argrefs,
76 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
88 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &argrefs,
91 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
104 EvalResultPtr Run(AnalysisEnginePtr engine, const ConfigPtrList &argrefs,
107 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
[all …]
|
| D | evaluator.cc | 62 void BaseFuncGraphEvaluator::EnterStackFrame(const AnalysisEnginePtr &engine, const StackFramePtr &… in EnterStackFrame()
95 void BaseFuncGraphEvaluator::LeaveStackFrame(const AnalysisEnginePtr &, const StackFramePtr ¤… in LeaveStackFrame() argument
112 AbstractBasePtr BaseFuncGraphEvaluator::LaunchStackFrame(const AnalysisEnginePtr &engine, const Fun… in LaunchStackFrame()
159 AbstractBasePtr BaseFuncGraphEvaluator::LaunchRecursiveEval(const AnalysisEnginePtr &engine, const … in LaunchRecursiveEval()
186 EvalResultPtr BaseFuncGraphEvaluator::Eval(AnalysisEnginePtr engine, const AbstractBasePtrList &arg… in Eval()
316 FuncGraphPtr FuncGraphEvaluator::GetFuncGraph(AnalysisEnginePtr engine, const AbstractBasePtrList &… in GetFuncGraph()
340 FuncGraphPtr MetaFuncGraphEvaluator::GetFuncGraph(AnalysisEnginePtr engine, const AbstractBasePtrLi… in GetFuncGraph()
362 EvalResultPtr Evaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_list, in Run()
395 EvalResultPtr TrivialPrimEvaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_li… in Run()
414 EvalResultPtr TransitionPrimEvaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf… in Run()
[all …]
|
| D | stack_frame.h | 60 StackFramePtr Jump(const AnalysisEnginePtr &engine);
62 EvalResultPtr Step(const AnalysisEnginePtr &engine);
64 …void Back(const AnalysisEnginePtr &engine, const StackFramePtr &last_stack_frame, const EvalResult…
123 …AbstractBasePtrList GenerateArgsAbsList(const AnalysisEnginePtr &engine, const EvaluatorPtr &evalu…
127 StackFramePtr DoJump(const AnalysisEnginePtr &engine, const CNodePtr current_cnode,
|
| D | stack_frame.cc | 23 AbstractBasePtrList StackFrame::GenerateArgsAbsList(const AnalysisEnginePtr &engine, const Evaluato… in GenerateArgsAbsList()
62 StackFramePtr StackFrame::DoJump(const AnalysisEnginePtr &engine, const CNodePtr current_cnode, in DoJump()
113 StackFramePtr StackFrame::Jump(const AnalysisEnginePtr &engine) { in Jump()
138 EvalResultPtr StackFrame::Step(const AnalysisEnginePtr &engine) { in Step()
151 void StackFrame::Back(const AnalysisEnginePtr &engine, const StackFramePtr &last_stack_frame, in Back()
|
| D | prim.cc | 53 EvalResultPtr DoSignatureEvaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_li… in Run()
142 EvalResultPtr UnpackGraphEvaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_conf_li… in Run()
256 EvalResultPtr MixedPrecisionCastEvaluator::Run(AnalysisEnginePtr engine, const ConfigPtrList &args_… in Run()
576 EvalResultPtr StandardPrimEvaluator::RunPyInferValue(const AnalysisEnginePtr &engine, const Abstrac… in RunPyInferValue()
611 EvalResultPtr StandardPrimEvaluator::EvalPyCheckPrim(const AnalysisEnginePtr &engine, const Abstrac… in EvalPyCheckPrim()
633 EvalResultPtr StandardPrimEvaluator::EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePt… in EvalPrim()
675 EvalResultPtr PythonPrimEvaluator::EvalPrim(const AnalysisEnginePtr &, const AbstractBasePtrList &a… in EvalPrim() argument
701 EvalResultPtr UniformPrimEvaluator::EvalPrim(const AnalysisEnginePtr &, const AbstractBasePtrList &… in EvalPrim() argument
790 FuncGraphPtr PyObjToGraph(const AnalysisEnginePtr &engine, const ValuePtr &method) { in PyObjToGraph()
808 inline void AddToManager(const AnalysisEnginePtr &engine, const FuncGraphPtr func_graph) { in AddToManager()
[all …]
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | static_analysis_test.cc | 38 AbstractBasePtr InferImplScalarAddStub(const AnalysisEnginePtr &engine, const PrimitivePtr &, in InferImplScalarAddStub()
55 AbstractBasePtr InferImplReturnStub(const AnalysisEnginePtr &engine, const PrimitivePtr &prim, in InferImplReturnStub()
80 AnalysisEnginePtr SetupAnalysisEngineStub() { in SetupAnalysisEngineStub()
84 …AnalysisEnginePtr engine = std::make_shared<AnalysisEngine>(PrimEvaluatorConstructorsStub, graph_m… in SetupAnalysisEngineStub()
122 AnalysisEnginePtr engine_;
181 AnalysisEnginePtr engine_;
321 AnalysisEnginePtr engine_;
370 AnalysisEnginePtr engine_;
406 AnalysisEnginePtr engine_;
433 AnalysisEnginePtr engine_;
|
| D | helper.cc | 24 AnalysisEnginePtr SetupAnalysisEngine() { in SetupAnalysisEngine()
27 …AnalysisEnginePtr engine = std::make_shared<AnalysisEngine>(GetPrimEvaluatorConstructors(), graph_… in SetupAnalysisEngine()
|
| D | helper.h | 24 AnalysisEnginePtr SetupAnalysisEngine();
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.h | 62 abstract::AnalysisEnginePtr engine() { return engine_; } in engine()
92 abstract::AnalysisEnginePtr engine_;
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/common/ |
| D | restore_abs_input_in_backed_infer_test.cc | 40 AbstractBasePtr InferImplAttrTest(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitiv… in InferImplAttrTest() argument
48 AbstractBasePtr InferImplDynamicInputTest(const abstract::AnalysisEnginePtr &, const PrimitivePtr &… in InferImplDynamicInputTest() argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | layer_norm_x_backprop_v2.h | 38 AbstractBasePtr LayerNormXBackpropV2Infer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &…
|
| D | LayerNormXBackprop.h | 38 AbstractBasePtr LayerNormXBackpropInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &pr…
|
| D | LayerNormBetaGammaBackprop.h | 38 AbstractBasePtr LayerNormBetaGammaBackpropInfer(const abstract::AnalysisEnginePtr &, const Primitiv…
|
| D | reduce_sum.h | 42 AbstractBasePtr ReduceSumInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive,
|
| D | apply_adagrad_d_a.h | 48 AbstractBasePtr ApplyAdagradDAInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primit…
|