| /external/tensorflow/tensorflow/core/kernels/ |
| conditional_accumulator_base.h | 94 DoneCallback done_callback; // must be run outside mu_ member in struct:tensorflow::ConditionalAccumulatorBase::Attempt
101 Attempt(int elements_requested, DoneCallback done_callback,
105 done_callback(std::move(done_callback)),
|
| queue_base.h | 160 DoneCallback done_callback; // must be run outside mu_ member in struct:tensorflow::QueueBase::Attempt
170 Attempt(int32 elements_requested, DoneCallback done_callback,
174 done_callback(done_callback),
|
| conditional_accumulator_base.cc | 85 attempt->context, attempt->done_callback);
123 std::swap(callback, attempt.done_callback);
154 clean_up->emplace_back(std::move(cur_attempt->done_callback),
|
| batch_kernels.cc | 242 AsyncOpKernel::DoneCallback done_callback) {
270 batch_components->done_callback = std::move(done_callback);
289 AsyncOpKernel::DoneCallback done_callback; member in struct:tensorflow::BatchResource::BatchTask
492 batch->mutable_task(i)->done_callback();
564 batch->task(batch->num_tasks() - 1).done_callback;
589 task.done_callback);
600 task.done_callback);
609 task.done_callback);
620 batch->mutable_task(task_idx)->done_callback();
[all...] |
| queue_base.cc | 208 std::swap(callback, attempt.done_callback);
230 callbacks.emplace_back(std::move(attempt.done_callback));
303 clean_up->emplace_back(std::move(cur_attempt->done_callback),
|
| collective_nccl_reducer.cc | 140 auto done_callback = [&nccl_done, &nccl_status](const Status& s) { local
147 col_params_->default_rank, std::move(done_callback));
|
| fifo_queue.cc | 179 attempt->done_callback = [callback, tuple]() { callback(tuple); };
340 attempt->done_callback = [callback, tuple]() {
|
| priority_queue.cc | 222 attempt->done_callback = [callback, tuple]() { callback(tuple); };
370 attempt->done_callback = [callback, tuple]() {
|
| random_shuffle_queue_op.cc | 258 attempt->done_callback = [callback, tuple]() { callback(tuple); };
418 attempt->done_callback = [callback, tuple]() {
|
| padding_fifo_queue.cc | 220 attempt->done_callback = [callback, tuple]() {
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| remote_execute_node.h | 36 done_callback)
41 done_callback_(std::move(done_callback)) {
|
| /external/grpc-grpc/src/python/grpcio/grpc/ |
| _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/grpc-grpc/src/python/grpcio/grpc/beta/ |
| 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)
|
| _client_adaptations.py | 149 def done_callback(): function in function:_Rendezvous.add_abortion_callback
153 registered = self._call.add_callback(done_callback)
154 return None if registered else done_callback()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| nccl_manager.h | 60 Tensor* output, int global_rank, DoneCallback done_callback)
68 done_callback(std::move(done_callback)),
81 // `done_callback` is not called until the next kernel launched on `stream`
83 // `done_callback` is called.
87 // Owned by the caller, who must keep it live until `done_callback` is
93 // Owned by the caller, who must keep it live until `done_callback` is
97 // Owned by the caller, who must keep it live until `done_callback` is
108 DoneCallback done_callback; member in struct:tensorflow::NcclManager::Participant
|
| nccl_manager.cc | 497 collective->participants[i]->done_callback(s);
617 // Run the done_callback when the nccl kernel finishes running.
618 auto done_callback = [collective, p_idx, nccl_result]() { local
620 collective->participants[p_idx]->done_callback(Status::OK());
624 collective->participants[p_idx]->done_callback(errors::Unknown(
637 p->event_mgr->ThenExecute(comm_stream, done_callback);
|
| /external/perfetto/src/traced/probes/filesystem/ |
| file_scanner_unittest.cc | 43 std::function<void()> done_callback)
45 done_callback_(std::move(done_callback)) {}
|
| /external/python/cpython3/Lib/asyncio/ |
| windows_events.py | 604 def _wait_cancel(self, event, done_callback):
608 fut._done_callback = done_callback
|
| /external/libchrome/mojo/public/cpp/bindings/tests/ |
| associated_interface_unittest.cc | [all...] |
