Trait burn_jit::JitElement

pub trait JitElement: Element + Debug + Send + Sync + 'static + Clone + Pod
where
    Self: Sized,
{
    // Required methods
    fn type_name() -> &'static str;
    fn as_bytes(slice: &[Self]) -> &[u8] ;
    fn from_bytes(bytes: &[u8]) -> &[Self];
    fn gpu_elem() -> Elem;
    fn maximum_value() -> Self;
    fn minimum_value() -> Self;
}

The base element trait for the jit backend.

Required Methods

fn type_name() -> &'static str

TODO: Remove when all wgsl static kernels are migrated.

fn as_bytes(slice: &[Self]) -> &[u8]

Convert a slice of elements to a slice of bytes.

fn from_bytes(bytes: &[u8]) -> &[Self]

Convert a slice of bytes to a slice of elements.

fn gpu_elem() -> Elem

Element representation for gpu.

fn maximum_value() -> Self

Highest possible value

fn minimum_value() -> Self

Lowest possible value

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl JitElement for f32

fn type_name() -> &'static str

fn as_bytes(slice: &[Self]) -> &[u8]

fn from_bytes(bytes: &[u8]) -> &[Self]

fn gpu_elem() -> Elem

fn maximum_value() -> Self

fn minimum_value() -> Self

impl JitElement for i32

fn type_name() -> &'static str

fn as_bytes(slice: &[Self]) -> &[u8]

fn from_bytes(bytes: &[u8]) -> &[Self]

fn gpu_elem() -> Elem

fn maximum_value() -> Self

fn minimum_value() -> Self

impl JitElement for u32

fn type_name() -> &'static str

fn as_bytes(slice: &[Self]) -> &[u8]

fn from_bytes(bytes: &[u8]) -> &[Self]

fn gpu_elem() -> Elem

fn maximum_value() -> Self

fn minimum_value() -> Self

Implementors