Enum ReduceOp 

pub enum ReduceOp {
    Sum,
    Product,
    Min,
    Max,
    Average,
}

Reduction operation for collective communication.

Variants

Sum

Sum all values.

Product

Compute product of all values.

Min

Find minimum value.

Max

Find maximum value.

Average

Compute average of all values.

Implementations

impl ReduceOp

pub fn apply_f32(&self, a: f32, b: f32) -> f32

Applies the reduction operation to two f32 values.

pub fn reduce_slices(&self, slices: &[Vec<f32>]) -> Vec<f32>

Applies the reduction operation to slices.

Trait Implementations

impl Clone for ReduceOp

fn clone(&self) -> ReduceOp

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ReduceOp

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for ReduceOp

fn eq(&self, other: &ReduceOp) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for ReduceOp

impl Eq for ReduceOp

impl StructuralPartialEq for ReduceOp