Interface WeightsManifestEntry

An entry in the weight manifest.

The entry contains specification of a weight.

interface WeightsManifestEntry {
    dtype: "string" | "float32" | "int32" | "bool" | "complex64";
    group?: WeightGroup;
    name: string;
    quantization?: {
        dtype: "uint16" | "uint8" | "float16";
        min?: number;
        scale?: number;
    };
    shape: number[];
}

Properties

dtype: "string" | "float32" | "int32" | "bool" | "complex64"

Data type of the weight.

group?: WeightGroup

Type of the weight.

Optional.

The value 'optimizer' indicates the weight belongs to an optimizer (i.e., used only during model training and not during inference).

name: string

Name of the weight, e.g., 'Dense_1/bias'

quantization?: {
    dtype: "uint16" | "uint8" | "float16";
    min?: number;
    scale?: number;
}

Information for dequantization of the weight.

shape: number[]

Shape of the weight.