Struct CubeFusionHandle

pub struct CubeFusionHandle<R: Runtime> {
    pub client: ComputeClient<R::Server, R::Channel>,
    pub handle: Handle,
    pub device: R::Device,
    pub dtype: DType,
    pub strides: Vec<usize>,
}

Handle to be used when fusing operations.

Fields

client: ComputeClient<R::Server, R::Channel>

Compute client for jit.

handle: Handle

The buffer where the data are stored.

device: R::Device

The device of the current tensor.

dtype: DType

The element type of the tensor.

strides: Vec<usize>

The strides of the tensor.

Implementations

impl<R: Runtime> CubeFusionHandle<R>

pub fn as_handle_ref<'a>(&'a self, shape: &'a [usize]) -> TensorHandleRef<'a, R>

Return the reference to a tensor handle.

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

Return the reference to a tensor argument.

Trait Implementations

impl<R: Runtime> Clone for CubeFusionHandle<R>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<R: Runtime> Debug for CubeFusionHandle<R>

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

Formats the value using the given formatter.

impl<R: Runtime> Send for CubeFusionHandle<R>

impl<R: Runtime> Sync for CubeFusionHandle<R>