| /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 | functional_ops.cc | 124 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
153 FHandle else_handle, DoneCallback done) in State()
187 DoneCallback captured_done(std::move(done_)); in Start()
199 DoneCallback done_;
217 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
249 std::vector<FHandle> branch_handles, DoneCallback done) in State()
286 DoneCallback captured_done(std::move(done_)); in Start()
297 DoneCallback done_;
330 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
356 FHandle body_handle, DoneCallback done) in State()
[all …]
|
| 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
198 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
211 DoneCallback callback) = 0;
218 DoneCallback callback) { in CheckSignature()
241 DoneCallback callback) override { in ComputeAsync()
|
| 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 | fifo_queue.h | 43 DoneCallback callback) override;
45 DoneCallback callback) override;
|
| D | priority_queue.h | 60 DoneCallback callback) override;
62 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;
|
| /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;
192 DoneCallback done;
194 DeferredCall(const ParsedKey& parsed, DoneCallback done);
224 void RecvLocalAsyncInternal(const ParsedKey& parsed, DoneCallback done);
|
| D | cluster_function_library_runtime.h | 42 FunctionLibraryRuntime::DoneCallback done) override;
47 FunctionLibraryRuntime::DoneCallback done) override;
50 FunctionLibraryRuntime::DoneCallback done) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 180 FunctionLibraryRuntime::DoneCallback done) const;
183 FunctionLibraryRuntime::DoneCallback done) const;
188 FunctionLibraryRuntime::DoneCallback done) const;
294 FunctionLibraryRuntime::DoneCallback done) const;
305 FunctionLibraryRuntime::DoneCallback done,
310 FunctionLibraryRuntime::DoneCallback ApplyCleanUpToDoneCallback(
312 FunctionLibraryRuntime::DoneCallback done) const;
364 FunctionLibraryRuntime::DoneCallback done);
385 FunctionLibraryRuntime::DoneCallback done) const;
388 FunctionLibraryRuntime::DoneCallback done) const;
[all …]
|
| D | device.h | 57 typedef std::function<void(const Status&)> DoneCallback; typedef
91 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
115 virtual void Sync(const DoneCallback& done);
|
| D | rendezvous_mgr.h | 51 DoneCallback done) override;
80 DoneCallback done) override;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.h | 49 FunctionLibraryRuntime::DoneCallback done) override;
54 FunctionLibraryRuntime::DoneCallback done) override;
59 FunctionLibraryRuntime::DoneCallback done) override;
62 FunctionLibraryRuntime::DoneCallback done) override;
|
| D | cluster_function_library_runtime.cc | 37 FunctionLibraryRuntime::DoneCallback done) { in Instantiate()
108 std::vector<Tensor>* rets, FunctionLibraryRuntime::DoneCallback done) { in Run()
116 FunctionLibraryRuntime::DoneCallback done) { in Run()
180 FunctionLibraryRuntime::DoneCallback done) { in CleanUp()
|
| /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()
131 const Rendezvous::DoneCallback& done() const { return done_; } in done()
158 Rendezvous::DoneCallback done_;
213 DoneCallback done) { in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | process_function_library_runtime.h | 57 FunctionLibraryRuntime::DoneCallback done) const override;
64 FunctionLibraryRuntime::DoneCallback done) const override;
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 68 bool try_rpc, AsyncOpKernel::DoneCallback done,
87 const AsyncOpKernel::DoneCallback done_;
100 AsyncOpKernel::DoneCallback done, in CallContainer()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 52 typedef std::function<void(Status)> DoneCallback; typedef
71 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()
136 DoneCallback done_callback;
|