Trait TensorType

pub trait TensorType: Sealed {
    type H;

    // Required methods
    fn owned() -> bool;
    fn host() -> bool;

    // Provided methods
    fn borrowed() -> bool { ... }
    fn device() -> bool { ... }
}

A trait to represent the type of a tensor

Required Associated Types

type H

The halide type of the tensor

Required Methods

fn owned() -> bool

Check if the tensor is owned

fn host() -> bool

Check if the tensor is allocated in the host

Provided Methods

fn borrowed() -> bool

Check if the tensor is borrowed

fn device() -> bool

Check if the tensor is allocated in the device

Object Safety

This trait is not object safe.

Implementors

impl<H: HalideType> TensorType for Device<H>

type H = H

impl<H: HalideType> TensorType for Host<H>

type H = H

impl<T: TensorType> TensorType for Ref<'_, T>

type H = <T as TensorType>::H

impl<T: TensorType> TensorType for RefMut<'_, T>

type H = <T as TensorType>::H