Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp127 #define current_graph_task (tls_current_graph_task.get()) macro
391 : last_graph_task_(std::move(current_graph_task)) { in GraphTaskGuard()
392 current_graph_task = std::move(graph_task); in GraphTaskGuard()
399 current_graph_task = std::move(last_graph_task_); in restore_current_graph_task()
406 return current_graph_task ? &current_graph_task->exec_info_ : nullptr; in get_current_graph_task_exec_info()
410 return current_graph_task ? &current_graph_task->nodes_in_graph_ : nullptr; in get_current_graph_task_nodes_in_graph()
414 return current_graph_task ? current_graph_task->id_ : -1; in get_current_graph_task_id()
418 return current_graph_task ? current_graph_task->keep_graph_ : true; in get_current_graph_task_keep_graph()
422 current_graph_task->exec_info_[fn].needed_ = true; in add_node_to_current_graph_task_exec_info()
427 std::shared_ptr<GraphTask> task = current_graph_task; in get_current_graph_task_execution_order()
[all …]