Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/autograd/
Dgraph_task.h34 std::unordered_set<Node*> nodes_in_graph_; member
Dengine.cpp410 return current_graph_task ? &current_graph_task->nodes_in_graph_ : nullptr; in get_current_graph_task_nodes_in_graph()
1173 const bool was_inserted = task.nodes_in_graph_.insert(next_ptr).second; in compute_dependencies()