| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.h | 57 typedef AsyncOpKernel::DoneCallback DoneCallback; typedef
61 DoneCallback callback);
94 DoneCallback done_callback; // must be run outside mu_
101 Attempt(int elements_requested, DoneCallback done_callback, in Attempt()
116 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
118 DoneCallback finished;
155 bool TakeGradLockedHelper(OpKernelContext* ctx, DoneCallback callback)
|
| D | queue_ops.cc | 34 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
51 DoneCallback callback) = 0;
84 DoneCallback callback) override { in ComputeAsync()
136 DoneCallback callback) override { in ComputeAsync()
186 DoneCallback callback) override { in ComputeAsync()
239 DoneCallback callback) override { in ComputeAsync()
325 DoneCallback callback) override { in ComputeAsync()
389 DoneCallback callback) override { in ComputeAsync()
414 DoneCallback callback) override { in ComputeAsync()
435 DoneCallback callback) override { in ComputeAsync()
|
| D | queue_base.h | 63 DoneCallback callback) override;
98 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
100 DoneCallback finished;
156 DoneCallback done_callback; // must be run outside mu_
166 Attempt(int32 elements_requested, DoneCallback done_callback, in Attempt()
|
| D | barrier_ops.cc | 46 typedef std::function<void()> DoneCallback; typedef in tensorflow::barrier::Barrier
91 const DoneCallback& callback) { in TryInsertMany()
255 const DoneCallback& callback) { in Close()
408 const DoneCallback& callback) in CloseQueueLocked()
508 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
520 DoneCallback callback) = 0;
534 DoneCallback callback) override { in ComputeAsync()
583 DoneCallback callback) override { in ComputeAsync()
643 DoneCallback callback) override { in ComputeAsync()
662 DoneCallback callback) override { in ComputeAsync()
[all …]
|
| D | conditional_accumulator_base_op.h | 37 typedef std::function<void()> DoneCallback; typedef
126 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
139 DoneCallback callback) = 0;
210 DoneCallback callback) = 0;
214 DoneCallback callback) override { in ComputeAsync()
|
| D | functional_ops.cc | 125 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
138 DoneCallback done) in State()
174 const DoneCallback done_;
195 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
238 FHandle body_handle, DoneCallback done) in State()
260 const DoneCallback done_;
|
| D | batch_kernels.cc | 238 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput()
276 AsyncOpKernel::DoneCallback done_callback;
327 AsyncOpKernel::DoneCallback last_task_callback = in ProcessBatch()
524 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
606 Status Compute(OpKernelContext* context, AsyncOpKernel::DoneCallback done) { in Compute()
655 std::vector<AsyncOpKernel::DoneCallback> done_callbacks_to_call; in Compute()
707 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute()
760 AsyncOpKernel::DoneCallback done;
790 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
827 const AsyncOpKernel::DoneCallback& done) in OutputBatch()
[all …]
|
| D | conditional_accumulator_base.cc | 61 DoneCallback callback) { in TryTakeGrad()
107 DoneCallback callback = nullptr; in Cancel()
186 DoneCallback callback) { in TakeGradLockedHelper()
|
| D | sendrecv_ops.cc | 146 Rendezvous::DoneCallback make_recv_callback(OpKernelContext* ctx, in make_recv_callback()
147 AsyncOpKernel::DoneCallback done) { in make_recv_callback()
150 [ctx](AsyncOpKernel::DoneCallback done, in make_recv_callback()
171 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()
|
| D | fifo_queue.h | 42 DoneCallback callback) override;
44 DoneCallback callback) override;
|
| D | priority_queue.h | 60 DoneCallback callback) override;
62 DoneCallback callback) override;
|
| D | queue_base.cc | 190 DoneCallback callback = nullptr; in Cancel()
221 std::vector<DoneCallback> callbacks; in CloseAndCancel()
234 for (const DoneCallback& callback : callbacks) { in CloseAndCancel()
241 DoneCallback callback) { in Close()
|
| D | critical_section.cc | 102 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
122 AsyncOpKernel::DoneCallback done_;
131 explicit Closure(AsyncOpKernel::DoneCallback done, OpKernelContext* ctx, in Closure()
|
| D | svd_op_gpu.cu.cc | 95 void RunSVD(OpKernelContext* context, DoneCallback done, int64 m, int64 n, in RunSVD()
171 void CheckResult(OpKernelContext* context, DoneCallback done, in CheckResult()
192 void PerformSVD_MgeqN(OpKernelContext* context, DoneCallback done, int64 m, in PerformSVD_MgeqN()
261 void PerformSVD_MlessN(OpKernelContext* context, DoneCallback done, int64 m, in PerformSVD_MlessN()
335 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync()
|
| D | dynamic_partition_op_gpu.cu.cc | 198 DoneCallback done) { in AllocateTempSpace()
212 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
227 void ComputeAsync(OpKernelContext* c, DoneCallback done) { in ComputeAsync()
325 Tensor* indices_out, DoneCallback done) { in RadixSort()
359 DoneCallback done) { in CountAndSortParts()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | queue_interface.h | 34 typedef AsyncOpKernel::DoneCallback DoneCallback; typedef
44 DoneCallback callback) = 0;
49 DoneCallback callback) = 0;
78 DoneCallback callback) = 0;
|
| D | rendezvous.h | 106 DoneCallback; typedef
109 DoneCallback done) = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.h | 79 Rendezvous::DoneCallback done) override;
136 DoneCallback done) override;
150 void RecvLocalAsync(const ParsedKey& parsed, DoneCallback done);
155 DoneCallback done) = 0;
189 DoneCallback done;
191 DeferredCall(const ParsedKey& parsed, DoneCallback done);
217 void RecvLocalAsyncInternal(const ParsedKey& parsed, DoneCallback done);
|
| D | base_rendezvous_mgr.cc | 70 Rendezvous::DoneCallback done) { in RecvLocalAsync()
73 Rendezvous::DoneCallback done_cb = std::bind( in RecvLocalAsync()
74 [rendez](Rendezvous::DoneCallback done, in RecvLocalAsync()
294 DoneCallback done) { in RecvAsync()
332 DoneCallback done) { in RecvLocalAsync()
352 DoneCallback done) { in RecvLocalAsyncInternal()
392 DoneCallback done) in DeferredCall()
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| D | nccl_manager.h | 38 typedef std::function<void(Status)> DoneCallback; typedef
63 const DoneCallback& done_callback);
71 const Tensor* in_t, DoneCallback done_callback);
76 Tensor* out_t, DoneCallback done_callback);
85 const Tensor* in_t, DoneCallback done_callback);
92 DoneCallback done_callback);
|
| D | nccl_ops.cc | 93 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
122 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
147 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
179 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
204 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
233 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
|
| D | nccl_manager.cc | 98 NcclManager::DoneCallback done_callback) in Participant()
130 NcclManager::DoneCallback done_callback;
305 const DoneCallback& done_callback) { in AddToAllReduce()
317 const Tensor* in_t, DoneCallback done_callback) { in AddBroadcastSend()
330 Tensor* out_t, DoneCallback done_callback) { in AddBroadcastRecv()
344 DoneCallback done_callback) { in AddReduceSend()
358 DoneCallback done_callback) { in AddReduceRecv()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 48 DoneCallback done) override;
63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init()
122 const Rendezvous::DoneCallback& done() const { return done_; } in done()
154 Rendezvous::DoneCallback done_;
209 DoneCallback done) { in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 38 DoneCallback done) override;
49 DoneCallback done) { in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.h | 107 typedef std::function<void(const Status&)> DoneCallback; typedef
108 virtual void RunAsync(const Args& args, DoneCallback done) = 0;
|