import { WebGPUProgram } from '@tensorflow/tfjs-backend-webgpu';
export default class AttentionMaskProgram32 implements WebGPUProgram {
    variableNames: string[];
    outputShape: number[];
    shaderKey: string;
    dispatchLayout: {
        x: number[];
    };
    dispatch: [number, number, number];
    uniforms: string;
    workgroupSize: [number, number, number];
    size: boolean;
    hs: number;
    nh: number;
    T1: number;
    T2: number;
    constructor(batch: number, nh: number, T1: number, T2: number, hs: number);
    getUserCode(): string;
}
