Home
last modified time | relevance | path

Searched refs:TpuTransferAsyncDynamicOrdinalOpKernel (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtransfer_ops.h88 class TpuTransferAsyncDynamicOrdinalOpKernel
91 explicit TpuTransferAsyncDynamicOrdinalOpKernel(
100 TpuTransferAsyncDynamicOrdinalOpKernel(
101 const TpuTransferAsyncDynamicOrdinalOpKernel&) = delete;
102 TpuTransferAsyncDynamicOrdinalOpKernel& operator=(
103 const TpuTransferAsyncDynamicOrdinalOpKernel&) = delete;
H A Doutfeed_ops.cc70 StreamExecutorOutfeedDequeueOp<TpuTransferAsyncDynamicOrdinalOpKernel>);
73 StreamExecutorOutfeedDequeueOp<TpuTransferAsyncDynamicOrdinalOpKernel>);
78 TpuTransferAsyncDynamicOrdinalOpKernel>);
81 TpuTransferAsyncDynamicOrdinalOpKernel>);
H A Dtransfer_ops.cc114 TpuTransferAsyncDynamicOrdinalOpKernel::TpuTransferAsyncDynamicOrdinalOpKernel( in TpuTransferAsyncDynamicOrdinalOpKernel() function in tensorflow::TpuTransferAsyncDynamicOrdinalOpKernel
120 Status TpuTransferAsyncDynamicOrdinalOpKernel::RunTransfer( in RunTransfer()