Struct kn_cuda_sys::wrapper::descriptor::TensorDescriptor

pub struct TensorDescriptor { /* private fields */ }

Implementations

impl TensorDescriptor

pub fn new_f32(shape: Vec<i32>, strides: Vec<i32>) -> Self

pub fn new(shape: Vec<i32>, strides: Vec<i32>, dtype: cudnnDataType_t) -> Self

pub fn rank(&self) -> usize

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

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

pub unsafe fn inner(&self) -> cudnnTensorDescriptor_t

pub fn size_bytes(&self) -> usize

pub fn has_positive_strides(&self) -> bool

Trait Implementations

impl Debug for TensorDescriptor

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

Formats the value using the given formatter.

impl Drop for TensorDescriptor

fn drop(&mut self)

Executes the destructor for this type.