Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/distributed/autograd/engine/
Ddist_engine.cpp348 bool incrementOutstandingTasks) { in execute_graph_task_until_ready_queue_empty() argument
361 cpu_ready_queue->push(std::move(node_task), incrementOutstandingTasks); in execute_graph_task_until_ready_queue_empty()
407 bool incrementOutstandingTasks) { in runEngineAndAccumulateGradients() argument
413 at::launch([this, graphTask, graphRoot, incrementOutstandingTasks]() { in runEngineAndAccumulateGradients()
416 /*incrementOutstandingTasks*/ incrementOutstandingTasks); in runEngineAndAccumulateGradients()
Ddist_engine.h123 bool incrementOutstandingTasks = true);
/external/pytorch/torch/csrc/autograd/
Dengine.h116 void push(NodeTask item, bool incrementOutstandingTasks = true);
Dengine.cpp233 auto ReadyQueue::push(NodeTask item, bool incrementOutstandingTasks) -> void { in push() argument
237 if (incrementOutstandingTasks) { in push()