Struct TensorHandleRef

pub struct TensorHandleRef<'a, R: Runtime> {
    pub handle: &'a Handle,
    pub strides: &'a [usize],
    pub shape: &'a [usize],
    pub elem_size: usize,
    pub runtime: PhantomData<R>,
}

Tensor representation with a reference to the server handle, the strides and the shape.

Fields

handle: &'a Handle

strides: &'a [usize]

shape: &'a [usize]

elem_size: usize

runtime: PhantomData<R>

Implementations

impl<R: Runtime> TensorHandleRef<'_, R>

pub fn size(&self) -> usize

impl<'a, R: Runtime> TensorHandleRef<'a, R>

pub fn as_tensor_arg(&'a self, vectorisation: u8) -> TensorArg<'a, R>

Convert the handle into a tensor argument.

pub unsafe fn from_raw_parts( handle: &'a Handle, strides: &'a [usize], shape: &'a [usize], elem_size: usize, ) -> Self

Create a handle from raw parts.

Safety

If you provide wrong strides or shapes, it might create undefined behavior caused by out-of-bounds reads and writes.

Trait Implementations

impl<R: Runtime> Debug for TensorHandleRef<'_, R>

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

Formats the value using the given formatter.