Trait DistributedRegistration 

pub trait DistributedRegistration: Send + Sync {
    // Required method
    fn on_register(
        &mut self,
        node_id: &NodeId,
        container: &mut TensorContainer<u64>,
    );
}

Trait for registering distributed gradients.

Required Methods

fn on_register( &mut self, node_id: &NodeId, container: &mut TensorContainer<u64>, )

Performs distributed registration operations on the tensor with the corresponding NodeId.

Dyn Compatibility

This trait is dyn compatible.

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

Implementors