| /external/tensorflow/tensorflow/core/kernels/ |
| D | queue_op.h | 70 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final;
74 DoneCallback callback) = 0;
100 DoneCallback callback) override;
125 DoneCallback callback) override;
146 DoneCallback callback) override;
171 DoneCallback callback) override;
214 DoneCallback callback) override;
232 DoneCallback callback) override;
251 DoneCallback callback) override;
263 DoneCallback callback) override;
|
| 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;
156 bool TakeGradLockedHelper(OpKernelContext* ctx, DoneCallback callback)
|
| D | queue_op.cc | 53 void QueueOpKernel::ComputeAsync(OpKernelContext* ctx, DoneCallback callback) { in ComputeAsync()
89 DoneCallback callback) { in ComputeAsync()
130 DoneCallback callback) { in ComputeAsync()
167 DoneCallback callback) { in ComputeAsync()
210 DoneCallback callback) { in ComputeAsync()
283 DoneCallback callback) { in ComputeAsync()
334 DoneCallback callback) { in ComputeAsync()
348 DoneCallback callback) { in ComputeAsync()
359 DoneCallback callback) { in ComputeAsync()
|
| D | queue_base.h | 60 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 | nccl_ops.cc | 97 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
133 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
164 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
203 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
265 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
|
| D | barrier_ops.cc | 46 typedef std::function<void()> DoneCallback; typedef in tensorflow::barrier::Barrier
91 const DoneCallback& callback) { in TryInsertMany()
254 const DoneCallback& callback) { in Close()
407 const DoneCallback& callback) in CloseQueueLocked()
507 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
519 DoneCallback callback) = 0;
533 DoneCallback callback) override { in ComputeAsync()
582 DoneCallback callback) override { in ComputeAsync()
641 DoneCallback callback) override { in ComputeAsync()
660 DoneCallback callback) override { in ComputeAsync()
[all …]
|
| D | conditional_accumulator_base_op.h | 37 typedef std::function<void()> DoneCallback; typedef
197 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
210 DoneCallback callback) = 0;
217 DoneCallback callback) { in CheckSignature()
240 DoneCallback callback) override { in ComputeAsync()
|
| D | functional_ops.cc | 137 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
158 FHandle else_handle, DoneCallback done) in State()
187 DoneCallback captured_done(std::move(done_)); in Start()
199 DoneCallback done_;
253 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
285 std::vector<FHandle> branch_handles, DoneCallback done) in State()
317 DoneCallback captured_done(std::move(done_)); in Start()
328 DoneCallback done_;
365 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
504 FHandle body_handle, DoneCallback done) in State()
[all …]
|
| D | conditional_accumulator_base.cc | 66 DoneCallback callback) { in TryTakeGrad()
112 DoneCallback callback = nullptr; in Cancel()
191 DoneCallback callback) { in TakeGradLockedHelper()
|
| D | partitioned_function_ops.h | 41 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
55 DoneCallback done);
|
| D | stack.h | 44 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
63 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
|
| D | batch_kernels.cc | 246 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
480 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
560 Status Compute(OpKernelContext* context, AsyncOpKernel::DoneCallback done) { in Compute()
599 std::vector<AsyncOpKernel::DoneCallback> done_callbacks_to_call; in Compute()
651 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute()
704 AsyncOpKernel::DoneCallback done;
735 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
770 const AsyncOpKernel::DoneCallback& done) in OutputBatch()
804 const AsyncOpKernel::DoneCallback& done) { in Compute()
893 AsyncOpKernel::DoneCallback done;
[all …]
|
| D | fifo_queue.h | 43 DoneCallback callback) override;
45 DoneCallback callback) override;
|
| /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 | device.h | 57 typedef std::function<void(const Status&)> DoneCallback; typedef
93 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
107 virtual void Sync(const DoneCallback& done);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.h | 79 Rendezvous::DoneCallback done) override;
133 DoneCallback done) override;
147 void RecvLocalAsync(const ParsedKey& parsed, DoneCallback done);
152 DoneCallback done) = 0;
187 DoneCallback done;
189 DeferredCall(const ParsedKey& parsed, DoneCallback done);
217 void RecvLocalAsyncInternal(const ParsedKey& parsed, DoneCallback done);
|
| D | cluster_function_library_runtime.h | 44 FunctionLibraryRuntime::DoneCallback done) override;
49 FunctionLibraryRuntime::DoneCallback done) override;
54 FunctionLibraryRuntime::DoneCallback done) override;
57 FunctionLibraryRuntime::DoneCallback done) override;
|
| /external/tensorflow/tensorflow/core/util/ |
| D | incremental_barrier.cc | 28 explicit InternalIncrementalBarrier(IncrementalBarrier::DoneCallback callback) in InternalIncrementalBarrier()
35 IncrementalBarrier::DoneCallback done_callback = in operator ()()
51 IncrementalBarrier::DoneCallback done_callback_;
54 IncrementalBarrier::IncrementalBarrier(DoneCallback done_callback) in IncrementalBarrier()
|
| D | incremental_barrier.h | 61 typedef std::function<void()> DoneCallback; typedef
63 explicit IncrementalBarrier(DoneCallback callback);
|
| D | incremental_barrier_test.cc | 55 IncrementalBarrier::DoneCallback done_callback = in TEST()
68 IncrementalBarrier::DoneCallback done_callback = in TEST()
96 IncrementalBarrier::DoneCallback done_callback = in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 54 FunctionLibraryRuntime::DoneCallback done) override;
66 FunctionLibraryRuntime::DoneCallback done) override;
74 FunctionLibraryRuntime::DoneCallback done) override;
77 FunctionLibraryRuntime::DoneCallback done) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 186 FunctionLibraryRuntime::DoneCallback done) const;
189 FunctionLibraryRuntime::DoneCallback done) const;
194 FunctionLibraryRuntime::DoneCallback done) const;
312 FunctionLibraryRuntime::DoneCallback done,
320 FunctionLibraryRuntime::DoneCallback ApplyCleanUpToDoneCallback(
322 FunctionLibraryRuntime::DoneCallback done, const int64 step_id,
374 FunctionLibraryRuntime::DoneCallback done);
396 FunctionLibraryRuntime::DoneCallback done) const;
399 FunctionLibraryRuntime::DoneCallback done) const;
426 FunctionLibraryRuntime::DoneCallback done);
|
| D | rendezvous_mgr.h | 51 DoneCallback done) override;
80 DoneCallback done) override;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 49 DoneCallback done) override;
64 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init()
132 const Rendezvous::DoneCallback& done() const { return done_; } in done()
180 Rendezvous::DoneCallback done_;
235 DoneCallback done) { in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 51 typedef std::function<void(Status)> DoneCallback; typedef
70 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()
125 DoneCallback done_callback;
|