Searched refs:get_current_graph_task_exec_info (Results 1 – 5 of 5) sorted by relevance
218 get_current_graph_task_exec_info();
423 const auto exec_info = get_current_graph_task_exec_info(); in task_should_compute_output()
405 get_current_graph_task_exec_info() { in get_current_graph_task_exec_info() function
110 const auto exec_info = get_current_graph_task_exec_info(); in apply_impl()
1186 const auto exec_info = torch::autograd::get_current_graph_task_exec_info(); in THPModule_willEngineExecuteNode()