Searched refs:curr_depth (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.h | 51 size_t curr_depth);
54 bool *is_next_reshape, size_t curr_depth);
|
| D | node_info.cc | 305 size_t curr_depth) { in FindReshapePreNodeStraCosts() argument
306 if (curr_depth > MAX_RECURSIVE_DEPTH) { in FindReshapePreNodeStraCosts()
349 …!FindReshapePreNodeStraCosts(cnode->inputs()[index], pre_operator_info, out_index, ++curr_depth)) { in FindReshapePreNodeStraCosts()
362 bool *is_next_reshape, size_t curr_depth) { in FindReshapeNextNodeStraCosts() argument
363 if (curr_depth > MAX_RECURSIVE_DEPTH) { in FindReshapeNextNodeStraCosts()
400 …ReshapeNextNodeStraCosts(use_apply, next_operator_info, in_index, is_next_reshape, ++curr_depth)) { in FindReshapeNextNodeStraCosts()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 1789 …st AnfNodePtr &node, vector<std::string> *unique_ids, vector<size_t> *indexes, size_t curr_depth) { in FindPreNodes() argument
1790 if (curr_depth > MAX_RECURSIVE_DEPTH) { in FindPreNodes()
1821 if (FindPreNodes(cnode, unique_ids, indexes, ++curr_depth)) { in FindPreNodes()
2038 std::shared_ptr<TensorLayout> FindParameterNextLayout(const AnfNodePtr &node, size_t curr_depth) { in FindParameterNextLayout() argument
2039 if (curr_depth > MAX_RECURSIVE_DEPTH) { in FindParameterNextLayout()
2049 auto layout_param = FindParameterNextLayout(node_pair.first, ++curr_depth); in FindParameterNextLayout()
2228 CNodePtr HandleDependLoss(const CNodePtr &cnode, size_t curr_depth) { in HandleDependLoss() argument
2229 if (curr_depth > MAX_RECURSIVE_DEPTH) { in HandleDependLoss()
2239 return HandleDependLoss(depend_before, ++curr_depth); in HandleDependLoss()
2615 ParameterMap NodeParameterName(const CNodePtr &node, int64_t index, size_t curr_depth) { in NodeParameterName() argument
[all …]
|
| D | step_parallel.h | 136 ParameterMap NodeParameterName(const CNodePtr &node, int64_t index, size_t curr_depth);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | graph_costmodel.cc | 124 auto curr_depth = next_level.front().second; in BFS() local
126 MS_LOG(INFO) << "curr_depth: " << curr_depth; in BFS()
127 curr_op->SetSelectedStrategy(configured_stra, curr_depth); in BFS()
135 if ((curr_depth > 0) && (configured_ops.find(next_op) != configured_ops.end())) { in BFS()
153 (void)next_level.emplace(std::make_pair(next_op, next_op_stra), curr_depth + 1); in BFS()
162 if ((curr_depth > 0) && (configured_ops.find(prev_op) != configured_ops.end())) { in BFS()
166 MS_LOG(ERROR) << "curr_depth: " << curr_depth; in BFS()
181 (void)next_level.emplace(std::make_pair(prev_op, prev_op_stra), curr_depth + 1); in BFS()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 1712 void OperatorInfo::SetSelectedStrategy(const StrategyPtr &s_strategy, size_t curr_depth) { in SetSelectedStrategy() argument
1714 if ((selected_strategy_depth_ != -1) && (SizeToLong(curr_depth) > selected_strategy_depth_)) { in SetSelectedStrategy()
1721 selected_strategy_depth_ = SizeToLong(curr_depth); in SetSelectedStrategy()
|