Struct nncombinator::cuda::kernel::activation::SoftMaxBackward

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

Implementations

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

pub fn new() -> SoftMaxBackward<T>

Create a SoftMaxForward instance

Trait Implementations

impl Kernel for SoftMaxBackward<f32>

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

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