Struct Im2colArgs 

pub struct Im2colArgs {
    pub pixel_box_lower_corner: Vec<i32>,
    pub pixel_box_upper_corner: Vec<i32>,
    pub channels_per_pixel: u32,
    pub pixels_per_column: u32,
}

Args for im2col tensor maps

Fields

pixel_box_lower_corner: Vec<i32>

Pixel box lower corner. This is the logical upper left corner in the input tensor, when offset is 0. The length of this value should equal the spatial dimensions of the input tensor (i.e. h, w for an NHWC tensor). Should normally be set to -padding.

pixel_box_upper_corner: Vec<i32>

Pixel box top corner. This is the logical lower right corner in the input tensor, when offset is 0. The length of this value should equal the spatial dimensions of the input tensor (i.e. h, w for an NHWC tensor). Should normally be set to padding - kernel_size - 1 (where kernel_size accounts for dilation). This is not equal to padding, it’s equal to the bounding box for the top left corner of the kernel.

channels_per_pixel: u32

Channels to load per pixel, should be a multiple or divisor of the matmul tile size. This is not the total number of channels in the tensor, but only the number loaded in each load. Must be <= 256 and aligned to 16 bytes.

pixels_per_column: u32

Pixels per column, equivalent to the m/n dimension of each tile in the matrix multiplication. i.e. NHW for a 4D tensor. Must be <= 256 and aligned to 16 bytes

Implementations

impl Im2colArgs

pub fn new( pixel_box_lower_corner: Vec<i32>, pixel_box_upper_corner: Vec<i32>, channels_per_pixel: u32, pixels_per_column: u32, ) -> Self

Constructs a new Im2colArgs.

Trait Implementations

impl Clone for Im2colArgs

fn clone(&self) -> Im2colArgs

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Im2colArgs

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

Formats the value using the given formatter.

impl Hash for Im2colArgs

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for Im2colArgs

fn eq(&self, other: &Im2colArgs) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for Im2colArgs

impl StructuralPartialEq for Im2colArgs