Home
last modified time | relevance | path

Searched refs:last_forward_node_ids (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc861 vector<std::string> last_forward_node_ids; in InsertVirtualOutput() local
863 FindLastNodesUniqueId(root, &last_forward_node_ids, &last_indexs); in InsertVirtualOutput()
864 MS_LOG(INFO) << "there are " << last_forward_node_ids.size() << " output nodes in eval/predict"; in InsertVirtualOutput()
871 …auto last_node_iter = std::find(last_forward_node_ids.begin(), last_forward_node_ids.end(), cnode-… in InsertVirtualOutput()
872 if (last_node_iter == last_forward_node_ids.end()) { in InsertVirtualOutput()
875 …for (size_t last_node_index = 0; last_node_index < last_forward_node_ids.size(); ++last_node_index… in InsertVirtualOutput()
876 if (last_forward_node_ids[last_node_index] != cnode->UniqueId()) { in InsertVirtualOutput()