Searched refs:TransferLiteralToDeviceAsync (Results 1 – 5 of 5) sorted by relevance
149 virtual Status TransferLiteralToDeviceAsync(153 Status TransferLiteralToDeviceAsync(se::Stream* stream, in TransferLiteralToDeviceAsync() function156 return TransferLiteralToDeviceAsync(stream, literal, device_buffer, in TransferLiteralToDeviceAsync()
48 Status TransferLiteralToDeviceAsync(
99 Status GenericTransferManager::TransferLiteralToDeviceAsync( in TransferLiteralToDeviceAsync() function in xla::GenericTransferManager
106 TF_RETURN_IF_ERROR(TransferLiteralToDeviceAsync( in TransferLiteralToDevice()
161 TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync( in CopyCPUTensorToDevice()