Trait rai_core::Eval

pub trait Eval<D, P>: DynClone + Sync + Send + 'static
where
    D: ?Sized,
    P: ?Sized,
{
    // Required method
    fn eval(
        &self,
        device: &D,
        primitive: &P,
        inputs: &[Tensor],
        output: &Tensor
    );
}

Required Methods

fn eval(&self, device: &D, primitive: &P, inputs: &[Tensor], output: &Tensor)

Implementors

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

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

impl<D, P, B> Eval<Box<dyn Device>, Box<dyn Primitive>> for BackendWrapper<D, P, B>

where D: Device + 'static + Sync + Send + Clone, P: Primitive + 'static + Sync + Send + Clone, B: Eval<D, P> + 'static + Clone,

impl<D, P, B> Eval<dyn Device, dyn Primitive> for BackendWrapper<D, P, B>

where D: Device + 'static + Sync + Send + Clone, P: Primitive + 'static + Sync + Send + Clone, B: Eval<D, P> + 'static + Clone,

impl<D, T> Eval<D, Arange<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: WithDType,

impl<D, T> Eval<D, FromArray<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: WithDType,

impl<D, T> Eval<D, Full<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: WithDType,

impl<D, T> Eval<D, Normal<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: FloatDType,

impl<D, T> Eval<D, Random<T>> for CandleBackend

where D: Device, for<'a> &'a D: Into<Device>, T: Type, T::Repr: FloatDType,

impl<D, T> Eval<D, ToDType<T>> for CandleBackend

where D: Device, T: Type + Into<DType>,

impl<D: Device> Eval<D, Abs> for CandleBackend

impl<D: Device> Eval<D, Add> for CandleBackend

impl<D: Device> Eval<D, ArgMax> for CandleBackend

impl<D: Device> Eval<D, ArgMin> for CandleBackend

impl<D: Device> Eval<D, AvgPool1d> for CandleBackend

impl<D: Device> Eval<D, AvgPool2d> for CandleBackend

impl<D: Device> Eval<D, Broadcast> for CandleBackend

impl<D: Device> Eval<D, Concatenate> for CandleBackend

impl<D: Device> Eval<D, Conv1d> for CandleBackend

impl<D: Device> Eval<D, Conv2d> for CandleBackend

impl<D: Device> Eval<D, ConvTranspose1d> for CandleBackend

impl<D: Device> Eval<D, ConvTranspose2d> for CandleBackend

impl<D: Device> Eval<D, Cos> for CandleBackend

impl<D: Device> Eval<D, Div> for CandleBackend

impl<D: Device> Eval<D, Equal> for CandleBackend

impl<D: Device> Eval<D, Erf> for CandleBackend

impl<D: Device> Eval<D, Exp> for CandleBackend

impl<D: Device> Eval<D, Gather> for CandleBackend

impl<D: Device> Eval<D, Greater> for CandleBackend

impl<D: Device> Eval<D, GreaterEqual> for CandleBackend

impl<D: Device> Eval<D, IndexAdd> for CandleBackend

impl<D: Device> Eval<D, IndexSelect> for CandleBackend

impl<D: Device> Eval<D, Less> for CandleBackend

impl<D: Device> Eval<D, LessEqual> for CandleBackend

impl<D: Device> Eval<D, Log2> for CandleBackend

impl<D: Device> Eval<D, Log10> for CandleBackend

impl<D: Device> Eval<D, Log> for CandleBackend

impl<D: Device> Eval<D, LogSoftmax> for CandleBackend

impl<D: Device> Eval<D, MatMul> for CandleBackend

impl<D: Device> Eval<D, MaxPool1d> for CandleBackend

impl<D: Device> Eval<D, MaxPool2d> for CandleBackend

impl<D: Device> Eval<D, Maximum> for CandleBackend

impl<D: Device> Eval<D, Minimum> for CandleBackend

impl<D: Device> Eval<D, Mul> for CandleBackend

impl<D: Device> Eval<D, Narrow> for CandleBackend

impl<D: Device> Eval<D, Negative> for CandleBackend

impl<D: Device> Eval<D, NotEqual> for CandleBackend

impl<D: Device> Eval<D, Permute> for CandleBackend

impl<D: Device> Eval<D, PowerFloat> for CandleBackend

impl<D: Device> Eval<D, ReduceMax> for CandleBackend

impl<D: Device> Eval<D, ReduceMin> for CandleBackend

impl<D: Device> Eval<D, ReduceSum> for CandleBackend

impl<D: Device> Eval<D, Reshape> for CandleBackend

impl<D: Device> Eval<D, Rsqrt> for CandleBackend

impl<D: Device> Eval<D, ScatterAdd> for CandleBackend

impl<D: Device> Eval<D, Sign> for CandleBackend

impl<D: Device> Eval<D, Sin> for CandleBackend

impl<D: Device> Eval<D, Softmax> for CandleBackend

impl<D: Device> Eval<D, Sqrt> for CandleBackend

impl<D: Device> Eval<D, Square> for CandleBackend

impl<D: Device> Eval<D, Sub> for CandleBackend

impl<D: Device> Eval<D, Tanh> for CandleBackend

impl<D: Device> Eval<D, ToContiguous> for CandleBackend

impl<D: Device> Eval<D, Transpose> for CandleBackend

impl<D: Device> Eval<D, UpsampleNearest1d> for CandleBackend

impl<D: Device> Eval<D, UpsampleNearest2d> for CandleBackend

impl<D: Device> Eval<D, Where> for CandleBackend