Enum ViewArg 

pub enum ViewArg<C: Coordinates, R: Runtime> {
    Array(BufferArg<R>, VirtualViewLayoutLaunch<C, Coords1d, BufferArg<R>, R>),
    Tensor(TensorArg<R>, VirtualViewLayoutLaunch<C, Coords1d, TensorArg<R>, R>),
    TensorMapTiled(TensorMapArg<R, Tiled>, VirtualViewLayoutLaunch<C, Sequence<i32>, TensorMapArg<R, Tiled>, R>),
    TensorMapIm2col(TensorMapArg<R, Im2col>, VirtualViewLayoutLaunch<C, (Sequence<i32>, Sequence<i32>), TensorMapArg<R, Im2col>, R>),
    Quantized {
        values: Box<ViewArg<C, R>>,
        scales: ScaleBindings<C, R>,
        scheme: QuantScheme,
    },
}

Variants

Array(BufferArg<R>, VirtualViewLayoutLaunch<C, Coords1d, BufferArg<R>, R>)

Tensor(TensorArg<R>, VirtualViewLayoutLaunch<C, Coords1d, TensorArg<R>, R>)

TensorMapTiled(TensorMapArg<R, Tiled>, VirtualViewLayoutLaunch<C, Sequence<i32>, TensorMapArg<R, Tiled>, R>)

TensorMapIm2col(TensorMapArg<R, Im2col>, VirtualViewLayoutLaunch<C, (Sequence<i32>, Sequence<i32>), TensorMapArg<R, Im2col>, R>)

Quantized

Fields

values: Box<ViewArg<C, R>>

scales: ScaleBindings<C, R>

scheme: QuantScheme

Implementations

impl<C: Coordinates, R: Runtime> ViewArg<C, R>

pub fn new_array<L: Layout<Coordinates = C, SourceCoordinates = Coords1d> + ViewLayoutLaunchArg>( buffer: BufferArg<R>, layout: L::RuntimeArg<R>, ) -> Self

pub fn new_tensor<L: Layout<Coordinates = C, SourceCoordinates = Coords1d> + ViewLayoutLaunchArg>( buffer: TensorArg<R>, layout: L::RuntimeArg<R>, ) -> Self

pub fn new_tensor_map_tiled<L: Layout<Coordinates = C, SourceCoordinates: IntoDyn> + ViewLayoutLaunchArg>( buffer: TensorMapArg<R, Tiled>, layout: L::RuntimeArg<R>, ) -> ViewArg<C, R>

pub fn new_tensor_map_im2col<L: Layout<Coordinates = C, SourceCoordinates = (P, O)> + ViewLayoutLaunchArg, P: IntoDyn, O: IntoDyn>( buffer: TensorMapArg<R, Im2col>, layout: L::RuntimeArg<R>, ) -> ViewArg<C, R>

pub fn new_quantized( values: Self, scales: ScaleBindings<C, R>, scheme: QuantScheme, ) -> Self

Create a new view arg that dequantizes on read, against one scale binding per scheme level. The inner scales layout should take values indices and map them to the corresponding scale.

Panics when the bindings and the scheme’s levels disagree in count, since a missing level would be dropped from the reconstruction, and for an global level this reader cannot serve. See quant::check_scale_bindings.