Home
last modified time | relevance | path

Searched refs:AbstractBasePtrList (Results 1 – 25 of 217) sorted by relevance

123456789

/third_party/mindspore/mindspore/core/abstract/
Dinfer_functions.h29 const AbstractBasePtrList &args_spec_list);
31 const AbstractBasePtrList &args_spec_list);
33 const AbstractBasePtrList &args_spec_list);
35 const AbstractBasePtrList &args_spec_list);
37 const AbstractBasePtrList &args_spec_list);
39 const AbstractBasePtrList &args_spec_list);
41 const AbstractBasePtrList &args_spec_list);
43 const AbstractBasePtrList &args_spec_list);
45 const AbstractBasePtrList &args_spec_list);
47 const AbstractBasePtrList &args_spec_list);
[all …]
Dprim_structures.cc26 const AbstractBasePtrList &args_spec_list) { in InferImplMakeTuple()
31 const AbstractBasePtrList &args_spec_list) { in InferImplMakeList()
36 const AbstractBasePtrList &args_spec_list) { in InferImplMakeDict()
50 AbstractBasePtrList key_list = keys->elements(); in InferImplMakeDict()
51 AbstractBasePtrList value_list = values->elements(); in InferImplMakeDict()
66 const AbstractBasePtrList &args_spec_list) { in InferImplMakeKwarg()
82 const AbstractBasePtrList &args_spec_list) { in InferImplExtractKwarg()
104 const AbstractBasePtrList &args_spec_list) { in InferImplMakeSlice()
108 AbstractBasePtrList slice_args; in InferImplMakeSlice()
159 AbstractBasePtr InferTupleOrListGetItem(const std::string &op_name, const AbstractBasePtrList &args… in InferTupleOrListGetItem()
[all …]
Dprim_others.cc37 const AbstractBasePtrList &args_spec_list) { in InferImplIdentity()
44 const AbstractBasePtrList &args_spec_list) { in InferImplEnvGetItem()
77 const AbstractBasePtrList &args_spec_list) { in InferImplEnvSetItem()
95 const AbstractBasePtrList &args_spec_list) { in InferImplEnvAdd()
101 …rImplMakeRefKey(const AnalysisEnginePtr &, const PrimitivePtr &prim, const AbstractBasePtrList &) { in InferImplMakeRefKey() argument
116 const AbstractBasePtrList &args_spec_list) { in InferImplMakeRef()
127 const AbstractBasePtrList &args_spec_list) { in InferImplGetRefKey()
142 const AbstractBasePtrList &args_spec_list) { in InferImplGetRefValue()
158 const AbstractBasePtrList &args_spec_list) { in InferImplStateSetItem()
171 const AbstractBasePtrList &args_spec_list) { in InferImplDepend()
[all …]
Danalysis_context.h35 …std::unordered_map<AbstractBasePtrList, AnalysisContextWeakPtr, AbstractBasePtrListHasher, Abstrac…
40 …xt(const AnalysisContextPtr &parent, const FuncGraphPtr &fg, const AbstractBasePtrList &args_spec_… in AnalysisContext()
49 …AnalysisContextPtr NewContext(const FuncGraphPtr &func_graph, const AbstractBasePtrList &args_spec…
61 AbstractBasePtrList args_spec_list() { return args_spec_list_; } in args_spec_list()
64 const AbstractBasePtrList &args_spec_list);
69 AbstractBasePtrList args_spec_list_;
Dprim_statement.cc26 const AbstractBasePtrList &args_spec_list) { in InferImplReturn()
38 const AbstractBasePtrList &args_spec_list) { in InferImplSwitch()
70 const AbstractBasePtrList &args_spec_list) { in InferImplSwitchLayer()
87 AbstractBasePtrList branches = branches_abs->elements(); in InferImplSwitchLayer()
127 const AbstractBasePtrList &args_spec_list) { in InferImplIs_()
143 const AbstractBasePtrList &args_spec_list) { in InferImplIsNot()
159 bool IsInDict(const PrimitivePtr &primitive, const AbstractBasePtrList &args_spec_list) { in IsInDict()
177 const AbstractBasePtrList &args_spec_list) { in InferImplInDict()
184 const AbstractBasePtrList &args_spec_list) { in InferImplNotInDict()
191 const AbstractBasePtrList &args_spec_list) { in InferImplIsConstant()
Dutils.h37 AbstractBasePtr AbstractJoin(const AbstractBasePtrList &args_spec_list);
38 AbstractBasePtrList AbstractJoin(const AbstractBasePtrList &spec1, const AbstractBasePtrList &spec2…
64 int64_t GetUnsortedSegmentOpScalarArg(const AbstractBasePtrList &args_spec_list, const std::string …
Dabstract_function.h188 PartialAbstractClosure(const AbstractFuncAtomPtr &fn, const AbstractBasePtrList &args_spec_list,
195 const AbstractBasePtrList &args() { return args_spec_list_; } in args()
209 AbstractBasePtrList args_spec_list_;
238 …VirtualAbstractClosure(const AbstractBasePtrList &args_spec_list, const AbstractBasePtr &output_sp… in VirtualAbstractClosure()
245 AbstractBasePtrList args_spec_list() { return args_spec_list_; } in MS_DECLARE_PARENT()
257 AbstractBasePtrList args_spec_list_;
267 TypedPrimitiveAbstractClosure(const PrimitivePtr prim, const AbstractBasePtrList &args_spec_list, in TypedPrimitiveAbstractClosure()
274 AbstractBasePtrList args_spec_list() { return args_spec_list_; } in args_spec_list()
286 AbstractBasePtrList args_spec_list_;
292 …PyInterpretAbstractClosure(const AbstractFuncAtomPtr &fn, const AbstractBasePtrList &args_spec_lis…
[all …]
Dprim_nn.cc42 const AbstractBasePtrList &args_spec_list) { in InferImplPooling()
108 const AbstractBasePtrList &args_spec_list) { in InferImplPoolingGrad()
128 const AbstractBasePtrList &args_spec_list) { in InferImplBatchNorm()
181 AbstractBasePtrList rets = {output, output_gamma, output_gamma, output_gamma, output_gamma}; in InferImplBatchNorm()
186 const AbstractBasePtrList &args_spec_list) { in InferImplFusedSparseAdam()
199 AbstractBasePtrList rets = {dx, dscale, dbias}; in InferImplFusedSparseAdam()
244 const AbstractBasePtrList &args_spec_list) { in InferImplConv2D()
345 const AbstractBasePtrList &args_spec_list) { in InferImplBiasAdd()
377 const AbstractBasePtrList &args_spec_list) { in InferImplBiasAddGrad()
400 const AbstractBasePtrList &args_spec_list) { in InferImplHSigmoid()
[all …]
Dprim_arrays.cc30 const AbstractBasePtrList &args_spec_list) { in InferImplScalarToArray()
39 const AbstractBasePtrList &args_spec_list) { in InferImplArrayToScalar()
53 const AbstractBasePtrList &args_spec_list) { in InferImplBroadCastShape()
84 AbstractBasePtrList elems; in InferImplBroadCastShape()
93 const AbstractBasePtrList &args_spec_list) { in InferImplStack()
135 const AbstractBasePtrList &args_spec_list) { in InferImplUnique()
175 AbstractBasePtrList elements = {ids, ids_idx}; in InferImplUnique()
180 const AbstractBasePtrList &args_spec_list) { in InferImplPadAndShift()
202 const AbstractBasePtrList &args_spec_list) { in InferImplUniqueGrad()
227 const AbstractBasePtrList &args_spec_list) { in InferImplUnsortedSegmentSum()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.h23 const AbstractBasePtrList &args_spec_list);
25 const AbstractBasePtrList &args_spec_list);
27 const AbstractBasePtrList &args_spec_list);
29 const AbstractBasePtrList &args_spec_list);
31 const AbstractBasePtrList &args_spec_list);
33 const AbstractBasePtrList &args_spec_list);
35 const AbstractBasePtrList &args_spec_list);
37 const AbstractBasePtrList &args_spec_list);
39 const AbstractBasePtrList &args_spec_list);
41 const AbstractBasePtrList &args_spec_list);
[all …]
Dops_front_infer_function.cc47 AbstractBasePtr InferImplTupleOrListEqual(const std::string &op_name, const AbstractBasePtrList &ar… in InferImplTupleOrListEqual()
58 void CalcSlidePara(const AbstractBasePtrList &args_spec_list, SlideInfo *slide) { in CalcSlidePara()
154 AbstractBasePtrList abs_list_x; in BroadcastGradientArgsDiff()
155 AbstractBasePtrList abs_list_y; in BroadcastGradientArgsDiff()
162 AbstractBasePtrList elem_list; in BroadcastGradientArgsDiff()
170 const AbstractBasePtrList &args_spec_list) { in InferImplTypeof()
183 const AbstractBasePtrList &args_spec_list) { in InferImplHasType()
222 AbstractBasePtrList values(x_rank, std::make_shared<AbstractScalar>(size)); in DoInferReduceShape()
235 AbstractBasePtrList values; in DoInferReduceShape()
251 const AbstractBasePtrList &args_spec_list) { in InferImplBroadcastGradientArgs()
[all …]
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Devaluator.h36 …std::unordered_map<AbstractBasePtrList, AttrValueMapPtr, AbstractBasePtrListHasher, AbstractBasePt…
37 using EvaluatorAttrCache = MultiThreadCache<AbstractBasePtrList, AttrValueMapPtr, EvaluatorAttrMap>;
55 virtual EvalResultPtr Eval(AnalysisEnginePtr engine, const AbstractBasePtrList &args_spec_list,
60 …virtual AbstractBasePtrList NormalizeArgs(const AbstractBasePtrList &args_spec_list) const { retur… in NormalizeArgs()
62 virtual AbstractBasePtrList BroadenUndeterminedArgs(const AbstractBasePtrList &args_spec_list) { in BroadenUndeterminedArgs()
66 virtual EvalResultPtr AbstractEval(const AbstractBasePtrList &args_spec_list) { in AbstractEval()
113 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) final… in Eval() argument
125 …virtual EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args_sp…
136 …virtual EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args_sp…
158 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
[all …]
Dprim.h39 EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args) override;
45 EvalResultPtr EvalPyCheckPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args);
47 const AbstractBasePtrList &args);
60 EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args) override;
76 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
91 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
107 …EvalResultPtr Eval(AnalysisEnginePtr, const AbstractBasePtrList &, const AnfNodeConfigPtr &) overr… in Eval() argument
143 EvalResultPtr EvalPrim(const AnalysisEnginePtr &engine, const AbstractBasePtrList &args) override;
147 AbstractBasePtrList NormalizeArgs(const AbstractBasePtrList &args_spec_list) const override { in NormalizeArgs()
149 AbstractBasePtrList broadened_args_spec_list; in NormalizeArgs()
Devaluator.cc33 string EvalEntryLogging(const EvaluatorPtr &evaluator, const AbstractBasePtrList &arg_spec_list, in EvalEntryLogging()
47 void EvalFailLogging(const EvaluatorPtr &evaluator, const AbstractBasePtrList &, const AnfNodeConfi… in EvalFailLogging() argument
186 EvalResultPtr BaseFuncGraphEvaluator::Eval(AnalysisEnginePtr engine, const AbstractBasePtrList &arg… in Eval()
276 void BroadenArgs(const AbstractBasePtrList &args_spec_list, AbstractBasePtrList *broaded_args) { in BroadenArgs()
288 AbstractBasePtrList FuncGraphEvaluator::NormalizeArgs(const AbstractBasePtrList &args_spec_list) co… in NormalizeArgs()
291 AbstractBasePtrList broaded_list; in NormalizeArgs()
300 AbstractBasePtrList FuncGraphEvaluator::BroadenUndeterminedArgs(const AbstractBasePtrList &args_spe… in BroadenUndeterminedArgs()
316 FuncGraphPtr FuncGraphEvaluator::GetFuncGraph(AnalysisEnginePtr engine, const AbstractBasePtrList &… in GetFuncGraph()
340 …taFuncGraphEvaluator::GetFuncGraph(AnalysisEnginePtr engine, const AbstractBasePtrList &args_spec_… in GetFuncGraph()
364 AbstractBasePtrList args_spec_list; in Run()
[all …]
Dprogram_specialize.h109 const AbstractBasePtrList &args_spec_list);
129 const AbstractBasePtrList &argvals);
131 … const AbstractFunctionPtr &func, const AbstractBasePtrList &args,
136 const AbstractBasePtrList &argvals,
137 std::pair<AbstractBasePtrList, AbstractBasePtr> *result);
141 std::pair<AbstractBasePtrList, AbstractBasePtr> BuildFromBroadedArgsVal(const EvaluatorPtr &eval);
Dstatic_analysis.h199 std::size_t operator()(const std::pair<AbstractFunctionPtr, AbstractBasePtrList> &p) const { in operator()
208 …EvaluatorArgs(const EvaluatorPtr &eval, const AbstractBasePtrList &args) : evaluator_(eval), args_… in EvaluatorArgs()
221 AbstractBasePtrList args_;
244 AnalysisResult Run(const FuncGraphPtr &func_graph, const AbstractBasePtrList &args_spec_list);
255 EvalResultPtr Execute(const AbstractFunctionPtr &fn, const AbstractBasePtrList &args_spec_list);
286 …static EvalResultPtr ProcessEvalResults(const AbstractBasePtrList &out_specs, const AnfNodePtr &no…
291 … const AbstractBasePtrList &args_spec_list, const EvalTraceRevIter &it,
297 …std::unordered_map<std::pair<AbstractFunctionPtr, AbstractBasePtrList>, EvaluatorPtr, PartialAppHa…
351 EvalResultPtr EvalOnePrim(const PrimitivePtr &p, const AbstractBasePtrList &arg_specs);
/third_party/mindspore/tests/ut/cpp/operator/
Dcomposite_test.cc92 AbstractBasePtrList eles; in TEST_F()
101 AbstractBasePtrList args_spec_list = {tuple_tensor, start_index, stop_index}; in TEST_F()
117 AbstractBasePtrList eles; in TEST_F()
125 AbstractBasePtrList args_spec_list = {tuple_tensor, start_index}; in TEST_F()
149 AbstractBasePtrList eles; in TEST_F()
160 AbstractBasePtrList args_spec_list = {tuple_tensor, slice}; in TEST_F()
175 AbstractBasePtrList eles; in TEST_F()
186 AbstractBasePtrList args_spec_list = {tuple_tensor, slice}; in TEST_F()
201 AbstractBasePtrList eles; in TEST_F()
212 AbstractBasePtrList args_spec_list = {tuple_tensor, slice}; in TEST_F()
[all …]
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dprim_test.cc65 AbstractBasePtrList te; in ShapeOf()
73 AbstractBasePtrList te; in ListShapeOf()
134 AbstractBasePtrList args_spec_list; in TEST_F()
150 AbstractBasePtrList args_spec_list; in TEST_F()
154 …auto abstract_list1 = std::make_shared<AbstractList>(AbstractBasePtrList({abstract_v1, abstract_u1… in TEST_F()
157 …auto abstract_list2 = std::make_shared<AbstractList>(AbstractBasePtrList({abstract_v2, abstract_u2… in TEST_F()
169 …AbstractBasePtrList({FromValue(static_cast<int64_t>(3), false), FromValue(static_cast<int64_t>(3),… in TEST_F()
177 AbstractBasePtrList args_spec_list; in TEST_F()
182 …auto abstract_list = std::make_shared<AbstractList>(AbstractBasePtrList({abstract_v1, abstract_v2}… in TEST_F()
200 AbstractBasePtrList args_spec_list; in TEST_F()
[all …]
Dstatic_analysis_test.cc39 const AbstractBasePtrList &args_spec_list) { in InferImplScalarAddStub()
56 const AbstractBasePtrList &args_spec_list) { in InferImplReturnStub()
156 AbstractBasePtrList args_spec_list; in TEST_F()
260 AbstractBasePtrList args_spec_list; in TEST_F()
297 AnalysisContextPtr f_context = dummy_context->NewContext(graph_f_, AbstractBasePtrList()); in TEST_F()
301 AnalysisContextPtr g_context = f_context->NewContext(graph_g_, AbstractBasePtrList()); in TEST_F()
306 AnalysisContextPtr alpha_context = dummy_context->NewContext(graph_alpha_, AbstractBasePtrList()); in TEST_F()
310 AnalysisContextPtr beta_context = alpha_context->NewContext(graph_beta_, AbstractBasePtrList()); in TEST_F()
354 AbstractBasePtrList args_spec_list; in TEST_F()
383 AbstractBasePtrList args_spec; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.h47 using TupleListKey = std::pair<PrimitivePtr, abstract::AbstractBasePtrList>;
50 …std::unordered_map<abstract::AbstractBasePtrList, PrimBpropOptGraphLevel2InfoPtr, abstract::Abstra…
152 …ECacheQrtRes GetOptBpfgFromCache(const PrimitivePtr &prim, const abstract::AbstractBasePtrList &ab…
157 …bs(const PrimitivePtr &prim, const ValuePtrList &op_args, abstract::AbstractBasePtrList *abs_list);
160 …abstract::AbstractBasePtrList AddOutToAbsList(const ValuePtr &out, const abstract::AbstractBasePtr…
167 … const abstract::AbstractBasePtrList &abs_list_input);
169 …void BindAbsToParameters(const FuncGraphPtr &bprop_fg, const abstract::AbstractBasePtrList &abs_li…
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.h69 …abstract::AbstractBasePtrList NormalizeArgs(const abstract::AbstractBasePtrList &args_spec_list) c…
111 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
128 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
138 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
154 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
172 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list);
184 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
194 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
204 FuncGraphPtr GenerateFuncGraph(const AbstractBasePtrList &args_spec_list) override;
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dlamb_next_mv_rule_test.cc60 AbstractBasePtrList args_spec_list; in TEST_F()
103 AbstractBasePtrList args_spec_list; in TEST_F()
146 AbstractBasePtrList args_spec_list; in TEST_F()
189 AbstractBasePtrList args_spec_list; in TEST_F()
232 AbstractBasePtrList args_spec_list; in TEST_F()
252 AbstractBasePtrList args_spec_list; in TEST_F()
272 AbstractBasePtrList args_spec_list; in TEST_F()
292 AbstractBasePtrList args_spec_list; in TEST_F()
312 AbstractBasePtrList args_spec_list; in TEST_F()
332 AbstractBasePtrList args_spec_list; in TEST_F()
[all …]
Dadam_apply_one_fusion_test.cc53 AbstractBasePtrList args_spec_list; in TEST_F()
91 AbstractBasePtrList args_spec_list; in TEST_F()
129 AbstractBasePtrList args_spec_list; in TEST_F()
167 AbstractBasePtrList args_spec_list; in TEST_F()
205 AbstractBasePtrList args_spec_list; in TEST_F()
225 AbstractBasePtrList args_spec_list; in TEST_F()
245 AbstractBasePtrList args_spec_list; in TEST_F()
265 AbstractBasePtrList args_spec_list; in TEST_F()
285 AbstractBasePtrList args_spec_list; in TEST_F()
305 AbstractBasePtrList args_spec_list; in TEST_F()
Dadam_apply_one_with_decay_rule_test.cc38 AbstractBasePtrList args_spec_list; in TEST_F()
59 AbstractBasePtrList args_spec_list; in TEST_F()
80 AbstractBasePtrList args_spec_list; in TEST_F()
101 AbstractBasePtrList args_spec_list; in TEST_F()
122 AbstractBasePtrList args_spec_list; in TEST_F()
143 AbstractBasePtrList args_spec_list; in TEST_F()
164 AbstractBasePtrList args_spec_list; in TEST_F()
185 AbstractBasePtrList args_spec_list; in TEST_F()
206 AbstractBasePtrList args_spec_list; in TEST_F()
227 AbstractBasePtrList args_spec_list; in TEST_F()
Dlamb_next_mv_with_decay_rule_test.cc61 AbstractBasePtrList args_spec_list; in TEST_F()
105 AbstractBasePtrList args_spec_list; in TEST_F()
150 AbstractBasePtrList args_spec_list; in TEST_F()
195 AbstractBasePtrList args_spec_list; in TEST_F()
240 AbstractBasePtrList args_spec_list; in TEST_F()
261 AbstractBasePtrList args_spec_list; in TEST_F()
281 AbstractBasePtrList args_spec_list; in TEST_F()
302 AbstractBasePtrList args_spec_list; in TEST_F()
322 AbstractBasePtrList args_spec_list; in TEST_F()
343 AbstractBasePtrList args_spec_list; in TEST_F()
[all …]

123456789