Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dcudnn_inplace_fusion.cc297 std::map<AnfNodePtr, int> topo_index; in TopoIndex() local
299 topo_index.insert(make_pair(node_list[i], i)); in TopoIndex()
301 return topo_index; in TopoIndex()
308 auto topo_index = TopoIndex(node_list); in Run() local
320 …std::sort(inplace_node.begin(), inplace_node.end(), [&topo_index](const AnfNodeIndex &n1, const An… in Run()
321 auto iter1 = topo_index.find(n1.node); in Run()
322 auto iter2 = topo_index.find(n2.node); in Run()
323 if (iter1 == topo_index.end() || iter2 == topo_index.end()) { in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.h40 const size_t topo_index() const { return topo_index_; } in topo_index() function
Dgraph_partitioner.cc249 size_t index1 = this->node_info_[node1].topo_index(); in CollectSegments()
250 size_t index2 = this->node_info_[node2].topo_index(); in CollectSegments()