Trait DistributedLayer 

pub trait DistributedLayer: Module {
    type Input;
    type Output;

    // Required method
    fn forward(&self, input: Self::Input) -> Self::Output;
}

One layer of a model split by LayerParallelism. It runs on the device its placement gives it, and what it returns moves to the next layer’s device.

Required Associated Types

type Input

What the layer takes, already on its device.

type Output

What the layer returns, left on its device.

Required Methods

fn forward(&self, input: Self::Input) -> Self::Output

Run the layer.

Dyn Compatibility

This trait is not dyn compatible.

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

Implementors