Trait ReplicatedTensorCollective 

pub trait ReplicatedTensorCollective<B>: TensorCollective<B>
where
    B: Backend,
{
    // Required method
    fn all_reduce_sum(
        &self,
        value: <B as BackendTypes>::FloatTensorPrimitive,
    ) -> Result<<B as BackendTypes>::FloatTensorPrimitive, Self::Error>;
}

Replicated reductions in addition to leading-axis shard collectives.

Required Methods

fn all_reduce_sum( &self, value: <B as BackendTypes>::FloatTensorPrimitive, ) -> Result<<B as BackendTypes>::FloatTensorPrimitive, Self::Error>

Sum corresponding tensor elements across ranks, retaining shape and dtype.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementors