Searched refs:next_operator_info (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.h | 53 bool FindReshapeNextNodeStraCosts(const CNodePtr &cnode, OperatorInfoPtr *next_operator_info, int64…
|
| D | node_info.cc | 361 bool FindReshapeNextNodeStraCosts(const CNodePtr &cnode, OperatorInfoPtr *next_operator_info, int64… in FindReshapeNextNodeStraCosts() argument
393 *next_operator_info = op_info; in FindReshapeNextNodeStraCosts()
400 …if (FindReshapeNextNodeStraCosts(use_apply, next_operator_info, in_index, is_next_reshape, ++curr_… in FindReshapeNextNodeStraCosts()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 881 OperatorInfoPtr next_operator_info; in ReshapeCostCompute() local
884 …bool find_next_node = FindReshapeNextNodeStraCosts(cnode, &next_operator_info, &in_index, &is_next… in ReshapeCostCompute()
894 next_stra_costs = next_operator_info->strategy_cost(); in ReshapeCostCompute()
895 reshape_info->set_next_operator_name(next_operator_info->name()); in ReshapeCostCompute()
|