Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TransferLiteralToDeviceAsync (0.32 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/xla_device_context.cc 

          // synchronized.
      TF_RETURN_IF_ERROR(host_to_device_stream_->WaitFor(stream_.get()));
    }

    TF_RETURN_IF_ERROR(transfer_manager_->TransferLiteralToDeviceAsync(
        host_to_device_stream_.get(), literal, xla_tensor->shaped_buffer()));

    if (UseMultipleStreams()) {
      TF_ASSIGN_OR_RETURN(std::shared_ptr<se::Event> event,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 12.7K bytes
    - Viewed (0)
Back to top