Home
last modified time | relevance | path

Searched refs:ComputeAsync (Results 1 – 25 of 55) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dqueue_op.h70 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final;
73 virtual void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
99 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
124 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
145 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
170 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
213 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
231 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
250 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
262 void ComputeAsync(OpKernelContext* ctx, QueueInterface* queue,
Dqueue_op.cc53 void QueueOpKernel::ComputeAsync(OpKernelContext* ctx, DoneCallback callback) { in ComputeAsync() function in tensorflow::QueueOpKernel
62 ComputeAsync(ctx, queue, [callback, queue]() { in ComputeAsync()
88 void EnqueueOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::EnqueueOp
129 void EnqueueManyOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::EnqueueManyOp
166 void DequeueOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::DequeueOp
209 void DequeueManyOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::DequeueManyOp
282 void DequeueUpToOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::DequeueUpToOp
333 void QueueCloseOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::QueueCloseOp
347 void QueueSizeOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::QueueSizeOp
358 void QueueIsClosedOp::ComputeAsync(OpKernelContext* ctx, QueueInterface* queue, in ComputeAsync() function in tensorflow::QueueIsClosedOp
Dnccl_ops.cc97 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclAllReduceOpKernel
133 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclReduceSendKernel
164 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclReduceRecvKernel
203 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclBroadcastSendKernel
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclBroadcastRecvKernel
265 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon6c22edc00111::NcclStubKernel
Dconditional_accumulator_base_op.h198 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function
202 ComputeAsync(ctx, accumulator, [callback, accumulator]() { in ComputeAsync()
209 virtual void ComputeAsync(OpKernelContext* ctx,
239 void ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function
Dbarrier_ops.cc507 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function in tensorflow::barrier::BarrierOpKernel
511 ComputeAsync(ctx, barrier, [callback, barrier]() { in ComputeAsync()
518 virtual void ComputeAsync(OpKernelContext* ctx, Barrier* barrier,
532 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::InsertManyOp
581 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::TakeManyOp
640 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierCloseOp
659 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierIncompleteSizeOp
678 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierReadySizeOp
Dstack.h44 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
63 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
Dcollective_ops.cc55 ComputeAsync(c, done); in CanProceedWithCompute()
93 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anond4bee1ea0111::CollectiveGatherOpKernel
217 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anond4bee1ea0111::CollectiveReduceOpKernel
289 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anond4bee1ea0111::CollectiveBcastSendOpKernel
365 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anond4bee1ea0111::CollectiveBcastRecvOpKernel
Dsendrecv_ops.h40 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
Dpartitioned_function_ops.h41 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
Dfunction_ops.h65 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
Dfunctional_ops.cc124 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon351ad7530111::IfOp
217 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon351ad7530111::CaseOp
330 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon351ad7530111::WhileOp
522 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon351ad7530111::ForOp
Dstack.cc220 void StackPushOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::StackPushOp
289 void StackPopOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::StackPopOp
Drpc_op.cc60 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::RpcOp
Ddeterminant_op.cc133 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::DeterminantOpGpu
273 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::LogDeterminantOpGpu
Dself_adjoint_eig_v2_op_gpu.cc48 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::SelfAdjointEigV2OpGpu
Dfunction_ops.cc242 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::SymbolicGradientOp
310 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::RemoteCallOp
Dcholesky_op.cc100 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::CholeskyOpGpu
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2b740a240111::CopyFromGpuToHostKernel
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2b740a240111::CopyFromHostToGpuKernel
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice.h90 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
92 op_kernel->ComputeAsync(context, std::move(done)); in ComputeAsync()
Drenamed_device.h136 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
138 underlying_device_->ComputeAsync(op_kernel, context, std::move(done)); in ComputeAsync()
Dtestlib_ops.cc92 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::test::DelayOp
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc234 void ComputeAsync(OpKernelContext* context, DoneCallback done) override;
243 void XRTExecuteOp::ComputeAsync(OpKernelContext* context, DoneCallback done) { in ComputeAsync() function in tensorflow::__anonbcd96c1c0111::XRTExecuteOp
319 void ComputeAsync(OpKernelContext* context, DoneCallback done) override;
328 void XRTExecuteChainedOp::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::__anonbcd96c1c0111::XRTExecuteChainedOp
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.h51 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override;
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc48 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::experimental::__anon13282d8d0111::ToTFRecordOp
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h142 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,

123