pub struct TensorDescriptor<'comm> { /* private fields */ }Implementations§
Source§impl<'comm> TensorDescriptor<'comm>
impl<'comm> TensorDescriptor<'comm>
pub fn create_for<T: DataTypeLike>( context: &Context<'comm>, shape: &[u64], ranks_per_mode: &[u64], rank_count: u32, ) -> Result<Self>
pub fn create( context: &Context<'comm>, config: TensorDescriptorConfig<'_>, ) -> Result<Self>
pub fn rank(&self) -> u32
pub fn shape(&self) -> &[u64]
pub fn element_stride(&self) -> Option<&[u64]>
pub fn block_size(&self) -> Option<&[u64]>
pub fn block_stride(&self) -> Option<&[u64]>
pub fn ranks_per_mode(&self) -> &[u64]
pub fn ranks(&self) -> Option<&[i32]>
pub fn rank_count(&self) -> u32
pub fn data_type(&self) -> DataType
pub const fn as_raw(&self) -> cutensorMpTensorDescriptor_t
Trait Implementations§
Source§impl<'comm> Debug for TensorDescriptor<'comm>
impl<'comm> Debug for TensorDescriptor<'comm>
Source§impl Drop for TensorDescriptor<'_>
impl Drop for TensorDescriptor<'_>
Auto Trait Implementations§
impl<'comm> Freeze for TensorDescriptor<'comm>
impl<'comm> RefUnwindSafe for TensorDescriptor<'comm>
impl<'comm> !Send for TensorDescriptor<'comm>
impl<'comm> !Sync for TensorDescriptor<'comm>
impl<'comm> Unpin for TensorDescriptor<'comm>
impl<'comm> UnsafeUnpin for TensorDescriptor<'comm>
impl<'comm> UnwindSafe for TensorDescriptor<'comm>
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more