Struct LinearBatchCrossEntropyMulticlass

pub struct LinearBatchCrossEntropyMulticlass<'a, T, const N: usize>
where
    T: DataTypeInfo + UnitValue<T>,
{ /* private fields */ }

Implementations

impl<'a, T, const N: usize> LinearBatchCrossEntropyMulticlass<'a, T, N>

where T: DataTypeInfo + UnitValue<T>,

pub fn new() -> LinearBatchCrossEntropyMulticlass<'a, T, N>

Create a LinearBatchCrossEntropyMulticlass instance

Trait Implementations

impl<'a, const N: usize> Kernel for LinearBatchCrossEntropyMulticlass<'a, f32, N>

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

Pointer to cuda kernel function

type Args = LinearBatchCrossEntropyMulticlassArgs<'a, f32, N>

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

fn device_synchronize(&self) -> Result<(), CudaRuntimeError>

Function that waits for the completion of the execution of the process passed to the Cuda kernel

impl<'a, const N: usize> Kernel for LinearBatchCrossEntropyMulticlass<'a, f64, N>

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

Pointer to cuda kernel function

type Args = LinearBatchCrossEntropyMulticlassArgs<'a, f64, N>

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

fn device_synchronize(&self) -> Result<(), CudaRuntimeError>

Function that waits for the completion of the execution of the process passed to the Cuda kernel