Home
last modified time | relevance | path

Searched defs:ComputeAsync (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/
Dqueue_op.cc53 void QueueOpKernel::ComputeAsync(OpKernelContext* ctx, DoneCallback callback) { in ComputeAsync() function in tensorflow::QueueOpKernel
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::__anon095167410111::NcclAllReduceOpKernel
133 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon095167410111::NcclReduceSendKernel
164 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon095167410111::NcclReduceRecvKernel
203 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon095167410111::NcclBroadcastSendKernel
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon095167410111::NcclBroadcastRecvKernel
265 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon095167410111::NcclStubKernel
Dbarrier_ops.cc503 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function in tensorflow::barrier::BarrierOpKernel
528 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::InsertManyOp
577 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::TakeManyOp
636 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierCloseOp
655 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierIncompleteSizeOp
674 void ComputeAsync(OpKernelContext* ctx, Barrier* barrier, in ComputeAsync() function in tensorflow::barrier::BarrierReadySizeOp
Dcollective_ops.cc60 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2d3cb82b0111::CollectiveOpV1Kernel
604 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2d3cb82b0111::CollectiveReduceV2OpKernel
655 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2d3cb82b0111::CollectiveGatherV2OpKernel
701 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2d3cb82b0111::CollectiveBcastSendV2OpKernel
750 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon2d3cb82b0111::CollectiveBcastRecvV2OpKernel
Dconditional_accumulator_base_op.h196 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function
237 void ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function
Dfunction_ops.cc213 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::SymbolicGradientOp
271 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::RemoteCallOp
Dstack.cc221 void StackPushOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::StackPushOp
290 void StackPopOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::StackPopOp
Dfunctional_ops.cc137 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8d98ad940111::IfOp
253 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8d98ad940111::CaseOp
365 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8d98ad940111::WhileOp
741 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8d98ad940111::ForOp
Dbatch_kernels.cc264 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::BatchFunctionKernel
553 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::BatchKernel
808 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::UnbatchKernel
994 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::UnbatchGradKernel
Dsegment_reduction_ops_impl.h231 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function
Dmutex_ops.cc141 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::MutexLockOp
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonea1f56a50111::CopyFromGpuToHostKernel
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonea1f56a50111::CopyFromHostToGpuKernel
/external/tensorflow/tensorflow/core/common_runtime/
Dtestlib_ops.cc92 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::test::DelayOp
Drenamed_device.h126 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
/external/tensorflow/tensorflow/core/kernels/linalg/
Dself_adjoint_eig_v2_op_gpu.cc48 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::SelfAdjointEigV2OpGpu
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
Dmatrix_inverse_op.cc98 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::MatrixInverseOpGpu
Dlu_op_gpu.cu.cc83 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::LuOpGpu
/external/tensorflow/tensorflow/core/framework/
Ddevice.h92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc51 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::experimental::__anona73d0a2e0111::ToTFRecordOp
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_reshard_variables_op.cc57 void TPUReshardVariablesOpKernel::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::TPUReshardVariablesOpKernel
Dhost_compute_ops.cc75 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon85ffe0630111::RecvAtHostOp
Dtransfer_ops.cc44 void TpuTransferAsyncOpKernelBase::ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function in tensorflow::TpuTransferAsyncOpKernelBase
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc264 void XRTExecuteOp::ComputeAsync(OpKernelContext* context, DoneCallback done) { in ComputeAsync() function in tensorflow::__anon1bdb8d580111::XRTExecuteOp
386 void XRTExecuteChainedOp::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::__anon1bdb8d580111::XRTExecuteChainedOp
Dxrt_execute_op.cc429 void XRTExecuteOp::ComputeAsync(OpKernelContext* context, DoneCallback done) { in ComputeAsync() function in tensorflow::__anondc65f35d0111::XRTExecuteOp
516 void XRTExecuteChainedOp::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::__anondc65f35d0111::XRTExecuteChainedOp

123