Enum DispatchDevice 

pub enum DispatchDevice {
    Host(HostDevice),
    Autodiff(Box<DispatchDevice>, GradientCheckpointingStrategy),
}

Variants

Host(HostDevice)

Autodiff(Box<DispatchDevice>, GradientCheckpointingStrategy)

Implementations

impl DispatchDevice

pub fn identity(&self) -> Option<DeviceIdentity>

pub fn autodiff(device: impl Into<Self>) -> Self

pub fn autodiff_with_gradient_checkpointing( device: impl Into<Self>, strategy: GradientCheckpointingStrategy, ) -> Self

pub fn context(&self) -> AutodiffContext

pub fn gradient_checkpointing_strategy(&self) -> GradientCheckpointingStrategy

pub fn with_context(&self, context: AutodiffContext) -> Self

Trait Implementations

impl Clone for DispatchDevice

fn clone(&self) -> Self

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 to_id(&self) -> DeviceId

Retrieve the device id from the device.

fn from_id(id: DeviceId) -> Self

Create a device from its id.

impl DeviceOps for DispatchDevice

fn defaults(&self) -> Option<DeviceSettings>

Device-specific defaults for dynamically selected or remote backends. Static backends may retain defaults derived from their element types.

fn inner(&self) -> &Self

Returns the inner device without autodiff enabled.

fn id(&self) -> DeviceId

Returns the device id.

impl Eq for DispatchDevice

impl From<HostDevice> for DispatchDevice

fn from(device: HostDevice) -> Self

Converts to this type from the input type.

impl PartialEq for DispatchDevice

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

Equality operator ==.

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

Inequality operator !=.