Home
last modified time | relevance | path

Searched refs:NodeTask (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/autograd/
Dengine.h47 struct NodeTask { struct
60 NodeTask( argument
87 bool operator()(NodeTask const& t1, NodeTask const& t2) { in operator()
108 std::priority_queue<NodeTask, std::vector<NodeTask>, CompareNodeTaskTime>
116 void push(NodeTask item, bool incrementOutstandingTasks = true);
118 NodeTask pop();
Dengine.cpp210 int NodeTask::getReentrantDepth() const { in getReentrantDepth()
233 auto ReadyQueue::push(NodeTask item, bool incrementOutstandingTasks) -> void { in push()
250 heap_.push(NodeTask({}, nullptr, InputBuffer(0), true)); in pushShutdownTask()
261 auto ReadyQueue::pop() -> NodeTask { in pop()
266 auto task = std::move(const_cast<NodeTask&>(heap_.top())); in pop()
537 NodeTask task = local_ready_queue->pop(); in thread_main()
608 ->push(NodeTask(local_graph_task, nullptr, InputBuffer(0))); in thread_main()
1115 NodeTask(graph_task, next.function, std::move(input_buffer))); in evaluate_function()
1130 NodeTask(graph_task, next.function, std::move(input_buffer))); in evaluate_function()
1329 NodeTask(graph_task, std::move(graph_root), std::move(input_buffer))); in execute_with_graph_task()
[all …]
/external/pytorch/torch/csrc/distributed/autograd/engine/
Ddist_engine.cpp25 using torch::autograd::NodeTask;
82 NodeTask task = ready_queue->pop(); in globalCpuThread()
106 /*node_task*/ NodeTask(graphTask, graphRoot, std::move(inputs)), in globalCpuThread()
347 NodeTask&& node_task, in execute_graph_task_until_ready_queue_empty()
370 NodeTask task = cpu_ready_queue->pop(); in execute_graph_task_until_ready_queue_empty()
415 /*node_task*/ NodeTask(graphTask, graphRoot, InputBuffer(0)), in runEngineAndAccumulateGradients()
551 /*node_task*/ NodeTask(graphTask, sendFunction, InputBuffer(0)), in executeSendFunctionAsync()
Ddist_engine.h122 torch::autograd::NodeTask&& node_task,