Struct TensorParallelGroupedOutput 

pub struct TensorParallelGroupedOutput<T> { /* private fields */ }

Rank-local grouped output split around the tensor-parallel reduction.

Implementations

impl<T> TensorParallelGroupedOutput<T>

pub fn new(reducible: T, post_reduce: Option<T>) -> Self

Creates one rank-local grouped projection output.

pub const fn reducible(&self) -> &T

Returns the rank-local reducible tensor.

pub const fn post_reduce(&self) -> Option<&T>

Returns the optional post-reduction term.

pub fn into_parts(self) -> (T, Option<T>)

Consumes the output into its reduction contribution and post-reduction term.

Trait Implementations

impl<T: Clone> Clone for TensorParallelGroupedOutput<T>

fn clone(&self) -> TensorParallelGroupedOutput<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug> Debug for TensorParallelGroupedOutput<T>

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

Formats the value using the given formatter.