Struct nncombinator::cuda::kernel::activation::SwishBackward

pub struct SwishBackward<T>where
    T: DataTypeInfo,{ /* private fields */ }

Swish activation function implementation

Implementations

impl<T> SwishBackward<T>where
    T: DataTypeInfo,

pub fn new() -> SwishBackward<T>

Create a SwishBackward instance

Trait Implementations

impl Kernel for SwishBackward<f32>

const FUNC_PTR: *const c_void = {alloc188: *const libc::c_void}

Pointer to cuda kernel function

type Args = ActivationBackwardArgs<f32>

Object to be converted into a list of arguments to be passed to the cuda kernel function

fn launch(
    &mut self,
    grid_dim: dim3,
    block_dim: dim3,
    args: &mut Self::Args,
    shared_mem: usize
) -> Result<(), CudaRuntimeError>

cuda kernel startup function

impl Kernel for SwishBackward<f64>

const FUNC_PTR: *const c_void = {alloc192: *const libc::c_void}

Pointer to cuda kernel function

type Args = ActivationBackwardArgs<f64>

Object to be converted into a list of arguments to be passed to the cuda kernel function

fn launch(
    &mut self,
    grid_dim: dim3,
    block_dim: dim3,
    args: &mut Self::Args,
    shared_mem: usize
) -> Result<(), CudaRuntimeError>

cuda kernel startup function