Struct Im2colLayout 

pub struct Im2colLayout {
    pub shape_out: Sequence<FastDivmod<u32>>,
    pub padded_channels: FastDivmod<u32>,
    pub rows: u32,
    pub cols: u32,
    pub params: ConvolutionParams,
    pub config: GlobalLayoutConfig,
}

Maps a 4D NHWC tensor to a 2D column matrix using the im2col transformation It first decomposes the (m, k) matrix into ((n, out_h, out_w), (k_h, k_w, c)), then applies the convolution parameters to calculate the position in the input tensor for that kernel element.

Fields

shape_out: Sequence<FastDivmod<u32>>

Shape of output DHW

padded_channels: FastDivmod<u32>

Shape of channel, for decomposing k

rows: u32

Shape of the combined m dimension, including padding

cols: u32

Shape of the combined k dimension, including padding

params: ConvolutionParams

Comptime parameters for the convolution

config: GlobalLayoutConfig

Global memory config for the backing tensor

Implementations

impl Im2colLayout

pub fn new<G: GlobalConfig>( rows: u32, cols: u32, padded_channels: FastDivmod<u32>, shape_out: Sequence<FastDivmod<u32>>, config: GlobalLayoutConfig, params: ConvolutionParams, ) -> Im2colLayout

pub fn __expand_new<G: GlobalConfig>( scope: &Scope, rows: <u32 as CubeType>::ExpandType, cols: <u32 as CubeType>::ExpandType, padded_channels: <FastDivmod<u32> as CubeType>::ExpandType, shape_out: <Sequence<FastDivmod<u32>> as CubeType>::ExpandType, config: GlobalLayoutConfig, params: ConvolutionParams, ) -> <Im2colLayout as CubeType>::ExpandType

Trait Implementations

impl Clone for Im2colLayout

fn clone(&self) -> Im2colLayout

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeType for Im2colLayout

type ExpandType = Im2colLayoutExpand

impl LaunchArg for Im2colLayout

type RuntimeArg<R: Runtime> = Im2colLayoutLaunch<R>

The runtime argument for the kernel.

type CompilationArg = Im2colLayoutCompilationArg

Compilation argument.

fn register<R: Runtime>( arg: Self::RuntimeArg<R>, launcher: &mut KernelLauncher<R>, ) -> Self::CompilationArg

fn expand( arg: &Self::CompilationArg, builder: &mut KernelBuilder, ) -> <Self as CubeType>::ExpandType

Register a variable during compilation that fill the KernelBuilder.

impl Layout for Im2colLayout

type Coordinates = (usize, u32, u32)

The coordinate type used by the conceptual tensor represented by this layout, i.e. (u32, u32, u32) for a fixed-rank 3D tensor. This does not have to match the rank of the underlying storage (if applicable). It’s only how the tensor is interpreted (viewed) by the code.

type SourceCoordinates = NhwcCoords

The coordinate type used by the inner storage wrapped in this layout, i.e. u32 for Array, or (u32, u32) for a 2D view.

fn to_source_pos(&self, pos: Self::Coordinates) -> NhwcCoords

Transform a set of n-dimensional coordinates to a source coordinate space. It is recommended to use absolute positions here, and handle the translation into vectors at the lowest level (global memory layout).

fn shape(&self) -> Self::Coordinates

The shape of the conceptual tensor represented by this layout. Not necessarily the extent of the underlying storage, but only this view of it.

fn to_source_pos_checked(&self, pos: Self::Coordinates) -> (NhwcCoords, bool)

Transform a set of n-dimensional coordinates to an offset into the underlying storage, and return whether the position is in bounds of this layout. See also Layout::to_source_pos

fn is_in_bounds(&self, pos: Self::Coordinates) -> bool

fn __expand_to_source_pos( scope: &Scope, this: &Self::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <Self::SourceCoordinates as CubeType>::ExpandType

fn __expand_to_source_pos_checked( scope: &Scope, this: &Self::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> (<Self::SourceCoordinates as CubeType>::ExpandType, <bool as CubeType>::ExpandType)

fn __expand_shape( scope: &Scope, this: &Self::ExpandType, ) -> <Self::Coordinates as CubeType>::ExpandType

fn __expand_is_in_bounds( scope: &Scope, this: &Self::ExpandType, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType