Searched defs:graph_task (Results 1 – 4 of 4) sorted by relevance
211 std::shared_ptr<GraphTask> graph_task = base_.lock(); in getReentrantDepth() local223 const std::shared_ptr<const GraphTask>& graph_task) in CheckpointValidGuard()238 std::shared_ptr<GraphTask> graph_task = item.base_.lock(); in push() local382 std::shared_ptr<GraphTask> graph_task = nullptr; in thread_init() local390 GraphTaskGuard::GraphTaskGuard(std::shared_ptr<GraphTask> graph_task) in GraphTaskGuard()519 auto Engine::thread_main(const std::shared_ptr<GraphTask>& graph_task) -> void { in thread_main()632 std::shared_ptr<GraphTask> graph_task = task.lock(); in reentrant_thread_init() local648 const std::shared_ptr<GraphTask>& graph_task, in thread_on_exception()942 std::shared_ptr<GraphTask>& graph_task, in call_function()990 std::shared_ptr<GraphTask>& graph_task, in evaluate_function()[all …]
103 const std::shared_ptr<GraphTask>& graph_task, in thread_on_exception()146 const std::shared_ptr<GraphTask>& graph_task, in execute_with_graph_task()
469 GraphTask& graph_task, in _compiled_autograd_impl()708 GraphTask& graph_task, in compiled_autograd()
354 auto graph_task = node_task.base_.lock(); in execute_graph_task_until_ready_queue_empty() local