Trait melange::tensor::shape::ReductionOptChunckSize

pub trait ReductionOptChunckSize<T, Ax>: StaticShape {
    type Output: Unsigned;
}

Trait operator that computes the intrinsic optimal chunk size i.e. the largest contiguous group of elements in storage after a reduction performed on the axis at (0-starting) index Ax (a type-level unsigned integer).

Note that this size is intrinsic to the shape and does not take into account the real layout.

Associated Types

type Output: Unsigned

Loading content...

Implementations on Foreign Types

impl<Ax, T> ReductionOptChunckSize<T, Ax> for ATerm

type Output = U1

impl<Ax, D, Ar, T> ReductionOptChunckSize<T, Ax> for TArr<D, Ar> where
    D: Mul<<Ar as ReductionOptChunckSize<T, Ax>>::Output>,
    Self: Len + StaticShape,
    Length<Self>: Sub<B1>,
    Ax: IsEqual<Sub1<Length<Self>>>,
    Ar: ReductionOptChunckSize<T, Ax> + NumElements<T>,
    Eq<Ax, Sub1<Length<Self>>>: If<U1, Prod<D, <Ar as ReductionOptChunckSize<T, Ax>>::Output>>,
    <Eq<Ax, Sub1<Length<Self>>> as If<U1, Prod<D, <Ar as ReductionOptChunckSize<T, Ax>>::Output>>>::Output: Unsigned, 

type Output = <Eq<Ax, Sub1<Length<Self>>> as If<U1, Prod<D, <Ar as ReductionOptChunckSize<T, Ax>>::Output>>>::Output

Loading content...

Implementors

Loading content...