Enum TensorRole 

pub enum TensorRole {
    Embedding,
    LmHead,
    FinalNorm,
    KdaProjection,
    KdaConv,
    KdaDecay,
    KdaGate,
    KdaNorm,
    MlaProjection,
    MlaNorm,
    Indexer,
    IndexerNorm,
    MoeRouter,
    MoeExpert,
    MoeShared,
    DenseFfn,
    LayerNorm,
    HyperConnection,
    MtpProjection,
    MtpNorm,
    Vision,
}

What a tensor is for. Deliberately coarse — Slice 1 proves coverage, not placement.

Variants

Embedding

Token embedding / final norm / lm_head.

LmHead

FinalNorm

KdaProjection

KDA linear-attention mixer (34 of the 45 text layers).

KdaConv

KdaDecay

KdaGate

KdaNorm

MlaProjection

NoPE sparse MLA (11 text layers + the MTP layer).

MlaNorm

Indexer

DSA top-k indexer that fronts each sparse-MLA layer.

IndexerNorm

MoeRouter

MoE router + experts + shared expert.

MoeExpert

MoeShared

DenseFfn

Dense FFN (layers 0..2, first_k_dense_replace = 3).

LayerNorm

Per-layer norms and mHC (hyper-connection) parameters.

HyperConnection

MtpProjection

MTP head at layer 45. 🪤 GLM does NOT use mtp.0.*.

MtpNorm

Vision

Vision tower — present in the checkpoint, out of scope for the text port.

Implementations

impl TensorRole

pub fn is_norm(self) -> bool

Norm-family roles. Used by the RMSNorm sanity pass: a tensor whose name says “norm” must land in one of these, or the classifier is lying about something. (Hazard carried from notavault-atlas: RMSNorm weights have silently corrupted Atlas numbers before.)

pub fn is_text_model(self) -> bool

Text-model roles Atlas must eventually implement. Vision is excluded.

Trait Implementations

impl Clone for TensorRole

fn clone(&self) -> TensorRole

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorRole

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

Formats the value using the given formatter.

impl Ord for TensorRole

fn cmp(&self, other: &TensorRole) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized,

Restrict a value to a certain interval.

impl PartialEq for TensorRole

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl PartialOrd for TensorRole

fn partial_cmp(&self, other: &TensorRole) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

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

Tests less than (for self and other) and is used by the < operator.

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

Tests less than or equal to (for self and other) and is used by the <= operator.

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

Tests greater than (for self and other) and is used by the > operator.

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

Tests greater than or equal to (for self and other) and is used by the >= operator.

impl Copy for TensorRole

impl Eq for TensorRole

impl StructuralPartialEq for TensorRole