Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc286 VisitNodesInfo GetVisitNodesInfo(const GraphNodesDependencyInfo &dependency_info, const std::string… in GetVisitNodesInfo() argument
289 auto independent_nodes = dependency_info.independent_nodes_; in GetVisitNodesInfo()
296 auto output_edges_iter = dependency_info.output_edges_.find(seed_node); in GetVisitNodesInfo()
297 …if (output_edges_iter != dependency_info.output_edges_.end() && output_edges_iter->second.size() =… in GetVisitNodesInfo()
299 auto input_num_iter = dependency_info.input_num_.find(cast_next_node); in GetVisitNodesInfo()
300 if (input_num_iter == dependency_info.input_num_.end()) { in GetVisitNodesInfo()
343 …itNodeEdges(const std::vector<AnfNodePtr> &output_edges, GraphNodesDependencyInfo *dependency_info, in ParallelSortVisitNodeEdges() argument
345 MS_EXCEPTION_IF_NULL(dependency_info); in ParallelSortVisitNodeEdges()
349 auto input_num_iter = dependency_info->input_num_.find(dst_node); in ParallelSortVisitNodeEdges()
350 if (input_num_iter == dependency_info->input_num_.end()) { in ParallelSortVisitNodeEdges()
[all …]