Searched refs:abs_list (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | prim_bprop_optimizer.cc | 209 abstract::AbstractBasePtrList abs_list; in GetOptBpropFromCache() local
210 ArgsToAbs(prim, op_args, &abs_list); in GetOptBpropFromCache()
214 …ECacheQrtRes cache_res = GetOptBpfgFromCache(prim, abs_list, &level_2_graph_info, &level_1_graph_i… in GetOptBpropFromCache()
232 auto new_abs_list = AddOutToAbsList(out, abs_list); in GetOptBpropFromCache()
237 level_1_graph_info->graph_level_2_cache_[abs_list] = level_2_graph_info; in GetOptBpropFromCache()
246 abstract::AbstractBasePtrList abs_list; in GenSpecOptBprop() local
247 ArgsToAbs(prim, op_args, &abs_list); in GenSpecOptBprop()
249 auto iter = tuple_list_bprop_cache_.find(std::pair(prim, abs_list)); in GenSpecOptBprop()
260 auto new_abs_list = AddOutToAbsList(out, abs_list); in GenSpecOptBprop()
265 …tuple_list_bprop_cache_[std::pair(prim, abs_list)] = BasicClone(level_2_graph_info->opt_func_graph… in GenSpecOptBprop()
[all …]
|
| D | prim_bprop_optimizer.h | 152 …QrtRes GetOptBpfgFromCache(const PrimitivePtr &prim, const abstract::AbstractBasePtrList &abs_list,
157 …bs(const PrimitivePtr &prim, const ValuePtrList &op_args, abstract::AbstractBasePtrList *abs_list);
160 …actBasePtrList AddOutToAbsList(const ValuePtr &out, const abstract::AbstractBasePtrList &abs_list);
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | shape.cc | 41 AbstractBasePtrList abs_list; in ShapeInfer() local
42 (void)std::transform(in_shape.begin(), in_shape.end(), std::back_inserter(abs_list), in ShapeInfer()
46 auto abs = std::make_shared<abstract::AbstractTuple>(abs_list); in ShapeInfer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | update_state_formatter.cc | 115 AbstractBasePtrList abs_list; in Run() local
116 std::transform(mt_inputs.begin(), mt_inputs.end(), std::back_inserter(abs_list), in Run()
120 mt_node->set_abstract(std::make_shared<abstract::AbstractTuple>(abs_list)); in Run()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 749 size_t CheckAndConvertUtils::GetRemoveMonadAbsNum(const AbstractBasePtrList &abs_list) { in GetRemoveMonadAbsNum() argument
750 size_t remove_monad_count = abs_list.size(); in GetRemoveMonadAbsNum()
751 for (const auto &item : abs_list) { in GetRemoveMonadAbsNum()
758 if (abs_list[i]->isa<abstract::AbstractMonad>()) { in GetRemoveMonadAbsNum()
765 bool CheckAndConvertUtils::HasDynamicShapeInput(const AbstractBasePtrList &abs_list) { in HasDynamicShapeInput() argument
766 for (const auto &item : abs_list) { in HasDynamicShapeInput()
|
| D | check_convert_utils.h | 313 static size_t GetRemoveMonadAbsNum(const AbstractBasePtrList &abs_list);
318 static bool HasDynamicShapeInput(const AbstractBasePtrList &abs_list);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | item_tuple_or_list_eliminate.h | 217 AbstractBasePtrList abs_list; in operator() local
221 abs_list.emplace_back(abs->Broaden()); in operator()
223 make_tuple->set_abstract(std::make_shared<abstract::AbstractTuple>(abs_list)); in operator()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | clean.cc | 78 auto abs_list = dyn_cast<AbstractList>(t); in AdaptAbs() local
79 return std::make_shared<AbstractTuple>(abs_list->elements()); in AdaptAbs()
|