module ptwm.classify

class AuditEntry:

def __init__(self, name: str, role: TensorRole, source: str, pattern: str) -> None

name: str

pattern: str

role: TensorRole

source: str

class AuditLog:

def __init__(self) -> None

def discovered_chains(self) -> tuple[DiscoveredChain, ...]

def entries(self) -> tuple[AuditEntry, ...]

def from_cbor(cls, blob: bytes) -> AuditLog

def record(self, name: str, c: TensorClassification) -> None

def record_discovered(self, chain_bytes: bytes, dtype_code: int, role: TensorRole, n_candidates_tried: int, sample_tensor_name: str) -> None

def to_cbor(self) -> bytes

class ClassifierChain:

def __init__(self, classifiers: Sequence[TensorClassifier]) -> None

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification

class DiscoveredChain:

def __init__(self, chain: bytes, dtype_code: int, role: str, n_candidates_tried: int, sample_tensor_name: str) -> None

chain: bytes

dtype_code: int

n_candidates_tried: int

role: str

sample_tensor_name: str

class ExplicitFlagsClassifier:

def __init__(self, rules: list[tuple[str, TensorRole]]) -> None

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification | None

def from_flag_strings(cls, flags: Sequence[str]) -> ExplicitFlagsClassifier

class HeuristicClassifier:

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification | None

class HfQuantConfigClassifier:

def __init__(self, quant_algo: str, exclude_modules: tuple[str, ...]) -> None

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification | None

def from_path(cls, path: str | Path) -> HfQuantConfigClassifier

class PtwmConfigClassifier:

def __init__(self, rules: list[tuple[str, TensorRole]]) -> None

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification | None

def from_path(cls, path: str | Path) -> PtwmConfigClassifier

class TensorClassification:

def __init__(self, role: TensorRole, source: str, pattern: str) -> None

pattern: str

role: TensorRole

source: str

class TensorClassifier(Protocol):

def classify(self, name: str, dtype: str, shape: tuple[int, ...], archive_keys: tuple[str, ...]) -> TensorClassification | None

class TensorRole(StrEnum):

PACKED_VALUES = 'packed_values'

SCALE_BLOCK = 'scale_block'

SCALE_GLOBAL = 'scale_global'

STANDARD = 'standard'