import { Tensor, Variable } from '@tensorflow/tfjs-core';
export default class WeightStore {
    private _variables;
    private touchedVariables;
    onWeightRead?: (name: string, variable: Variable) => Tensor;
    saveWeights(map: Map<string, Tensor[]>): void;
    loadWeights(weights: Map<string, Tensor[]>, reference: boolean, trainable?: boolean): void;
    addVariable(name: string, variable?: Variable): void;
    deleteVariable(name: string): void;
    get variables(): Variable[];
    get variableNames(): string[];
    get trainableVariables(): Variable[];
    getRawVariable(name: string): Variable;
    getVariable(name: string): Tensor;
    setTrainable(names: string[]): void;
    hasVariable(name: string): boolean;
    setVariable(name: string, variable: Variable): void;
    touchVariables(names: string[]): void;
    dispose(): void;
}
