Home
last modified time | relevance | path

Searched refs:abstract_list (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dadd_input_to_output.cc44 AbstractBasePtrList abstract_list; in AddOutputs() local
51 …:copy(origin_abstract_list.begin(), origin_abstract_list.end(), std::back_inserter(abstract_list)); in AddOutputs()
53 abstract_list.emplace_back(origin_abstract); in AddOutputs()
65 abstract_list.emplace_back(node_to_output->abstract()); in AddOutputs()
73 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in AddOutputs()
Dfused_batch_norm_fusion.cc104 AbstractBasePtrList abstract_list{data_input1->abstract(), data_input2->abstract()}; in CreateBNTrainingReduce() local
105 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce()
128 … std::vector<AbstractBasePtr> *abstract_list) const { in GetBNTrainingUpdateAbstractList()
131 MS_EXCEPTION_IF_NULL(abstract_list); in GetBNTrainingUpdateAbstractList()
142 …*abstract_list = {bn_abstract_tuple->elements()[kIndex0], variable_input0->abstract(), variable_in… in GetBNTrainingUpdateAbstractList()
159 AbstractBasePtrList abstract_list; in CreateBNTrainingUpdate() local
160 GetBNTrainingUpdateAbstractList(equiv, bn, &abstract_list); in CreateBNTrainingUpdate()
161 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdate()
Dfused_batch_norm_fusion.h54 std::vector<AbstractBasePtr> *abstract_list) const;
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcombine_momentum_fusion.cc88 AbstractBasePtrList abstract_list; in Run() local
92 abstract_list.push_back(cnode->abstract()); in Run()
96 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run()
Dcombine_cast_fusion.cc105 AbstractBasePtrList abstract_list; in Run() local
109 abstract_list.push_back(cnode->abstract()); in Run()
113 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_bert_fission.cc79 AbstractBasePtrList abstract_list{bn_input1->abstract(), bn_input2->abstract()}; in CreateBNTrainingReduce() local
80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce()
118 std::vector<AbstractBasePtr> abstract_list{bn_abstract_tuple->elements()[kIndex0], in CreateBNTrainingUpdateV2() local
121 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateV2()
Dsingle_batch_norm_fission.cc45 AbstractBasePtrList abstract_list{bn_input1->abstract(), bn_input1->abstract()}; in CreateBNTrainingReduce() local
46 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce()
Dbatch_norm_grad_infer_fission.cc137 std::vector<AbstractBasePtr> abstract_list{bn_grad_abstract_tuple->elements()[kIndex1], in CreateBNTrainingUpdateGrad() local
139 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateGrad()
/third_party/mindspore/mindspore/core/ops/
Dprimitive_c.cc29 AbstractBasePtr PrimitiveC::Infer(const AbstractBasePtrList &abstract_list) { in Infer() argument
36 return infer_function(nullptr, shared_from_base<Primitive>(), abstract_list); in Infer()
Dprimitive_c.h44 AbstractBasePtr Infer(const AbstractBasePtrList &abstract_list);
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dreplace_node_by_proxy.cc76 AbstractBasePtrList abstract_list; in Run() local
79 abstract_list.push_back(cnode->abstract()); in Run()
80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Doptimizer_unify_output.cc48 std::vector<AbstractBasePtr> abstract_list; in ProcessOutput() local
50 abstract_list.push_back(abstract->Clone()); in ProcessOutput()
52 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in ProcessOutput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_cast.cc44 AbstractBasePtrList abstract_list; in InsertCastForMultipleOutput() local
76 abstract_list.emplace_back(replace_node->abstract()); in InsertCastForMultipleOutput()
81 make_tuple->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in InsertCastForMultipleOutput()
Ddeal_ref_and_split_unsupported_transdata.cc185 AbstractBasePtrList abstract_list; in DealRefForMultipleOutput() local
196 abstract_list.push_back(final_node->abstract()); in DealRefForMultipleOutput()
202 make_tuple->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in DealRefForMultipleOutput()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc76 auto abstract_list = abstract_tuple->elements(); in GetShapeVectorFromCNode() local
77 if (abstract_list.size() <= idx) { in GetShapeVectorFromCNode()
81 cnode_abstract = abstract_list[idx]; in GetShapeVectorFromCNode()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc198 AbstractBasePtrList abstract_list; in IdentifyWhileNodeOutput() local
199 abstract_list.resize(output_exit_nodes_.size()); in IdentifyWhileNodeOutput()
200 std::transform(output_exit_nodes_.begin(), output_exit_nodes_.end(), abstract_list.begin(), in IdentifyWhileNodeOutput()
202 while_node_->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in IdentifyWhileNodeOutput()
220 AbstractBasePtrList abstract_list; in UpdateExitNodeUser() local
226 abstract_list.emplace_back(abstract); in UpdateExitNodeUser()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc339 AbstractBasePtrList abstract_list; in SetSubGraphAbstract() local
344 abstract_list.emplace_back(abstract_base->Clone()); in SetSubGraphAbstract()
368 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in SetSubGraphAbstract()
372 if (abstract_list.size() != 1) { in SetSubGraphAbstract()
375 cnode->set_abstract(abstract_list.front()); in SetSubGraphAbstract()
Ddecrease_transpose_algo.cc587 AbstractBasePtrList abstract_list; in SetSubGraphAbstract() local
592 abstract_list.emplace_back(abstract_base->Clone()); in SetSubGraphAbstract()
616 cnode->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in SetSubGraphAbstract()
618 if (abstract_list.size() != 1) { in SetSubGraphAbstract()
621 cnode->set_abstract(abstract_list.front()); in SetSubGraphAbstract()
Dnode_infershape.cc502 AbstractBasePtrList abstract_list; in SetCNodeAbstract() local
519 abstract_list.emplace_back(new_abstract); in SetCNodeAbstract()
521 auto new_abstract_list = std::make_shared<abstract::AbstractTuple>(abstract_list); in SetCNodeAbstract()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Deliminate_redundant_output.cc231 AbstractBasePtrList abstract_list; in ReplaceMakeTuple() local
239 abstract_list.push_back(old_maketuple->input(i + 1)->abstract()); in ReplaceMakeTuple()
253 make_tuple->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in ReplaceMakeTuple()
Ddecrease_transfer_precision.cc211 AbstractBasePtrList abstract_list; in Process_Father() local
213 (void)abstract_list.emplace_back(tuple_output->input(i)->abstract()); in Process_Father()
215 tuple_output->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in Process_Father()
/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc185 AbstractBasePtrList abstract_list; in DoFusion() local
189 abstract_list.push_back(cnode->abstract()); in DoFusion()
191 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in DoFusion()
/third_party/mindspore/mindspore/lite/tools/converter/acl/
Dacl_pass.cc329 AbstractBasePtrList abstract_list; in SetMultiOutputs() local
336 abstract_list.emplace_back(abstract_tensor); in SetMultiOutputs()
338 new_cnode->set_abstract(std::make_shared<abstract::AbstractTuple>(abstract_list)); in SetMultiOutputs()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dclean.cc84 …std::vector<AbstractBasePtr> abstract_list{abs_sparse->indices(), abs_sparse->values(), abs_sparse… in AdaptAbs() local
85 return std::make_shared<AbstractTuple>(abstract_list); in AdaptAbs()
90 std::vector<AbstractBasePtr> abstract_list{abs_row_tensor->indices(), abs_row_tensor->values(), in AdaptAbs() local
92 return std::make_shared<AbstractTuple>(abstract_list); in AdaptAbs()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc581 AbstractBasePtrList abstract_list; in ConvertTop() local
588 abstract_list.emplace_back(abstract); in ConvertTop()
604 auto abstract = std::make_shared<abstract::AbstractTuple>(abstract_list); in ConvertTop()

12