Struct cubecl_core::frontend::TensorHandleRef

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

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

Fields

handle: &'a Handle<R::Server>

strides: &'a [usize]

shape: &'a [usize]

Implementations

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<R::Server>, strides: &'a [usize], shape: &'a [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.