Trait WithDType 

pub trait WithDType:
    Copy
    + Clone
    + 'static
    + Zero
    + One {
    const DTYPE: DType;

    // Required methods
    fn to_cpu_storage(data: &[Self]) -> CpuStorageRef<'_>;
    fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[Self], Error>;
    fn to_cpu_storage_mut(data: &mut [Self]) -> CpuStorageRefMut<'_>;
    fn from_cpu_storage_mut(
        data: CpuStorageRefMut<'_>,
    ) -> Result<&mut [Self], Error>;
}

Required Associated Constants

const DTYPE: DType

Required Methods

fn to_cpu_storage(data: &[Self]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[Self], Error>

fn to_cpu_storage_mut(data: &mut [Self]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut( data: CpuStorageRefMut<'_>, ) -> Result<&mut [Self], Error>

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types

impl WithDType for f32

const DTYPE: DType = DType::F32

fn to_cpu_storage_mut(data: &mut [f32]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut(data: CpuStorageRefMut<'_>) -> Result<&mut [f32], Error>

fn to_cpu_storage(data: &[f32]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[f32], Error>

impl WithDType for i32

const DTYPE: DType = DType::I32

fn to_cpu_storage_mut(data: &mut [i32]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut(data: CpuStorageRefMut<'_>) -> Result<&mut [i32], Error>

fn to_cpu_storage(data: &[i32]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[i32], Error>

impl WithDType for i64

const DTYPE: DType = DType::I64

fn to_cpu_storage_mut(data: &mut [i64]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut(data: CpuStorageRefMut<'_>) -> Result<&mut [i64], Error>

fn to_cpu_storage(data: &[i64]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[i64], Error>

impl WithDType for bf16

const DTYPE: DType = DType::BF16

fn to_cpu_storage_mut(data: &mut [bf16]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut( data: CpuStorageRefMut<'_>, ) -> Result<&mut [bf16], Error>

fn to_cpu_storage(data: &[bf16]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[bf16], Error>

impl WithDType for f16

const DTYPE: DType = DType::F16

fn to_cpu_storage_mut(data: &mut [f16]) -> CpuStorageRefMut<'_>

fn from_cpu_storage_mut(data: CpuStorageRefMut<'_>) -> Result<&mut [f16], Error>

fn to_cpu_storage(data: &[f16]) -> CpuStorageRef<'_>

fn from_cpu_storage(data: CpuStorageRef<'_>) -> Result<&[f16], Error>

Implementors