Searched refs:cpu_ready_queue (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/autograd/ |
| D | engine.h | 180 const std::shared_ptr<ReadyQueue>& cpu_ready_queue);
215 std::shared_ptr<ReadyQueue> cpu_ready_queue,
218 std::shared_ptr<ReadyQueue> cpu_ready_queue,
|
| D | engine.cpp | 662 std::shared_ptr<ReadyQueue> cpu_ready_queue, in GraphTask() argument
670 cpu_ready_queue_(std::move(cpu_ready_queue)), in GraphTask()
993 const std::shared_ptr<ReadyQueue>& cpu_ready_queue) { in evaluate_function() argument
1113 auto queue = ready_queue(cpu_ready_queue, input_buffer.device()); in evaluate_function()
1128 auto queue = ready_queue(cpu_ready_queue, input_buffer.device()); in evaluate_function()
1437 std::shared_ptr<ReadyQueue> cpu_ready_queue, in ready_queue() argument
1443 TORCH_INTERNAL_ASSERT(cpu_ready_queue); in ready_queue()
1444 return cpu_ready_queue; in ready_queue()
1456 std::shared_ptr<ReadyQueue> cpu_ready_queue, in ready_queue_by_index() argument
1460 TORCH_INTERNAL_ASSERT(cpu_ready_queue); in ready_queue_by_index()
[all …]
|
| D | graph_task.h | 196 std::shared_ptr<ReadyQueue> cpu_ready_queue,
|
| /external/pytorch/torch/csrc/distributed/autograd/engine/ |
| D | dist_engine.cpp | 353 std::shared_ptr<ReadyQueue> cpu_ready_queue = std::make_shared<ReadyQueue>(); in execute_graph_task_until_ready_queue_empty() local
361 cpu_ready_queue->push(std::move(node_task), incrementOutstandingTasks); in execute_graph_task_until_ready_queue_empty()
364 while (!cpu_ready_queue->empty()) { in execute_graph_task_until_ready_queue_empty()
370 NodeTask task = cpu_ready_queue->pop(); in execute_graph_task_until_ready_queue_empty()
379 local_graph_task, task.fn_.get(), task.inputs_, cpu_ready_queue); in execute_graph_task_until_ready_queue_empty()
|