Home
last modified time | relevance | path

Searched refs:TransferLiteralToDeviceAsync (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h155 virtual Status TransferLiteralToDeviceAsync(
159 Status TransferLiteralToDeviceAsync(se::Stream* stream, in TransferLiteralToDeviceAsync() function
162 return TransferLiteralToDeviceAsync(stream, literal, device_buffer, in TransferLiteralToDeviceAsync()
Dgeneric_transfer_manager.h48 Status TransferLiteralToDeviceAsync(
Dgeneric_transfer_manager.cc92 Status GenericTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync() function in xla::GenericTransferManager
Dtransfer_manager.cc109 TF_RETURN_IF_ERROR(TransferLiteralToDeviceAsync( in TransferLiteralToDevice()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.h44 Status TransferLiteralToDeviceAsync(
Dtpu_transfer_manager.cc67 Status TpuTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync() function in tensorflow::tpu::TpuTransferManager
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc169 TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync( in CopyCPUTensorToDevice()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc791 TF_CHECK_OK(transfer_manager->TransferLiteralToDeviceAsync( in BufferFromHostBuffer()
796 TF_CHECK_OK(transfer_manager->TransferLiteralToDeviceAsync( in BufferFromHostBuffer()
902 TF_CHECK_OK(transfer_manager->TransferLiteralToDeviceAsync( in BufferFromHostLiteral()