Home
last modified time | relevance | path

Searched refs:transfer_manager_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtransfer_manager_test.cc45 return transfer_manager_->GetByteSizeRequirement(shape); in TransferManagerTest()
56 return transfer_manager_ in AllocateDeviceBuffer()
77 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
81 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
93 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
97 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
111 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
115 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
127 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
131 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
[all …]
Dlocal_client_test_base.h127 TransferManager* transfer_manager_; variable
Dlocal_client_test_base.cc126 transfer_manager_ = in LocalClientTestBase()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc84 transfer_manager_(client->backend().transfer_manager()), in XlaDeviceContext()
154 !transfer_manager_->CanShapedBufferBeAccessedNow( in CopyCPUTensorToDevice()
161 TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync( in CopyCPUTensorToDevice()
247 transfer_manager_->TransferLiteralFromDevice( in CopyDeviceTensorToCPU()
Dxla_device_context.h81 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager()
109 xla::TransferManager* transfer_manager_; variable
/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.h94 TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager()
169 TransferManager* transfer_manager_; variable
Dbackend.cc150 transfer_manager_(transfer_manager), in Backend()
225 return transfer_manager_->ResetDevices(stream_executors_); in ResetDevices()