Enum DispatchDevice 

pub enum DispatchDevice {
    NdArray(NdArrayDevice),
}

Represents a device for the Dispatch.

Each variant corresponds to a backend that the Dispatch can dispatch operations to.

Example

use burn::DispatchDevice;

#[cfg(feature = "cpu")]
let cpu_device = DispatchDevice::Cpu(Default::default());

#[cfg(feature = "cuda")]
let cuda_device = DispatchDevice::Cuda(Default::default());

Variants

NdArray(NdArrayDevice)

Available on crate feature ndarray only.

The NdArray backend device (CPU-only).

Trait Implementations

impl Clone for DispatchDevice

fn clone(&self) -> DispatchDevice

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DispatchDevice

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for DispatchDevice

fn default() -> Self

Returns the “default value” for a type.

impl Device for DispatchDevice

fn from_id(device_id: DeviceId) -> Self

Create a device from its id.

fn to_id(&self) -> DeviceId

Retrieve the device id from the device.

fn device_count(type_id: u16) -> usize

Returns the number of devices available under the provided type id.

fn device_count_total() -> usize

Returns the total number of devices that can be handled by the runtime.

impl DeviceOps for DispatchDevice

fn inner(&self) -> &Self

Returns the inner device without autodiff enabled.

fn id(&self) -> DeviceId

Returns the device id.

impl From<NdArrayDevice> for DispatchDevice

Available on crate feature ndarray only.

fn from(device: NdArrayDevice) -> Self

Converts to this type from the input type.

impl PartialEq for DispatchDevice

fn eq(&self, other: &Self) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for DispatchDevice