| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 60 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()
68 done_callback(std::move(done_callback)), in Participant()
108 DoneCallback done_callback; member
|
| D | nccl_manager.cc | 497 collective->participants[i]->done_callback(s); in RunCollective()
618 auto done_callback = [collective, p_idx, nccl_result]() { in LoopKernelLaunches() local
620 collective->participants[p_idx]->done_callback(Status::OK()); in LoopKernelLaunches()
624 collective->participants[p_idx]->done_callback(errors::Unknown( in LoopKernelLaunches()
637 p->event_mgr->ThenExecute(comm_stream, done_callback); in LoopKernelLaunches()
|
| /external/grpc-grpc/src/python/grpcio/grpc/beta/ |
| D | utilities.py | 69 for done_callback in done_callbacks:
71 done_callback, _DONE_CALLBACK_EXCEPTION_LOG_MESSAGE, self)
84 for done_callback in done_callbacks:
86 done_callback, _DONE_CALLBACK_EXCEPTION_LOG_MESSAGE, self)
|
| D | _client_adaptations.py | 149 def done_callback(): function
153 registered = self._call.add_callback(done_callback)
154 return None if registered else done_callback()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.h | 94 DoneCallback done_callback; // must be run outside mu_ member
101 Attempt(int elements_requested, DoneCallback done_callback, in Attempt()
105 done_callback(std::move(done_callback)), in Attempt()
|
| D | queue_base.h | 160 DoneCallback done_callback; // must be run outside mu_ member
170 Attempt(int32 elements_requested, DoneCallback done_callback, in Attempt()
174 done_callback(done_callback), in Attempt()
|
| D | conditional_accumulator_base.cc | 85 attempt->context, attempt->done_callback); in TryTakeGrad()
123 std::swap(callback, attempt.done_callback); in Cancel()
154 clean_up->emplace_back(std::move(cur_attempt->done_callback), in TryAttemptLocked()
|
| D | batch_kernels.cc | 242 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput() argument
270 batch_components->done_callback = std::move(done_callback); in RegisterInput()
289 AsyncOpKernel::DoneCallback done_callback; member
492 batch->mutable_task(i)->done_callback(); in ProcessFuncBatch()
564 batch->task(batch->num_tasks() - 1).done_callback; in ProcessBatch()
589 task.done_callback); in ProcessBatch()
600 task.done_callback); in ProcessBatch()
609 task.done_callback); in ProcessBatch()
620 batch->mutable_task(task_idx)->done_callback(); in ProcessBatch()
982 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute() local
[all …]
|
| D | queue_base.cc | 208 std::swap(callback, attempt.done_callback); in Cancel()
230 callbacks.emplace_back(std::move(attempt.done_callback)); in CloseAndCancel()
303 clean_up->emplace_back(std::move(cur_attempt->done_callback), in TryAttemptLocked()
|
| D | collective_nccl_reducer.cc | 140 auto done_callback = [&nccl_done, &nccl_status](const Status& s) { in Run() local
147 col_params_->default_rank, std::move(done_callback)); in Run()
|
| D | fifo_queue.cc | 179 attempt->done_callback = [callback, tuple]() { callback(tuple); }; in TryDequeue()
340 attempt->done_callback = [callback, tuple]() { in TryDequeueMany()
|
| D | priority_queue.cc | 222 attempt->done_callback = [callback, tuple]() { callback(tuple); }; in TryDequeue()
370 attempt->done_callback = [callback, tuple]() { in TryDequeueMany()
|
| D | random_shuffle_queue_op.cc | 258 attempt->done_callback = [callback, tuple]() { callback(tuple); }; in TryDequeue()
418 attempt->done_callback = [callback, tuple]() { in TryDequeueMany()
|
| D | padding_fifo_queue.cc | 220 attempt->done_callback = [callback, tuple]() { in TryDequeueMany()
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| D | _utilities.py | 100 for done_callback in done_callbacks:
102 done_callback, _DONE_CALLBACK_EXCEPTION_LOG_MESSAGE, self)
115 for done_callback in done_callbacks:
117 done_callback, _DONE_CALLBACK_EXCEPTION_LOG_MESSAGE, self)
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 36 done_callback) in RemoteExecuteNode()
41 done_callback_(std::move(done_callback)) { in RemoteExecuteNode()
|
| /external/perfetto/src/traced/probes/filesystem/ |
| D | file_scanner_unittest.cc | 43 std::function<void()> done_callback) in TestDelegate() argument
45 done_callback_(std::move(done_callback)) {} in TestDelegate()
|
| /external/python/cpython3/Lib/asyncio/ |
| D | windows_events.py | 604 def _wait_cancel(self, event, done_callback): argument
608 fut._done_callback = done_callback
|
| /external/libchrome/mojo/public/cpp/bindings/tests/ |
| D | associated_interface_unittest.cc | 1070 auto done_callback = base::Bind( in TEST_F() local
1084 ->Echo(123, base::Bind(done_callback, main_task_runner, in TEST_F()
|