Searched refs:TpuTransferAsyncOpKernelBase (Results 1 – 2 of 2) sorted by relevance
44 class TpuTransferAsyncOpKernelBase : public AsyncOpKernel {46 explicit TpuTransferAsyncOpKernelBase(67 TpuTransferAsyncOpKernelBase(const TpuTransferAsyncOpKernelBase&) = delete;68 TpuTransferAsyncOpKernelBase& operator=(const TpuTransferAsyncOpKernelBase&) =72 class TpuTransferAsyncOpKernel : public TpuTransferAsyncOpKernelBase {89 : public TpuTransferAsyncOpKernelBase {
32 TpuTransferAsyncOpKernelBase::TpuTransferAsyncOpKernelBase( in TpuTransferAsyncOpKernelBase() function in tensorflow::TpuTransferAsyncOpKernelBase44 void TpuTransferAsyncOpKernelBase::ComputeAsync(OpKernelContext* ctx, in ComputeAsync()76 Status TpuTransferAsyncOpKernelBase::RunTransferWithOrdinal( in RunTransferWithOrdinal()98 : TpuTransferAsyncOpKernelBase(ctx, transfer_type, number_of_threads, in TpuTransferAsyncOpKernel()117 : TpuTransferAsyncOpKernelBase(ctx, transfer_type, number_of_threads, in TpuTransferAsyncDynamicOrdinalOpKernel()