Searched refs:GraphTaskGuard (Results 1 – 3 of 3) sorted by relevance
206 class GraphTaskGuard {208 explicit GraphTaskGuard(std::shared_ptr<GraphTask> graph_task);209 ~GraphTaskGuard();
390 GraphTaskGuard::GraphTaskGuard(std::shared_ptr<GraphTask> graph_task) in GraphTaskGuard() function in torch::autograd::GraphTaskGuard394 GraphTaskGuard::~GraphTaskGuard() { in ~GraphTaskGuard()398 void GraphTaskGuard::restore_current_graph_task() { in restore_current_graph_task()567 GraphTaskGuard guard(local_graph_task); in thread_main()716 GraphTaskGuard guard(shared_from_this()); in exec_post_processing()
22 using torch::autograd::GraphTaskGuard;377 GraphTaskGuard guard(local_graph_task); in execute_graph_task_until_ready_queue_empty()