Trait burn_wgpu::kernel::StaticKernelSource

pub trait StaticKernelSource: Send + 'static + Sync {
    // Required method
    fn source() -> SourceTemplate;
}

Static wgpu kernel to create a source template.

Required Methods

fn source() -> SourceTemplate

Source template for the kernel.

Object Safety

This trait is not object safe.

Implementors

impl StaticKernelSource for Conv2d

impl StaticKernelSource for ConvTranspose2d

impl StaticKernelSource for MatmulMemCoalescingRaw

impl StaticKernelSource for MatmulNaiveRaw

impl StaticKernelSource for MatmulTiling2DUnpaddedRaw

impl StaticKernelSource for MatmulTiling2Dvec4Raw

impl StaticKernelSource for MatmulTiling2DVec4LhsRaw

impl StaticKernelSource for AvgPool2dBackwardRaw

impl StaticKernelSource for AvgPool2dRaw

impl StaticKernelSource for MaxPool2d

impl StaticKernelSource for MaxPool2dWithIndices

impl StaticKernelSource for MaxPool2dWithIndicesBackward

impl StaticKernelSource for RecursiveSumRaw

impl StaticKernelSource for ReductionArgsRaw

impl StaticKernelSource for ReductionDimRaw

impl StaticKernelSource for ReductionDimSharedMemoryRaw

impl StaticKernelSource for CastRaw

impl StaticKernelSource for ContiguousRaw

impl<K: StaticKernelSource, E: WgpuElement, I: WgpuElement, const WORKGROUP_X_SIZE: usize, const WORKGROUP_Y_SIZE: usize, const WORKGROUP_Z_SIZE: usize> StaticKernelSource for KernelSettings<K, E, I, WORKGROUP_X_SIZE, WORKGROUP_Y_SIZE, WORKGROUP_Z_SIZE>