import * as tf from '@tensorflow/tfjs';
import type { Denoiser } from './denoiser';
import type { DenoiserProps, InputOptions } from 'types/types';
export declare function setupBackend(denoiser: Denoiser, prefered?: string, canvasOrDevice?: HTMLCanvasElement | GPUDevice): Promise<tf.KernelBackend | undefined>;
export declare function determineTensorMap(props: DenoiserProps): {
    tensorMapLabel: string;
    size: "small" | "large" | "default";
};
export declare function adjustSize(denoiser: Denoiser, options: InputOptions): void;
export declare function warmstart(model: tf.LayersModel, height: number, width: number, channels: number): void;
export declare function handleInputTensors(denoiser: Denoiser, name: 'color' | 'albedo' | 'normal', inputTensor: tf.Tensor3D, options: InputOptions): Promise<void>;
export declare function handleModelInput(denoiser: Denoiser): Promise<tf.Tensor4D>;
export declare function handleModelOutput(denoiser: Denoiser, result: tf.Tensor4D): Promise<tf.Tensor3D>;
export declare function handleCallback(denoiser: Denoiser, outputTensor: tf.Tensor3D, returnType: string, callback?: Function | undefined): Promise<tf.Tensor3D | ImageData | WebGLTexture>;
export declare function profileModel(model: tf.LayersModel, inputTensor: tf.Tensor4D): Promise<void>;