Trait dfdx::devices::HasDevice

source · [−]

pub trait HasDevice: HasArrayType {
    type Device: Device<Self::Array>;
}

Expand description

A crate::arrays::HasArrayType that has a Device for its crate::arrays::HasArrayType::Array

Required Associated Types

type Device: Device<Self::Array>

Implementors

impl<H> HasDevice for Tensor0D<H>

type Device = Cpu

impl<T: HasDevice> HasDevice for PhantomTensor<T>

type Device = <T as HasDevice>::Device

impl<const M: usize, H> HasDevice for Tensor1D<M, H>

type Device = Cpu

impl<const M: usize, const N: usize, H> HasDevice for Tensor2D<M, N, H>

type Device = Cpu

impl<const M: usize, const N: usize, const O: usize, H> HasDevice for Tensor3D<M, N, O, H>

type Device = Cpu

impl<const M: usize, const N: usize, const O: usize, const P: usize, H> HasDevice for Tensor4D<M, N, O, P, H>

type Device = Cpu