Searched refs:restore_current_graph_task (Results 1 – 2 of 2) sorted by relevance
211 void restore_current_graph_task();
395 restore_current_graph_task(); in ~GraphTaskGuard()398 void GraphTaskGuard::restore_current_graph_task() { in restore_current_graph_task() function in torch::autograd::GraphTaskGuard