Struct nncombinator::cuda::kernel::activation::ReLuBackward

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

Implementations

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

pub fn new() -> ReLuBackward<T>

Create a ReLuBackward instance

Trait Implementations

impl Kernel for ReLuBackward<f32>

const FUNC_PTR: *const c_void = {alloc140: *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 ReLuBackward<f64>

const FUNC_PTR: *const c_void = {alloc144: *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