import { TensorInfo } from '@tensorflow/tfjs-core';
import { MathBackendWebGL } from '@tensorflow/tfjs-backend-webgl';
export declare const MATMUL_SHARED_DIM_THRESHOLD = 1000;
interface BatchMatMulConfig {
    a: TensorInfo;
    b: TensorInfo;
    transposeA: boolean;
    transposeB: boolean;
    backend: MathBackendWebGL;
    activationSnippet?: string;
    multiplier?: TensorInfo;
}
export declare function batchMatMulGeluImpl({ a, b, transposeA, transposeB, backend, activationSnippet, multiplier, }: BatchMatMulConfig): TensorInfo;
export declare function batchMatMulKernel(args: {
    inputs: {
        x: TensorInfo;
        kernel: TensorInfo;
    };
    backend: MathBackendWebGL;
}): TensorInfo;
export {};
