Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Doutfeed_ops.cc57 transfer_manager->TransferLiteralFromOutfeed(stream_executor, literal)); in DoWork()
100 transfer_manager->TransferLiteralFromOutfeed(stream_executor, literal)); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.h43 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
Dgpu_transfer_manager.cc116 Status GpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::gpu::GpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.h44 Status TransferLiteralFromOutfeed(se::StreamExecutor* executor,
Dcpu_transfer_manager.cc178 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.cc147 Status GenericTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in xla::GenericTransferManager
Dtransfer_manager.h208 virtual Status TransferLiteralFromOutfeed(
Dhlo_runner.cc289 TF_CHECK_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in ExecuteReplicatedImpl()
Dservice.cc1039 execute_backend_->transfer_manager()->TransferLiteralFromOutfeed( in TransferFromOutfeed()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.h57 Status TransferLiteralFromOutfeed(
Dtpu_transfer_manager.cc129 Status TpuTransferManager::TransferLiteralFromOutfeed( in TransferLiteralFromOutfeed() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc440 return backend().transfer_manager()->TransferLiteralFromOutfeed(executor, in TransferFromOutfeedLocal()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtuple_test.cc575 TF_EXPECT_OK(backend().transfer_manager()->TransferLiteralFromOutfeed( in XLA_TEST_F()