Searched refs:TransferLiteralFromOutfeed (Results 1 – 11 of 11) sorted by relevance
43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
147 Status GpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::gpu::GpuTransferManager
177 Status CpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::CpuTransferManager
55 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
159 Status GenericTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::GenericTransferManager
192 virtual Status TransferLiteralFromOutfeed(
356 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in ExecuteReplicated()
1062 execute_backend_->transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeed()
304 TF_RETURN_IF_ERROR(backend().transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeedLocal()
575 TF_EXPECT_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in XLA_TEST_F()