import type { GraphModel, Tensor } from '@tensorflow/tfjs';
export declare function getSpatialNodeNames(model: GraphModel): {
    name: string;
    layer: number;
}[];
export declare function getLastSpatialNodeName(model: GraphModel): string;
export type SpatialNode = {
    name: string;
    layer: number;
    shape: number[];
};
export declare function getSpatialNodes(args: {
    model: GraphModel;
    tf: {
        zeros: (shape: number[]) => Tensor;
    };
}): SpatialNode[];
export declare function filterSpatialNodesWithUniqueShapes(spatialNodes: SpatialNode[]): SpatialNode[];
export declare function getSpatialNodesWithUniqueShapes(args: {
    model: GraphModel;
    tf: {
        zeros: (shape: number[]) => Tensor;
    };
}): SpatialNode[];