| /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 | functional_ops.cc | 138 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
167 FHandle else_handle, DoneCallback done) in State()
195 DoneCallback captured_done(std::move(done_)); in Start()
207 DoneCallback done_;
225 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
257 std::vector<FHandle> branch_handles, DoneCallback done) in State()
288 DoneCallback captured_done(std::move(done_)); in Start()
299 DoneCallback done_;
332 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
358 FHandle body_handle, DoneCallback done) in State()
[all …]
|
| D | queue_base.h | 64 DoneCallback callback) override;
102 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
104 DoneCallback finished;
160 DoneCallback done_callback; // must be run outside mu_
170 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()
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
129 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync()
142 DoneCallback callback) = 0;
213 DoneCallback callback) = 0;
217 DoneCallback callback) override { in ComputeAsync()
|
| D | nccl_ops.cc | 93 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
130 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
162 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
202 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync()
266 void ComputeAsync(OpKernelContext* c, DoneCallback done) 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 | 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()
170 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync()
|
| 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 | 242 AsyncOpKernel::DoneCallback done_callback) { in RegisterInput()
289 AsyncOpKernel::DoneCallback done_callback;
563 AsyncOpKernel::DoneCallback last_task_callback = in ProcessBatch()
721 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
801 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
881 Status Compute(OpKernelContext* context, AsyncOpKernel::DoneCallback done) { in Compute()
930 std::vector<AsyncOpKernel::DoneCallback> done_callbacks_to_call; in Compute()
982 for (const AsyncOpKernel::DoneCallback& done_callback : in Compute()
1035 AsyncOpKernel::DoneCallback done;
1065 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync()
[all …]
|
| D | svd_op_gpu.cu.cc | 95 void RunSVD(OpKernelContext* context, DoneCallback done, int64 m, int64 n, in RunSVD()
217 void CheckResult(OpKernelContext* context, DoneCallback done, in CheckResult()
238 void PerformSVD_MgeqN(OpKernelContext* context, DoneCallback done, int64 m, in PerformSVD_MgeqN()
263 void PerformSVD_MlessN(OpKernelContext* context, DoneCallback done, int64 m, in PerformSVD_MlessN()
291 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync()
|
| 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;
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()
335 DoneCallback done) { in RecvLocalAsync()
355 DoneCallback done) { in RecvLocalAsyncInternal()
395 DoneCallback done) in DeferredCall()
|
| /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()
163 Rendezvous::DoneCallback done_;
218 DoneCallback done) { in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 60 typedef std::function<void(const Status&)> DoneCallback; typedef
94 AsyncOpKernel::DoneCallback done) { in ComputeAsync()
124 virtual void Sync(const DoneCallback& done);
|
| D | rendezvous_mgr.cc | 118 DoneCallback done) { in RecvAsync()
125 [this, parsed](DoneCallback done, in RecvAsync()
137 [send_args, recv_args, out, is_dead](DoneCallback done, in RecvAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 45 typedef std::function<void(Status)> DoneCallback; typedef
60 Tensor* output, int global_rank, DoneCallback done_callback) in Participant()
108 DoneCallback done_callback;
|
| /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/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync()
|