Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc291 auto seed_node = independent_nodes.top(); in GetVisitNodesInfo() local
293 MS_EXCEPTION_IF_NULL(seed_node); in GetVisitNodesInfo()
294 auto node_target = GetCNodeTarget(seed_node); in GetVisitNodesInfo()
295 if (IsPrimitiveCNode(seed_node, prim::kPrimCast)) { in GetVisitNodesInfo()
296 auto output_edges_iter = dependency_info.output_edges_.find(seed_node); in GetVisitNodesInfo()
304 result.seed_cast_next_node_[cast_next_node] = seed_node; in GetVisitNodesInfo()
310 result.default_target_nodes_.push(seed_node); in GetVisitNodesInfo()
312 result.other_target_nodes_.push(seed_node); in GetVisitNodesInfo()