Struct MetalStorage

pub struct MetalStorage;

Trait Implementations

impl BackendStorage for MetalStorage

type Device = MetalDevice

fn try_clone(&self, _: &Layout) -> Result<Self>

fn dtype(&self) -> DType

fn device(&self) -> &Self::Device

fn to_cpu_storage(&self) -> Result<CpuStorage>

fn affine(&self, _: &Layout, _: f64, _: f64) -> Result<Self>

fn powf(&self, _: &Layout, _: f64) -> Result<Self>

fn elu(&self, _: &Layout, _: f64) -> Result<Self>

fn reduce_op(&self, _: ReduceOp, _: &Layout, _: &[usize]) -> Result<Self>

fn cmp(&self, _: CmpOp, _: &Self, _: &Layout, _: &Layout) -> Result<Self>

fn to_dtype(&self, _: &Layout, _: DType) -> Result<Self>

fn unary_impl<B: UnaryOpT>(&self, _: &Layout) -> Result<Self>

fn binary_impl<B: BinaryOpT>( &self, _: &Self, _: &Layout, _: &Layout, ) -> Result<Self>

fn where_cond( &self, _: &Layout, _: &Self, _: &Layout, _: &Self, _: &Layout, ) -> Result<Self>

fn conv1d( &self, _: &Layout, _: &Self, _: &Layout, _: &ParamsConv1D, ) -> Result<Self>

fn conv_transpose1d( &self, _l: &Layout, _kernel: &Self, _kernel_l: &Layout, _params: &ParamsConvTranspose1D, ) -> Result<Self>

fn conv2d( &self, _: &Layout, _: &Self, _: &Layout, _: &ParamsConv2D, ) -> Result<Self>

fn conv_transpose2d( &self, _l: &Layout, _kernel: &Self, _kernel_l: &Layout, _params: &ParamsConvTranspose2D, ) -> Result<Self>

fn index_select( &self, _: &Self, _: &Layout, _: &Layout, _: usize, ) -> Result<Self>

fn gather(&self, _: &Layout, _: &Self, _: &Layout, _: usize) -> Result<Self>

fn scatter_add( &self, _: &Layout, _: &Self, _: &Layout, _: &Self, _: &Layout, _: usize, ) -> Result<Self>

fn index_add( &self, _: &Layout, _: &Self, _: &Layout, _: &Self, _: &Layout, _: usize, ) -> Result<Self>

fn matmul( &self, _: &Self, _: (usize, usize, usize, usize), _: &Layout, _: &Layout, ) -> Result<Self>

fn copy_strided_src(&self, _: &mut Self, _: usize, _: &Layout) -> Result<()>

fn copy2d( &self, _: &mut Self, _: usize, _: usize, _: usize, _: usize, _: usize, _: usize, ) -> Result<()>

fn avg_pool2d( &self, _: &Layout, _: (usize, usize), _: (usize, usize), ) -> Result<Self>

fn max_pool2d( &self, _: &Layout, _: (usize, usize), _: (usize, usize), ) -> Result<Self>

fn upsample_nearest1d(&self, _: &Layout, _: usize) -> Result<Self>

fn upsample_nearest2d(&self, _: &Layout, _: usize, _: usize) -> Result<Self>

impl Debug for MetalStorage

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.