import { DTypeGpu, GPUTensorConstructor, GPUTensorI } from '../../../tensor/gpu/interface';
import { GPUMemoryAllocator } from '../../../tensor/gpu/memory';
import { Input, Operation } from '../../../ops/gpu/operation';
import { Dispatcher } from '../../../ops/gpu/dispatcher';
export interface UpdateValueInfo {
    shapeValue?: readonly number[];
    widthValue?: number;
    heightValue?: number;
    shapeMoments?: readonly number[];
    widthMoments?: number;
    heightMoments?: number;
    shapeOutput?: readonly number[];
    widthOutput?: number;
    heightOutput?: number;
    alpha?: number;
    epsilon?: number;
}
export interface UpdateValueInput {
    Value: GPUTensorI;
    Moments: GPUTensorI;
    alpha: number;
    epsilon: number;
}
export declare class UpdateValueOperation<GPUTensor extends GPUTensorI> extends Operation<GPUTensor, UpdateValueInfo, UpdateValueInput> {
    protected maxIterations: number;
    constructor(tensorConstructor: GPUTensorConstructor<GPUTensor>, dtype: DTypeGpu, allocator?: GPUMemoryAllocator);
    getVariables(): string;
    getFragmentShader(info: UpdateValueInfo): string;
    getTextureNames(): string[];
    getUniformAttrs(): Input[];
    calc(input: UpdateValueInput): GPUTensor;
    getOutputShape(input: UpdateValueInput): readonly number[];
    compile(info: UpdateValueInfo): void;
    getCompilationInfo(input: UpdateValueInput): UpdateValueInfo;
    getInputInfoString(input: UpdateValueInput): string;
}
export declare const defaultUpdateValueD: Dispatcher<GPUTensorI, unknown, unknown, UpdateValueOperation<import("../../../tensor/gpu/tensor").GPUTensor<DTypeGpu>>>;