Home
last modified time | relevance | path

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

12

/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.cc93 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclAllReduceOpKernel
130 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclReduceSendKernel
162 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclReduceRecvKernel
202 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclBroadcastSendKernel
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclBroadcastRecvKernel
266 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon49c74ebe0111::NcclStubKernel
Dcollective_ops.cc87 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon947389680111::CollectiveGatherOpKernel
185 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon947389680111::CollectiveReduceOpKernel
250 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon947389680111::CollectiveBcastSendOpKernel
319 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon947389680111::CollectiveBcastRecvOpKernel
Dbarrier_ops.cc507 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function in tensorflow::barrier::BarrierOpKernel
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
Dconditional_accumulator_base_op.h129 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function
215 void ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function
Drpc_op.cc61 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::RpcOp
Dfunction_ops.cc241 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::SymbolicGradientOp
303 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::RemoteCallOp
Dself_adjoint_eig_v2_op_gpu.cc48 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::SelfAdjointEigV2OpGpu
Dfunctional_ops.cc138 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonf4cf7ed10111::IfOp
225 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonf4cf7ed10111::CaseOp
332 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonf4cf7ed10111::WhileOp
510 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anonf4cf7ed10111::ForOp
Dcholesky_op.cc100 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::CholeskyOpGpu
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
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
Dsendrecv_ops.cc170 void RecvOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::RecvOp
Dcheck_numerics_op.cc126 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4ae96e6e0111::CheckNumericsOp
Dmatrix_inverse_op.cc98 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::MatrixInverseOpGpu
Dlu_op_gpu.cu.cc81 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::LuOpGpu
Dpartitioned_function_ops.cc80 void PartitionedCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::PartitionedCallOp
Dcrop_and_resize_op.cc121 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeOp
344 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeGradImageOp
526 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeGradBoxesOp
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgpu_swapping_kernels.cc29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon145ac5220111::CopyFromGpuToHostKernel
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon145ac5220111::CopyFromHostToGpuKernel
/external/tensorflow/tensorflow/core/common_runtime/
Dtestlib_ops.cc84 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::test::DelayOp
Ddevice.h93 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
Drenamed_device.h103 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfunction_ops.cc66 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon0de038b60111::SymbolicGradientOp
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dto_tf_record_op.cc46 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anon18e6484b0111::ToTFRecordOp
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc570 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anon9c98ff710411::ToSingleElementOp
650 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anon9c98ff710411::ReduceDatasetOp
794 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anon9c98ff710411::OneShotIteratorOp
947 void IteratorGetNextOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::data::IteratorGetNextOp
997 void IteratorGetNextAsOptionalOp::ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function in tensorflow::data::IteratorGetNextAsOptionalOp

12