Trait TensorElement 

pub trait TensorElement: Element + Wire {
    const ELEMENT_TYPE: ElementType;
}

Storage types supported by the shared collective device adapter.

The sealed implementations preserve F32/F16/BF16/I32 storage bits on the little-endian rank transport; they do not convert half values through F32.

Required Associated Constants

const ELEMENT_TYPE: ElementType

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementations on Foreign Types

impl TensorElement for bf16

const ELEMENT_TYPE: ElementType = ElementType::BF16

impl TensorElement for f16

const ELEMENT_TYPE: ElementType = ElementType::F16

impl TensorElement for f32

const ELEMENT_TYPE: ElementType = ElementType::F32

impl TensorElement for i32

const ELEMENT_TYPE: ElementType = ElementType::I32

Implementors