| /external/tensorflow/tensorflow/core/kernels/ |
| D | queue_op.cc | 53 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
|
| D | nccl_ops.cc | 97 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclAllReduceOpKernel
133 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclReduceSendKernel
164 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclReduceRecvKernel
203 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclBroadcastSendKernel
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclBroadcastRecvKernel
265 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon05def67a0111::NcclStubKernel
|
| D | barrier_ops.cc | 503 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
|
| D | collective_ops.cc | 73 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveOpV1Kernel
715 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveReduceV2OpKernel
766 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveGatherV2OpKernel
812 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveBcastSendV2OpKernel
861 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveBcastRecvV2OpKernel
975 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveInitializeCommunicatorOpKernel
1181 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveReduceV3OpKernel
1231 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon8b310e240111::CollectiveAllToAllV3OpKernel
|
| D | conditional_accumulator_base_op.h | 196 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function
237 void ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function
|
| D | function_ops.cc | 227 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::SymbolicGradientOp
286 void RemoteCallOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::RemoteCallOp
|
| D | stack.cc | 221 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
|
| D | batch_kernels.cc | 288 void BatchFunctionKernel::ComputeAsync(OpKernelContext* c, DoneCallback done) { in ComputeAsync() function in tensorflow::BatchFunctionKernel
560 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::BatchKernel
822 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::UnbatchKernel
1018 void ComputeAsync(OpKernelContext* c, DoneCallback done) final { in ComputeAsync() function in tensorflow::UnbatchGradKernel
|
| D | sparse_slice_op.cc | 174 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::SparseSliceGPUOp
|
| D | functional_ops.cc | 135 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anoneb8d038d0111::IfOp
251 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anoneb8d038d0111::CaseOp
364 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anoneb8d038d0111::WhileOp
740 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anoneb8d038d0111::ForOp
|
| D | sparse_split_op.cc | 173 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::SparseSplitGPUOp
|
| D | segment_reduction_ops_impl.h | 230 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | gpu_swapping_kernels.cc | 29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon857460de0111::CopyFromGpuToHostKernel
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon857460de0111::CopyFromHostToGpuKernel
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | testlib_ops.cc | 92 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::test::DelayOp
|
| D | renamed_device.h | 127 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/sleep/ |
| D | sleep_kernel.cc | 54 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::custom_op_examples::AsyncSleepOp
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 133 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
|
| D | self_adjoint_eig_v2_op_gpu.cc | 48 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::SelfAdjointEigV2OpGpu
|
| D | matrix_inverse_op.cc | 98 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::MatrixInverseOpGpu
|
| D | lu_op_gpu.cu.cc | 83 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::LuOpGpu
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device.h | 92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 51 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::experimental::__anonad335b870111::ToTFRecordOp
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_reshard_variables_op.cc | 57 void TPUReshardVariablesOpKernel::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::TPUReshardVariablesOpKernel
|
| D | host_compute_ops.cc | 75 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon80850d1c0111::RecvAtHostOp
|
| D | transfer_ops.cc | 44 void TpuTransferAsyncOpKernelBase::ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function in tensorflow::TpuTransferAsyncOpKernelBase
|