import { DataType, TensorInfo } from '@tensorflow/tfjs-core';
import { WebGPUBackend } from '@tensorflow/tfjs-backend-webgpu/dist/webgpu';
import { WebGPUProgram } from './webgpu_program';
type ProgramUniform = {
    type: string;
    data: number[];
}[];
interface ExtendedAdapterInfo extends GPUAdapterInfo {
    subgroupMaxSize?: number;
    subgroupMinSize?: number;
}
export default class WebGPUBackendPatch extends WebGPUBackend {
    readonly subgroupMaxSize: number;
    readonly subgroupMinSize: number;
    constructor(device: GPUDevice, adapterInfo?: ExtendedAdapterInfo);
    runWebGPUProgram(program: WebGPUProgram, inputs: TensorInfo[], outputDtype: DataType, programDefinedUniform?: ProgramUniform, output?: TensorInfo): TensorInfo;
}
export {};
