Home
last modified time | relevance | path

Searched refs:TransferLiteralToInfeed (Results 1 – 10 of 10) 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.h41 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.cc154 Status GenericTransferManager::TransferLiteralToInfeed( in TransferLiteralToInfeed() function in xla::GenericTransferManager
Dtransfer_manager.h187 virtual Status TransferLiteralToInfeed(se::StreamExecutor* executor,
Dhlo_runner.cc337 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralToInfeed( in ExecuteReplicated()
Dservice.cc1034 return execute_backend_->transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeed()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc295 return backend().transfer_manager()->TransferLiteralToInfeed(executor, in TransferToInfeedLocal()