Skip to main content

segmented_projection_group

Function segmented_projection_group 

Source
pub fn segmented_projection_group<T, M>(
    logical_name: impl Into<String>,
    role: ParameterRole,
    fused: &M,
    row: &M,
    segments: Vec<Range<usize>>,
    preferred_units: usize,
) -> Result<ParameterGroupSpec, ParallelPlanError>
where T: Tensor, M: Parameterized<T>,
Expand description

Describes a component-major fused column projection and its row-parallel output as one shared logical partition.

The same ordered segment selection is attached to the fused weight and all encoding companions exposed by the module. The row projection consumes the corresponding local hidden partition and is reduced once by the backend.