Enum MetadataTensorMut 

pub enum MetadataTensorMut<'a> {
    I32(&'a mut Tensor<i32>),
    Bool(&'a mut Tensor<u8>),
}

Erased mutable metadata tensor reference.

Examples

use tenferro_device::LogicalMemorySpace;
use tenferro_prims::{MetadataDType, MetadataTensorMut};
use tenferro_tensor::{MemoryOrder, Tensor};

let mut tensor = Tensor::<u8>::zeros(
    &[2, 2],
    LogicalMemorySpace::MainMemory,
    MemoryOrder::ColumnMajor,
);
let metadata = MetadataTensorMut::Bool(&mut tensor);
assert_eq!(metadata.dtype(), MetadataDType::Bool);

Variants

I32(&'a mut Tensor<i32>)

A metadata tensor stored as i32.

Bool(&'a mut Tensor<u8>)

A logical bool/mask metadata tensor backed by u8 for now.

Implementations

impl<'a> MetadataTensorMut<'a>

pub const fn dtype(&self) -> MetadataDType

Return the logical dtype carried by this metadata tensor handle.

Examples

use tenferro_device::LogicalMemorySpace;
use tenferro_prims::{MetadataDType, MetadataTensorMut};
use tenferro_tensor::{MemoryOrder, Tensor};

let mut tensor = Tensor::<i32>::zeros(
    &[1],
    LogicalMemorySpace::MainMemory,
    MemoryOrder::ColumnMajor,
);
let metadata = MetadataTensorMut::I32(&mut tensor);
assert_eq!(metadata.dtype(), MetadataDType::I32);

Trait Implementations

impl<'a> Debug for MetadataTensorMut<'a>

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

Formats the value using the given formatter.