Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.h149 virtual Status TransferLiteralToDeviceAsync(
153 Status TransferLiteralToDeviceAsync(se::Stream* stream, in TransferLiteralToDeviceAsync() function
156 return TransferLiteralToDeviceAsync(stream, literal, device_buffer, in TransferLiteralToDeviceAsync()
Dgeneric_transfer_manager.h48 Status TransferLiteralToDeviceAsync(
Dgeneric_transfer_manager.cc99 Status GenericTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync() function in xla::GenericTransferManager
Dtransfer_manager.cc106 TF_RETURN_IF_ERROR(TransferLiteralToDeviceAsync( in TransferLiteralToDevice()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc161 TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync( in CopyCPUTensorToDevice()