Home
last modified time | relevance | path

Searched refs:TransferLiteralToInfeed (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.h41 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
Dgpu_transfer_manager.cc48 Status GpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::gpu::GpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.h42 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
Dcpu_transfer_manager.cc93 Status CpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::CpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.h53 Status TransferLiteralToInfeed(se::StreamExecutor* executor,
Dgeneric_transfer_manager.cc142 Status GenericTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::GenericTransferManager
Dtransfer_manager.h202 virtual Status TransferLiteralToInfeed(se::StreamExecutor* executor,
Dhlo_runner.cc270 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralToInfeed( in ExecuteReplicatedImpl()
Dservice.cc1011 return execute_backend_->transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeed()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.h54 Status TransferLiteralToInfeed(stream_executor::StreamExecutor* executor,
Dtpu_transfer_manager.cc89 Status TpuTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/core/tpu/kernels/
Dinfeed_ops.cc446 transfer_manager->TransferLiteralToInfeed(stream_executor, literal)); in DoWork()
514 transfer_manager->TransferLiteralToInfeed(stream_executor, tuple)); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc432 return backend().transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeedLocal()