Struct kn_cuda_eval::shape::StridedShape

pub struct StridedShape { /* private fields */ }

Implementations

impl StridedShape

pub fn new(shape: Vec<usize>, strides: Vec<isize>) -> Self

pub fn new_simple(shape: Vec<usize>) -> Self

pub fn shape(&self) -> &[usize]

pub fn strides(&self) -> &[isize]

pub fn rank(&self) -> usize

pub fn has_simple_strides(&self) -> bool

pub fn has_dense_strides(&self) -> bool

pub fn visit_strided_indices(&self, f: impl FnMut(isize))

pub fn size(&self) -> usize

pub fn slice(&self, axis: usize, range: SliceRange) -> StridedShape

pub fn flip(&self, axis: usize) -> StridedShape

pub fn broadcast(&self, new_shape: Vec<usize>) -> StridedShape

pub fn view(&self, new_shape: Vec<usize>) -> Result<StridedShape, ViewError>

pub fn permute(&self, permutation: &[usize]) -> StridedShape

pub fn repeat_unary(&self, axis: usize, count: usize) -> StridedShape

pub fn descriptor(&self, dtype: cudnnDataType_t) -> TensorDescriptor

pub fn filter_descriptor(&self, dtype: cudnnDataType_t) -> FilterDescriptor

pub fn matrix_layout(&self) -> MatrixLayout

pub fn remove(&self, axis: usize) -> StridedShape

Trait Implementations

impl Clone for StridedShape

fn clone(&self) -> StridedShape

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for StridedShape

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

Formats the value using the given formatter.

impl PartialEq for StridedShape

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

This method tests for self and other values to be equal, and is used by ==.

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

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

impl Eq for StridedShape

impl StructuralEq for StridedShape

impl StructuralPartialEq for StridedShape