Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dgraph_task.h211 void restore_current_graph_task();
H A Dengine.cpp395 restore_current_graph_task(); in ~GraphTaskGuard()
398 void GraphTaskGuard::restore_current_graph_task() { in restore_current_graph_task() function in torch::autograd::GraphTaskGuard