Struct LocalModelLayout 

pub struct LocalModelLayout<P = TensorPlacement> { /* private fields */ }

Complete rank-local model geometry produced alongside checkpoint placement.

Implementations

impl<P> LocalModelLayout<P>

pub fn contains(&self, target: &str) -> bool

Returns whether a physical target has already been planned.

pub fn insert(&mut self, target: String, layout: LocalTensorLayout<P>)

Inserts one planner-produced physical layout.

pub fn tensor(&self, target: &str) -> Option<&LocalTensorLayout<P>>

Returns one physical tensor layout by rewritten target name.

pub fn tensors(&self) -> impl Iterator<Item = (&str, &LocalTensorLayout<P>)>

Iterates physical layouts in deterministic target-name order.

pub fn len(&self) -> usize

Returns the number of planned physical tensors.

pub fn is_empty(&self) -> bool

Returns whether no physical tensors were planned.

Trait Implementations

impl<P: Clone> Clone for LocalModelLayout<P>

fn clone(&self) -> LocalModelLayout<P>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<P: Debug> Debug for LocalModelLayout<P>

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

Formats the value using the given formatter.

impl<P> Default for LocalModelLayout<P>

fn default() -> Self

Returns the “default value” for a type.

impl<P: Eq> Eq for LocalModelLayout<P>

impl<P: PartialEq> PartialEq for LocalModelLayout<P>

fn eq(&self, other: &LocalModelLayout<P>) -> bool

Equality operator ==.

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

Inequality operator !=.

impl<P: PartialEq> StructuralPartialEq for LocalModelLayout<P>