Home
last modified time | relevance | path

Searched defs:TransferLiteralToInfeed (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc45 Status CpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::CpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_manager.cc57 Status InfeedManager::TransferLiteralToInfeed(se::StreamExecutor* executor, in TransferLiteralToInfeed() function in xla::gpu::InfeedManager
Dgpu_transfer_manager.cc56 Status GpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::gpu::GpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.cc146 Status GenericTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::GenericTransferManager
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc159 Status StreamExecutorTransferOpImpl::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in tensorflow::StreamExecutorTransferOpImpl
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_transfer_manager.cc91 Status TpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in tensorflow::tpu::TpuTransferManager