Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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
H A Dnccl_ops.cc97 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclAllReduceOpKernel
133 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclReduceSendKernel
164 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclReduceRecvKernel
203 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclBroadcastSendKernel
232 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclBroadcastRecvKernel
265 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon4086b40b0111::NcclStubKernel
H A Dcollective_ops.cc73 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveOpV1Kernel
715 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveReduceV2OpKernel
766 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveGatherV2OpKernel
812 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveBcastSendV2OpKernel
861 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveBcastRecvV2OpKernel
975 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveInitializeCommunicatorOpKernel
1181 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveReduceV3OpKernel
1231 void ComputeAsync(OpKernelContext* c, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon3faef4750111::CollectiveAllToAllV3OpKernel
H A 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
H A Dbatch_kernels.cc288 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
H A Dconditional_accumulator_base_op.h196 void ComputeAsync(OpKernelContext* ctx, DoneCallback callback) final { in ComputeAsync() function
237 void ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function
H A Dfunctional_ops.cc135 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anona00ae9de0111::IfOp
251 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anona00ae9de0111::CaseOp
364 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anona00ae9de0111::WhileOp
740 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anona00ae9de0111::ForOp
H A Dfunction_ops.cc227 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
H A 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
H A Dsparse_slice_op.cc174 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::SparseSliceGPUOp
H A Dsparse_split_op.cc173 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::SparseSplitGPUOp
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgpu_swapping_kernels.cc29 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon542a216f0111::CopyFromGpuToHostKernel
62 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::__anon542a216f0111::CopyFromHostToGpuKernel
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A 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
H A Dself_adjoint_eig_v2_op_gpu.cc48 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::SelfAdjointEigV2OpGpu
H A Dmatrix_inverse_op.cc98 void ComputeAsync(OpKernelContext* context, DoneCallback done) final { in ComputeAsync() function in tensorflow::MatrixInverseOpGpu
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dcrop_and_resize_op.cc128 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeOp
364 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeGradImageOp
586 void ComputeAsync(OpKernelContext* context, DoneCallback done) override { in ComputeAsync() function in tensorflow::CropAndResizeGradBoxesOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.cc586 void HybridAsyncOpKernel::ComputeAsync(OpKernelContext* ctx, in ComputeAsync() function in tensorflow::data::HybridAsyncOpKernel
633 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anonfca7ab5e0711::ToSingleElementOp
739 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::__anonfca7ab5e0711::OneShotIteratorOp
/aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/
H A Drpc_ops.cc539 void RpcClientOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::rpc::RpcClientOp
758 void RpcCheckStatusOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::rpc::RpcCheckStatusOp
792 void RpcGetValueOp::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { in ComputeAsync() function in tensorflow::rpc::RpcGetValueOp
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_execute_op.cc431 void XRTExecuteOp::ComputeAsync(OpKernelContext* context, DoneCallback done) { in ComputeAsync() function in tensorflow::__anon174cd9a70111::XRTExecuteOp
518 void XRTExecuteChainedOp::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::__anon174cd9a70111::XRTExecuteChainedOp
H A Dtpu_execute_op.cc264 void XRTExecuteOp::ComputeAsync(OpKernelContext* context, DoneCallback done) { in ComputeAsync() function in tensorflow::__anon56c273a20111::XRTExecuteOp
386 void XRTExecuteChainedOp::ComputeAsync(OpKernelContext* context, in ComputeAsync() function in tensorflow::__anon56c273a20111::XRTExecuteChainedOp
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dtestlib_ops.cc92 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::test::DelayOp
H A Drenamed_device.h127 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/sleep/
H A Dsleep_kernel.cc54 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::custom_op_examples::AsyncSleepOp
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddevice.h92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dto_tf_record_op.cc51 void ComputeAsync(OpKernelContext* ctx, DoneCallback done) override { in ComputeAsync() function in tensorflow::data::experimental::__anon52a12e380111::ToTFRecordOp

123