Struct PackedFormat 

pub struct PackedFormat {
    pub dt: DatumType,
    pub r: usize,
    pub alignment_bytes: usize,
    pub end_padding_record: usize,
}

Fields

dt: DatumType

r: usize

alignment_bytes: usize

end_padding_record: usize

Implementations

impl PackedFormat

pub const fn new( dt: DatumType, nr: usize, alignment_bytes: usize, ) -> PackedFormat

pub const fn with_end_padding_record( self, end_padding_record: usize, ) -> PackedFormat

pub fn align(self, alignment: usize) -> PackedFormat

pub fn alignment(&self) -> usize

pub fn panel_width(&self) -> usize

pub fn len<D>(&self, k: D, n: D) -> D

where D: DimLike,

pub fn single_panel_len<D>(&self, k: D) -> D

where D: DimLike,

pub fn single_panel_layout(&self, k: usize, item_size: usize) -> Layout

pub fn pack_tensor( &self, t: &Tensor, k_axis: usize, mn_axis: usize, ) -> Result<Box<dyn MMMInputValue>, Error>

pub fn pack_tensor_view( &self, t: &TensorView<'_>, k_axis: usize, mn_axis: usize, ) -> Result<Box<dyn MMMInputValue>, Error>

pub unsafe fn pack<'a, 'b>( &self, pb: impl BorrowMut<TensorView<'a>>, b: impl Borrow<TensorView<'b>>, k_axis: usize, mn_axis: usize, )

pub unsafe fn pack_t<T>( &self, pb: *mut T, b: *const T, mn: usize, k_stride: isize, mn_stride: isize, k_range: Range<usize>, mn_range: Range<usize>, )

where T: Datum + Copy,

pub unsafe fn pack_segment<'a, 'b>( &self, pb: impl BorrowMut<TensorView<'a>>, b: impl Borrow<TensorView<'b>>, k_axis: usize, mn_axis: usize, k_range: Range<usize>, mn_range: Range<usize>, )

pub fn write_with_k_outer<'p, T>( &self, pb: *mut T, k: usize, mn: usize, ) -> KOutWriter<'p, T>

where T: Copy + Debug,

pub fn write_single_panel_with_k_outer<'p, T>( &self, pb: *mut T, ) -> KOutSinglePanelWriter<'p, T>

where T: Copy + Debug,

pub fn write_with_k_inner<'p, T>( &self, pb: *mut T, k: usize, mn: usize, ) -> KInWriter<'p, T>

where T: Copy + Debug,

Trait Implementations

impl Clone for PackedFormat

fn clone(&self) -> PackedFormat

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PackedFormat

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

Formats the value using the given formatter.

impl Display for PackedFormat

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

Formats the value using the given formatter.

impl Hash for PackedFormat

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

where __H: Hasher,

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 MMMInputFormat for PackedFormat

fn prepare_tensor( &self, t: &Tensor, k_axis: usize, mn_axis: usize, ) -> Result<Tensor, Error>

fn prepare_one( &self, t: &Tensor, k_axis: usize, mn_axis: usize, ) -> Result<Box<dyn MMMInputValue>, Error>

fn precursor(&self) -> WeightType

fn r(&self) -> usize

fn k_alignment(&self) -> usize

fn same_as(&self, other: &(dyn MMMInputFormat + 'static)) -> bool

fn merge_with<'o, 'a, 'b>( &'a self, other: &'b (dyn MMMInputFormat + 'static), ) -> Option<&'o (dyn MMMInputFormat + 'static)>

where 'a: 'o, 'b: 'o,

fn mem_size(&self, k: TDim, mn: TDim) -> TDim

fn extract_at_mn_f16( &self, data: &EagerPackedInput, mn: usize, slice: &mut [f16], ) -> Result<(), Error>

fn extract_at_mn_f32( &self, data: &EagerPackedInput, mn: usize, slice: &mut [f32], ) -> Result<(), Error>

impl PartialEq for PackedFormat

fn eq(&self, other: &PackedFormat) -> 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 PackedFormat

impl StructuralPartialEq for PackedFormat