Struct Layer

pub struct Layer {
    pub elem2idx: Vec<usize>,
    pub idx2vtx: Vec<usize>,
}

Fields

elem2idx: Vec<usize>

idx2vtx: Vec<usize>

Trait Implementations

impl CustomOp1 for Layer

fn bwd( &self, vtx2xy: &Tensor, _elem2cog: &Tensor, dw_elem2cog: &Tensor, ) -> Result<Option<Tensor>>

This function takes as argument the argument arg used in the forward pass, the result produced by the forward operation res and the gradient of the result grad_res. The function should return the gradient of the argument.

fn name(&self) -> &'static str

fn cpu_fwd( &self, storage: &CpuStorage, layout: &Layout, ) -> Result<(CpuStorage, Shape)>

The forward pass, as run on a cpu device. Note that the storage can use arbitrary strides, offsets etc so the associated layout should be used to access it.

fn cuda_fwd( &self, _storage: &CudaStorage, _layout: &Layout, ) -> Result<(CudaStorage, Shape), Error>

The forward pass, as run on a gpu device. Note that the storage can use arbitrary strides, offsets etc so the associated layout should be used to access it.

fn metal_fwd( &self, _storage: &MetalStorage, _layout: &Layout, ) -> Result<(MetalStorage, Shape), Error>

The forward pass, as run on a metal gpu device. Note that the storage can use arbitrary strides, offsets etc so the associated layout should be used to access it.