| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | lamb_next_right_rule.cc | 80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(new_node_abstract_list); in Process() local
81 MS_EXCEPTION_IF_NULL(abstract_tuple); in Process()
82 new_node->set_abstract(abstract_tuple); in Process()
|
| D | add_input_to_output.cc | 73 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in AddOutputs() local
74 cnode->set_abstract(abstract_tuple); in AddOutputs()
|
| D | fused_batch_norm_fusion.cc | 105 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce() local
106 bn_training_reduce->set_abstract(abstract_tuple); in CreateBNTrainingReduce()
161 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdate() local
162 bn_training_update->set_abstract(abstract_tuple); in CreateBNTrainingUpdate()
|
| D | lamb_next_mv_rule.cc | 84 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(new_abstracts); in CreateLambNextMVNode() local
85 MS_EXCEPTION_IF_NULL(abstract_tuple); in CreateLambNextMVNode()
86 lamb_next_mv_rule->set_abstract(abstract_tuple); in CreateLambNextMVNode()
|
| D | lamb_next_mv_with_decay_rule.cc | 41 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(new_node_list); in GetLambNextMVWithDecayOutput() local
42 MS_EXCEPTION_IF_NULL(abstract_tuple); in GetLambNextMVWithDecayOutput()
43 new_node->set_abstract(abstract_tuple); in GetLambNextMVWithDecayOutput()
|
| D | adam_apply_one_fusion.cc | 273 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(new_node_abstract_list); in Process() local
274 new_node->set_abstract(abstract_tuple); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | combine_momentum_fusion.cc | 96 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run() local
97 MS_EXCEPTION_IF_NULL(abstract_tuple); in Run()
98 combine_mom->set_abstract(abstract_tuple); in Run()
|
| D | combine_cast_fusion.cc | 113 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run() local
114 MS_EXCEPTION_IF_NULL(abstract_tuple); in Run()
115 cast_all->set_abstract(abstract_tuple); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | batch_norm_bert_fission.cc | 80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce() local
81 bn_training_reduce->set_abstract(abstract_tuple); in CreateBNTrainingReduce()
121 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateV2() local
122 bn_training_update_v2->set_abstract(abstract_tuple); in CreateBNTrainingUpdateV2()
|
| D | single_batch_norm_fission.cc | 46 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce() local
47 bn_training_reduce->set_abstract(abstract_tuple); in CreateBNTrainingReduce()
|
| D | batch_norm_grad_infer_fission.cc | 139 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateGrad() local
140 bn_training_update_grad->set_abstract(abstract_tuple); in CreateBNTrainingUpdateGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.cc | 80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in Run() local
81 MS_EXCEPTION_IF_NULL(abstract_tuple); in Run()
82 proxy_node->set_abstract(abstract_tuple); in Run()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | neighborexchange.cc | 113 auto abstract_tuple = input_args[0]->cast<abstract::AbstractTuplePtr>(); in Check() local
114 MS_EXCEPTION_IF_NULL(abstract_tuple); in Check()
115 auto abstract_element = abstract_tuple->elements(); in Check()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.cc | 191 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in DoFusion() local
192 MS_EXCEPTION_IF_NULL(abstract_tuple); in DoFusion()
193 fused_cnode->set_abstract(abstract_tuple); in DoFusion()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | optimizer_unify_output.cc | 52 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in ProcessOutput() local
53 cnode_ptr->set_abstract(abstract_tuple); in ProcessOutput()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/common/ |
| D | utils.cc | 74 auto abstract_tuple = in GetShapeVectorFromCNode() local
76 auto abstract_list = abstract_tuple->elements(); in GetShapeVectorFromCNode()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | infershape_pass.cc | 368 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in SetSubGraphAbstract() local
369 CHECK_NULL_RETURN(abstract_tuple); in SetSubGraphAbstract()
370 cnode->set_abstract(abstract_tuple); in SetSubGraphAbstract()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | kpynative.cc | 664 abstract::AbstractTuplePtr abstract_tuple = nullptr; in BackPropagate() local
669 abstract_tuple = bprop_app_abstract->cast<abstract::AbstractTuplePtr>(); in BackPropagate()
670 if (abstract_tuple->size() != (cnode_primal->size() - 1)) { in BackPropagate()
671 MS_LOG(EXCEPTION) << "AbstractTuple size: " << abstract_tuple->ToString() in BackPropagate()
692 if (abstract_tuple != nullptr) { in BackPropagate()
694 din->set_abstract((*abstract_tuple)[i - 1]); in BackPropagate()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.cc | 83 auto abstract_tuple = utils::cast<abstract::AbstractTuplePtr>(abstract_base); in ModifyCNode() local
84 abstracts = abstract_tuple->elements(); in ModifyCNode()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 604 … auto abstract_tuple = utils::cast<abstract::AbstractTuplePtr>(get_item_input_cnode->abstract()); in GetCNodeInputAbstract() local
605 auto abstract_list = abstract_tuple->elements(); in GetCNodeInputAbstract()
1200 auto abstract_tuple = abstract_base->cast<abstract::AbstractTuplePtr>(); in GetDataTypeFromAnfNode() local
1201 if (abstract_tuple->elements().empty()) { in GetDataTypeFromAnfNode()
1205 abstract_base = abstract_tuple->elements().front(); in GetDataTypeFromAnfNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 794 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in ConvertOutputTensor() local
795 MSLITE_CHECK_PTR(abstract_tuple); in ConvertOutputTensor()
796 dst_cnode->set_abstract(abstract_tuple); in ConvertOutputTensor()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 703 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in SetAbstractTuple() local
704 if (abstract_tuple == nullptr) { in SetAbstractTuple()
708 cnode->set_abstract(abstract_tuple); in SetAbstractTuple()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1233 auto abstract_tuple = std::make_shared<AbstractTuple>(abstract_list); in SetOutputTypeAndDetailShape() local
1234 node->set_abstract(abstract_tuple); in SetOutputTypeAndDetailShape()
1287 auto abstract_tuple = std::make_shared<AbstractTuple>(abstract_list); in SetOutputInferTypeAndShape() local
1288 node->set_abstract(abstract_tuple); in SetOutputInferTypeAndShape()
2246 auto abstract_tuple = real_abs->cast<abstract::AbstractTuplePtr>(); in InferShape() local
2247 MS_EXCEPTION_IF_NULL(abstract_tuple); in InferShape()
2248 auto tuple_elements = abstract_tuple->elements()[tuple_get_item_index]; in InferShape()
|