Searched refs:pre_operator_info (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 304 bool FindReshapePreNodeStraCosts(const AnfNodePtr &node, OperatorInfoPtr *pre_operator_info, int64_… in FindReshapePreNodeStraCosts() argument
324 *pre_operator_info = node_op_info; in FindReshapePreNodeStraCosts()
340 *pre_operator_info = pre_op_info; in FindReshapePreNodeStraCosts()
349 …if (!FindReshapePreNodeStraCosts(cnode->inputs()[index], pre_operator_info, out_index, ++curr_dept… in FindReshapePreNodeStraCosts()
|
| D | node_info.h | 50 bool FindReshapePreNodeStraCosts(const AnfNodePtr &node, OperatorInfoPtr *pre_operator_info, int64_…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 865 OperatorInfoPtr pre_operator_info; in ReshapeCostCompute() local
871 pre_operator_info = reshape_info1; in ReshapeCostCompute()
874 if (!FindReshapePreNodeStraCosts(pre_node, &pre_operator_info, &out_index, 0)) { in ReshapeCostCompute()
877 pre_stra_costs = pre_operator_info->strategy_cost(); in ReshapeCostCompute()
891 reshape_info->set_pre_operator_name(pre_operator_info->name()); in ReshapeCostCompute()
|