import { Tensor } from '@tensorflow/tfjs-core';
import { GPTConfig } from '../models/config';
export default class RoPECache {
    private readonly config;
    readonly rotaryDim: number;
    private ropeBase;
    private ropeInvFreq;
    private ropeCos;
    private ropeSin;
    private ropeNegSin;
    private ropeCacheLen;
    constructor(config: GPTConfig);
    ensureRopeCache(needed: number): void;
    getCos(): Tensor | null;
    getSin(): Tensor | null;
    getNegSin(): Tensor | null;
    dispose(): void;
}
