Searched defs:curr_depth (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 305 size_t curr_depth) { in FindReshapePreNodeStraCosts()
362 bool *is_next_reshape, size_t 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()
2038 std::shared_ptr<TensorLayout> FindParameterNextLayout(const AnfNodePtr &node, size_t curr_depth) { in FindParameterNextLayout()
2228 CNodePtr HandleDependLoss(const CNodePtr &cnode, size_t curr_depth) { in HandleDependLoss()
2615 ParameterMap NodeParameterName(const CNodePtr &node, int64_t index, size_t curr_depth) { in NodeParameterName()
2797 static AnfNodePtr FindGrad(const CNodePtr &cnode, size_t curr_depth) { in FindGrad()
|
| /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()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | graph_costmodel.cc | 124 auto curr_depth = next_level.front().second; in BFS() local
|