GpuOptimizer

Trait GpuOptimizer 

Source
pub trait GpuOptimizer<A: Float, D: Dimension> {
    // Required methods
    fn is_gpu_available(&self) -> bool;
    fn to_gpu(&mut self) -> Result<(), GpuOptimError>;
    fn to_cpu(&mut self) -> Result<(), GpuOptimError>;
    fn step_gpu(
        &mut self,
        params: &mut Array<A, D>,
        gradients: &Array<A, D>,
    ) -> Result<(), GpuOptimError>;
}
Expand description

Trait for GPU-accelerated optimizers

Required Methods§

Source

fn is_gpu_available(&self) -> bool

Check if GPU acceleration is available

Source

fn to_gpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state to GPU

Source

fn to_cpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state back to CPU

Source

fn step_gpu( &mut self, params: &mut Array<A, D>, gradients: &Array<A, D>, ) -> Result<(), GpuOptimError>

Perform optimization step on GPU

Implementors§