Searched refs:get_current_graph_task_id (Results 1 – 3 of 3) sorted by relevance
223 TORCH_API int get_current_graph_task_id();
413 int get_current_graph_task_id() { in get_current_graph_task_id() function
1244 return THPUtils_packInt64(torch::autograd::get_current_graph_task_id()); in THPModule_getCurrentGraphTaskId()