import { MemoryInfo } from '@tensorflow/tfjs-core';
export interface ExtendedMemoryInfo extends MemoryInfo {
    numBytesInGPUAllocated?: number;
    numBytesAllocatedInGPU?: number;
}
export default class MemoryProfiler {
    private log;
    private maxMemory;
    private maxLabel?;
    private lastMemInfo;
    private peakMemory;
    startMemory(): void;
    getPeakMemory(): number;
    getMaxMemory(): number;
    endMemory(label: string): void;
    printSummary(): void;
}
