Struct ParameterMemberSpec 

pub struct ParameterMemberSpec { /* private fields */ }

One physical tensor belonging to a logical parameter group.

Implementations

impl ParameterMemberSpec

pub fn new( target: impl Into<String>, global_shape: impl Into<Vec<usize>>, sharding: MemberSharding, ) -> Self

Creates a member with an exact pre-selection checkpoint shape.

pub fn target(&self) -> &str

Returns the rewritten checkpoint target.

pub fn global_shape(&self) -> &[usize]

Returns the complete source shape.

pub const fn sharding(&self) -> &MemberSharding

Returns the requested rank-local selection.

pub const fn linear_companion(&self) -> Option<LinearCompanionRole>

Returns this member’s encoded-linear companion role, when present.

pub fn linear_companion_of(&self) -> Option<&str>

Returns the primary linear weight owning this companion.

Trait Implementations

impl Clone for ParameterMemberSpec

fn clone(&self) -> ParameterMemberSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ParameterMemberSpec

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

Formats the value using the given formatter.

impl Eq for ParameterMemberSpec

impl PartialEq for ParameterMemberSpec

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for ParameterMemberSpec