Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.h227 void add_thread_pool_task(const std::weak_ptr<GraphTask>& graph_task);
H A Dengine.cpp1354 add_thread_pool_task(graph_task); in execute_with_graph_task()
1520 void Engine::add_thread_pool_task(const std::weak_ptr<GraphTask>& graph_task) { in add_thread_pool_task() function in torch::autograd::Engine