Home
last modified time | relevance | path

Searched refs:get_current_graph_task_exec_info (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dgraph_task.h218 get_current_graph_task_exec_info();
H A Dfunction.h423 const auto exec_info = get_current_graph_task_exec_info(); in task_should_compute_output()
H A Dengine.cpp405 get_current_graph_task_exec_info() { in get_current_graph_task_exec_info() function
/aosp_15_r20/external/pytorch/torch/csrc/autograd/functions/
H A Dtensor.cpp110 const auto exec_info = get_current_graph_task_exec_info(); in apply_impl()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1186 const auto exec_info = torch::autograd::get_current_graph_task_exec_info(); in THPModule_willEngineExecuteNode()