Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dgraph_task.h223 TORCH_API int get_current_graph_task_id();
H A Dengine.cpp413 int get_current_graph_task_id() { in get_current_graph_task_id() function
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1244 return THPUtils_packInt64(torch::autograd::get_current_graph_task_id()); in THPModule_getCurrentGraphTaskId()