View Source EXLA.Backend (EXLA v0.2.1)

A Nx tensor backend for the data kept on the device.

You can directly transfer to this backend by calling Nx.backend_transfer/2 or Nx.backend_copy/2. It allows the following options:

  • :client - the client to store the data on. Defaults to the client configured in Nx.Defn, otherwise uses :host.

  • :device_id - which device to store it on

To get the data out of the device backend into a regular tensor, call Nx.backend_transfer/1 (with the device tensor as the single argument).