Skip to main content

parallel_step_array1

optirs_core::parallel_optimizer

Function parallel_step_array1 

Source 
pub fn parallel_step_array1<O, A>(
    optimizer: &mut O,
    params_list: &[Array1<A>],
    grads_list: &[Array1<A>],
) -> Result<Vec<Array1<A>>>
where
    O: Optimizer<A, Ix1> + Clone + Send + Sync,
    A: Float + ScalarOperand + Debug + Send + Sync,
Expand description

Multi-group processing for Array1 specifically (optimized path)

See parallel_step for the state and parallelism semantics.