Struct rai_core::dispatch::Dispatch

source · 
pub struct Dispatch<B, P> { /* private fields */ }

Trait Implementations§

source§

impl<B: Clone, P: Clone> Clone for Dispatch<B, P>

source§

fn clone(&self) -> Dispatch<B, P>

Returns a copy of the value. Read more
1.0.0 · source§

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

Performs copy-assignment from source. Read more
source§

impl<B: Debug, P: Debug> Debug for Dispatch<B, P>

source§

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

Formats the value using the given formatter. Read more
source§

impl<B, P> Eval<Box<dyn Backend>, Box<dyn Primitive>> for Dispatch<B, P>
where B: Backend + 'static + Sync + Send, P: Primitive + 'static + Sync + Send, Self: Eval<B, P> + 'static,

source§

fn eval( &self, backend: &Box<dyn Backend>, primitive: &Box<dyn Primitive>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Abs> for Dispatch<Cpu, Abs>

source§

fn eval(&self, _: &Cpu, _: &Abs, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Add> for Dispatch<Cpu, Add>

source§

fn eval(&self, _: &Cpu, _: &Add, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Arange<F16>> for Dispatch<Cpu, Arange<F16>>

source§

fn eval(&self, _: &Cpu, primitive: &Arange<F16>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Arange<F32>> for Dispatch<Cpu, Arange<F32>>

source§

fn eval(&self, _: &Cpu, primitive: &Arange<F32>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Arange<F64>> for Dispatch<Cpu, Arange<F64>>

source§

fn eval(&self, _: &Cpu, primitive: &Arange<F64>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Arange<U32>> for Dispatch<Cpu, Arange<U32>>

source§

fn eval(&self, _: &Cpu, primitive: &Arange<U32>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Arange<U8>> for Dispatch<Cpu, Arange<U8>>

source§

fn eval(&self, _: &Cpu, primitive: &Arange<U8>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, ArgMax> for Dispatch<Cpu, ArgMax>

source§

fn eval(&self, _: &Cpu, primitive: &ArgMax, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, ArgMin> for Dispatch<Cpu, ArgMin>

source§

fn eval(&self, _: &Cpu, primitive: &ArgMin, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, AsType<F16>> for Dispatch<Cpu, AsType<F16>>

source§

fn eval( &self, _: &Cpu, primitive: &AsType<F16>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, AsType<F32>> for Dispatch<Cpu, AsType<F32>>

source§

fn eval( &self, _: &Cpu, primitive: &AsType<F32>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, AsType<F64>> for Dispatch<Cpu, AsType<F64>>

source§

fn eval( &self, _: &Cpu, primitive: &AsType<F64>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, AsType<U32>> for Dispatch<Cpu, AsType<U32>>

source§

fn eval( &self, _: &Cpu, primitive: &AsType<U32>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, AsType<U8>> for Dispatch<Cpu, AsType<U8>>

source§

fn eval( &self, _: &Cpu, primitive: &AsType<U8>, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Broadcast> for Dispatch<Cpu, Broadcast>

source§

fn eval( &self, _: &Cpu, primitive: &Broadcast, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Concatenate> for Dispatch<Cpu, Concatenate>

source§

fn eval( &self, _: &Cpu, primitive: &Concatenate, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Cos> for Dispatch<Cpu, Cos>

source§

fn eval(&self, _: &Cpu, _: &Cos, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Div> for Dispatch<Cpu, Div>

source§

fn eval(&self, _: &Cpu, _: &Div, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Equal> for Dispatch<Cpu, Equal>

source§

fn eval(&self, _: &Cpu, _: &Equal, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Erf> for Dispatch<Cpu, Erf>

source§

fn eval(&self, _: &Cpu, _: &Erf, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Exp> for Dispatch<Cpu, Exp>

source§

fn eval(&self, _: &Cpu, _: &Exp, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, FromArray<F16>> for Dispatch<Cpu, FromArray<F16>>

source§

fn eval( &self, _: &Cpu, primitive: &FromArray<F16>, _: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, FromArray<F32>> for Dispatch<Cpu, FromArray<F32>>

source§

fn eval( &self, _: &Cpu, primitive: &FromArray<F32>, _: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, FromArray<F64>> for Dispatch<Cpu, FromArray<F64>>

source§

fn eval( &self, _: &Cpu, primitive: &FromArray<F64>, _: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, FromArray<U32>> for Dispatch<Cpu, FromArray<U32>>

source§

fn eval( &self, _: &Cpu, primitive: &FromArray<U32>, _: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, FromArray<U8>> for Dispatch<Cpu, FromArray<U8>>

source§

fn eval( &self, _: &Cpu, primitive: &FromArray<U8>, _: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Full<F16>> for Dispatch<Cpu, Full<F16>>

source§

fn eval(&self, _: &Cpu, primitive: &Full<F16>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Full<F32>> for Dispatch<Cpu, Full<F32>>

source§

fn eval(&self, _: &Cpu, primitive: &Full<F32>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Full<F64>> for Dispatch<Cpu, Full<F64>>

source§

fn eval(&self, _: &Cpu, primitive: &Full<F64>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Full<U32>> for Dispatch<Cpu, Full<U32>>

source§

fn eval(&self, _: &Cpu, primitive: &Full<U32>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Full<U8>> for Dispatch<Cpu, Full<U8>>

source§

fn eval(&self, _: &Cpu, primitive: &Full<U8>, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Gather> for Dispatch<Cpu, Gather>

source§

fn eval(&self, _: &Cpu, primitive: &Gather, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Greater> for Dispatch<Cpu, Greater>

source§

fn eval(&self, _: &Cpu, _: &Greater, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, GreaterEqual> for Dispatch<Cpu, GreaterEqual>

source§

fn eval(&self, _: &Cpu, _: &GreaterEqual, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, IndexSelect> for Dispatch<Cpu, IndexSelect>

source§

fn eval( &self, _: &Cpu, primitive: &IndexSelect, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Less> for Dispatch<Cpu, Less>

source§

fn eval(&self, _: &Cpu, _: &Less, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, LessEqual> for Dispatch<Cpu, LessEqual>

source§

fn eval(&self, _: &Cpu, _: &LessEqual, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Log> for Dispatch<Cpu, Log>

source§

fn eval(&self, _: &Cpu, _: &Log, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Log10> for Dispatch<Cpu, Log10>

source§

fn eval(&self, _: &Cpu, _: &Log10, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Log2> for Dispatch<Cpu, Log2>

source§

fn eval(&self, _: &Cpu, _: &Log2, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, LogSoftmax> for Dispatch<Cpu, LogSoftmax>

source§

fn eval( &self, _: &Cpu, primitive: &LogSoftmax, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, MatMul> for Dispatch<Cpu, MatMul>

source§

fn eval(&self, _: &Cpu, _: &MatMul, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Maximum> for Dispatch<Cpu, Maximum>

source§

fn eval(&self, _: &Cpu, _: &Maximum, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Mul> for Dispatch<Cpu, Mul>

source§

fn eval(&self, _: &Cpu, _: &Mul, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Narrow> for Dispatch<Cpu, Narrow>

source§

fn eval(&self, _: &Cpu, primitive: &Narrow, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Negative> for Dispatch<Cpu, Negative>

source§

fn eval(&self, _: &Cpu, _: &Negative, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Normal> for Dispatch<Cpu, Normal>

source§

fn eval(&self, _: &Cpu, _: &Normal, _: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, NotEqual> for Dispatch<Cpu, NotEqual>

source§

fn eval(&self, _: &Cpu, _: &NotEqual, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, PowerFloat> for Dispatch<Cpu, PowerFloat>

source§

fn eval( &self, _: &Cpu, primitive: &PowerFloat, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, ReduceMax> for Dispatch<Cpu, ReduceMax>

source§

fn eval( &self, _: &Cpu, primitive: &ReduceMax, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, ReduceMin> for Dispatch<Cpu, ReduceMin>

source§

fn eval( &self, _: &Cpu, primitive: &ReduceMin, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, ReduceSum> for Dispatch<Cpu, ReduceSum>

source§

fn eval( &self, _: &Cpu, primitive: &ReduceSum, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Reshape> for Dispatch<Cpu, Reshape>

source§

fn eval(&self, _: &Cpu, primitive: &Reshape, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Rsqrt> for Dispatch<Cpu, Rsqrt>

source§

fn eval(&self, _: &Cpu, _: &Rsqrt, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Sign> for Dispatch<Cpu, Sign>

source§

fn eval(&self, _: &Cpu, _: &Sign, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Sin> for Dispatch<Cpu, Sin>

source§

fn eval(&self, _: &Cpu, _: &Sin, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Softmax> for Dispatch<Cpu, Softmax>

source§

fn eval(&self, _: &Cpu, primitive: &Softmax, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Sqrt> for Dispatch<Cpu, Sqrt>

source§

fn eval(&self, _: &Cpu, _: &Sqrt, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Square> for Dispatch<Cpu, Square>

source§

fn eval(&self, _: &Cpu, _: &Square, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Sub> for Dispatch<Cpu, Sub>

source§

fn eval(&self, _: &Cpu, _: &Sub, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Tanh> for Dispatch<Cpu, Tanh>

source§

fn eval(&self, _: &Cpu, _: &Tanh, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, ToContiguous> for Dispatch<Cpu, ToContiguous>

source§

fn eval(&self, _: &Cpu, _: &ToContiguous, inputs: &[Tensor], output: &Tensor)

source§

impl Eval<Cpu, Transpose> for Dispatch<Cpu, Transpose>

source§

fn eval( &self, _: &Cpu, primitive: &Transpose, inputs: &[Tensor], output: &Tensor )

source§

impl Eval<Cpu, Where> for Dispatch<Cpu, Where>

source§

fn eval(&self, _: &Cpu, _: &Where, inputs: &[Tensor], output: &Tensor)

source§

impl<B, P> Eval<dyn Backend, dyn Primitive> for Dispatch<B, P>
where B: Backend + 'static + Sync + Send, P: Primitive + 'static + Sync + Send, Self: Eval<B, P> + 'static,

source§

fn eval( &self, backend: &dyn Backend, primitive: &dyn Primitive, inputs: &[Tensor], output: &Tensor )

Auto Trait Implementations§

§

impl<B, P> RefUnwindSafe for Dispatch<B, P>
where B: RefUnwindSafe, P: RefUnwindSafe,

§

impl<B, P> Send for Dispatch<B, P>
where B: Send, P: Send,

§

impl<B, P> Sync for Dispatch<B, P>
where B: Sync, P: Sync,

§

impl<B, P> Unpin for Dispatch<B, P>
where B: Unpin, P: Unpin,

§

impl<B, P> UnwindSafe for Dispatch<B, P>
where B: UnwindSafe, P: UnwindSafe,

Blanket Implementations§

source§

impl<T> Any for T
where T: 'static + ?Sized,

source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
source§

impl<T> Borrow<T> for T
where T: ?Sized,

source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
source§

impl<T> DynClone for T
where T: Clone,

source§

fn __clone_box(&self, _: Private) -> *mut ()

source§

impl<T> From<T> for T

source§

fn from(t: T) -> T

Returns the argument unchanged.

source§

impl<T> Instrument for T

source§

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
source§

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more
source§

impl<T, U> Into<U> for T
where U: From<T>,

source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

§

impl<T> Pointable for T

§

const ALIGN: usize = _

The alignment of pointer.
§

type Init = T

The type for initializers.
§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
source§

impl<T> ToOwned for T
where T: Clone,

§

type Owned = T

The resulting type after obtaining ownership.
source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

§

type Error = Infallible

The type returned in the event of a conversion error.
source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

§

fn vzip(self) -> V

source§

impl<T> WithSubscriber for T

source§

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
source§

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more
§

impl<T> ErasedDestructor for T
where T: 'static,