Struct Adagrad

pub struct Adagrad<'a, T> { /* private fields */ }

Implementation Adagrad optimizer

Implementations

impl<'a, T> Adagrad<'a, T>

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

Create a Adagrad optimizer instance

Trait Implementations

impl<'a> Kernel for Adagrad<'a, f32>

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

Pointer to cuda kernel function

type Args = AdagradArgs<'a, 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

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> Kernel for Adagrad<'a, f64>

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

Pointer to cuda kernel function

type Args = AdagradArgs<'a, 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

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

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