Home
last modified time | relevance | path

Searched refs:transfer_manager_ (Results 1 – 8 of 8) 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()
115 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
119 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
133 ASSERT_IS_OK(transfer_manager_->TransferLiteralToDevice(stream_, literal, in XLA_TEST_F()
137 transfer_manager_->TransferLiteralFromDevice(stream_, device_buffer)); in XLA_TEST_F()
[all …]
Dlocal_client_test_base.h138 TransferManager* transfer_manager_; variable
Dlocal_client_test_base.cc132 transfer_manager_ = in LocalClientTestBase()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc89 transfer_manager_(client->backend().transfer_manager()), in XlaDeviceContext()
162 !transfer_manager_->CanShapedBufferBeAccessedNow( in CopyCPUTensorToDevice()
169 TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync( in CopyCPUTensorToDevice()
257 transfer_manager_->TransferLiteralFromDevice( in CopyDeviceTensorToCPU()
Dxla_device_context.h81 xla::TransferManager* transfer_manager() const { return transfer_manager_; } in transfer_manager()
112 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.cc128 transfer_manager_(transfer_manager), in Backend()
198 return transfer_manager_->ResetDevices(stream_executors_); in ResetDevices()
/external/tensorflow/tensorflow/core/platform/s3/
Ds3_file_system.cc241 transfer_manager_(transfer_manager), in S3RandomAccessFile()
273 transfer_manager_.get()->DownloadFile(bucket_.c_str(), object_.c_str(), in ReadS3TransferManager()
287 transfer_manager_.get()->RetryDownload(handle); in ReadS3TransferManager()
343 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager_; member in tensorflow::__anon6d29161a0111::S3RandomAccessFile
356 transfer_manager_(transfer_manager), in S3WritableFile()
402 transfer_manager_.get()->UploadFile( in Sync()
413 transfer_manager_.get()->RetryUpload(outfile_, handle); in Sync()
434 std::shared_ptr<Aws::Transfer::TransferManager> transfer_manager_; member in tensorflow::__anon6d29161a0111::S3WritableFile