Enum TensorPlacement 

pub enum TensorPlacement {
    Replicated,
    Local,
    Omit,
    Rank {
        rank: usize,
    },
    Shard {
        axis: usize,
        index: usize,
        parts: usize,
    },
    Range {
        axis: usize,
        start: usize,
        end: usize,
    },
    Indices {
        axis: usize,
        indices: Vec<usize>,
    },
}

Backend-neutral placement decision for one physical tensor.

Variants

Replicated

Materialize the complete tensor on every rank.

Local

Materialize the complete tensor on this rank.

Omit

Intentionally omit this tensor on this rank.

Rank

Materialize the complete tensor only on one global rank.

Fields

rank: usize

Owning global rank.

Shard

Materialize an equal contiguous source-tensor slice.

Fields

axis: usize

Source tensor axis being sharded.

index: usize

Shard index.

parts: usize

Total shard count.

Range

Materialize an explicit contiguous source-tensor range.

Fields

axis: usize

Source tensor axis being sliced.

start: usize

Inclusive element offset on axis.

end: usize

Exclusive element offset on axis.

Indices

Materialize selected source-tensor indices in the supplied order.

Fields

axis: usize

Source tensor axis being selected.

indices: Vec<usize>

Distinct source indices in local output order.

Trait Implementations

impl Clone for TensorPlacement

fn clone(&self) -> TensorPlacement

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorPlacement

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

Formats the value using the given formatter.

impl Eq for TensorPlacement

impl PartialEq for TensorPlacement

fn eq(&self, other: &TensorPlacement) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for TensorPlacement