Home
last modified time | relevance | path

Searched refs:TransferLiteralFromOutfeed (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.h43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
Dgpu_transfer_manager.cc147 Status GpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::gpu::GpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.h43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
Dcpu_transfer_manager.cc177 Status CpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::CpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.h55 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
Dgeneric_transfer_manager.cc159 Status GenericTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::GenericTransferManager
Dtransfer_manager.h192 virtual Status TransferLiteralFromOutfeed(
Dhlo_runner.cc356 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in ExecuteReplicated()
Dservice.cc1062 execute_backend_->transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc304 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeedLocal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtuple_test.cc575 TF_EXPECT_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in XLA_TEST_F()