Struct rai_core::primitives::ToDevice

pub struct ToDevice<D: Device + Clone> {
    pub device: D,
}

Fields

device: D

Implementations

impl<D: Device + Clone> ToDevice<D>

pub fn new(device: D) -> Self

pub fn device(&self) -> &D

Trait Implementations

impl<D: Clone + Device + Clone> Clone for ToDevice<D>

fn clone(&self) -> ToDevice<D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<D: Debug + Device + Clone> Debug for ToDevice<D>

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

Formats the value using the given formatter.

impl<D1, D2> Eval<D1, ToDevice<D2>> for CandleBackend

where D1: Device, D2: Device + Clone, for<'a> &'a D2: Into<Device>,

fn eval( &self, _: &D1, primitive: &ToDevice<D2>, inputs: &[Tensor], output: &Tensor )

impl<D: Device + Clone + 'static> Primitive for ToDevice<D>

fn clone_boxed(&self) -> Box<dyn Primitive>

fn as_any(&self) -> &dyn Any

fn dot_label(&self) -> String

fn jvp( &self, _output: &Tensor, primals: &[Tensor], tangents: &[Tensor] ) -> Tensor

fn vjp( &self, _output: &Tensor, primals: &[Tensor], cotangent: &Tensor ) -> Vec<Tensor>