oidn-web 0.3.5 → 0.5.0

package/lib/graphOptimizer.js

@@ -0,0 +1,215 @@
+function nodeInputs(node) {
+    return node.op === 'concat' ? node.inputs : [node.input];
+}
+function buildConsumers(spec) {
+    const consumers = new Map();
+    for (const node of spec.nodes) {
+        for (const input of nodeInputs(node)) {
+            const list = consumers.get(input) ?? [];
+            list.push(node);
+            consumers.set(input, list);
+        }
+    }
+    return consumers;
+}
+function onlyConsumer(consumers, value, op) {
+    const list = consumers.get(value);
+    if (list?.length !== 1 || list[0].op !== op)
+        return undefined;
+    return list[0];
+}
+/**
+ * Applies topology-only fusions. It never depends on a particular OIDN model
+ * name, so new descriptors automatically benefit from known graph patterns.
+ */
+export function optimizeModelGraph(validated, options = {}) {
+    const spec = validated.spec;
+    const consumers = buildConsumers(spec);
+    const nodesById = new Map(spec.nodes.map((node) => [node.id, node]));
+    const eliminated = new Set();
+    const fusedAt = new Map();
+    let convPool = 0;
+    let upsampleConcatConv = 0;
+    if (options.fuseConvPool !== false) {
+        for (const node of spec.nodes) {
+            if (node.op !== 'conv2d' || node.activation !== 'relu')
+                continue;
+            const pool = onlyConsumer(consumers, node.id, 'maxPool2d');
+            if (!pool || pool.size !== 2 || pool.stride !== 2)
+                continue;
+            eliminated.add(node.id);
+            fusedAt.set(pool.id, {
+                op: 'fusedConvReluMaxPool2d',
+                id: pool.id,
+                input: node.input,
+                conv: node,
+                pool
+            });
+            convPool++;
+        }
+    }
+    if (options.fuseUpsampleConcatConv !== false) {
+        for (const node of spec.nodes) {
+            if (node.op !== 'conv2d')
+                continue;
+            const concat = nodesById.get(node.input);
+            if (concat?.op !== 'concat' || concat.inputs.length !== 2)
+                continue;
+            if (onlyConsumer(consumers, concat.id, 'conv2d') !== node)
+                continue;
+            // The native blocked layout can remove concat only when the source
+            // boundary is also a vec4 boundary. Other graphs keep the generic ops.
+            const firstInputChannels = validated.channelsByValue.get(concat.inputs[0]);
+            if (firstInputChannels === undefined || firstInputChannels % 4 !== 0) {
+                continue;
+            }
+            const inputs = concat.inputs.map((value) => {
+                const candidate = nodesById.get(value);
+                if (candidate?.op === 'upsample2d' &&
+                    candidate.scale === 2 &&
+                    candidate.mode === 'nearest' &&
+                    onlyConsumer(consumers, candidate.id, 'concat') === concat) {
+                    return { value: candidate.input, upsample: candidate };
+                }
+                return { value };
+            });
+            const upsampleCount = inputs.filter((input) => input.upsample).length;
+            if (upsampleCount !== 1)
+                continue;
+            eliminated.add(concat.id);
+            for (const value of concat.inputs) {
+                const candidate = nodesById.get(value);
+                if (candidate?.op === 'upsample2d')
+                    eliminated.add(candidate.id);
+            }
+            fusedAt.set(node.id, {
+                op: 'fusedUpsampleConcatConv2d',
+                id: node.id,
+                inputs,
+                conv: node
+            });
+            upsampleConcatConv++;
+        }
+    }
+    const nodes = [];
+    for (const node of spec.nodes) {
+        const fused = fusedAt.get(node.id);
+        if (fused) {
+            nodes.push(fused);
+        }
+        else if (!eliminated.has(node.id)) {
+            nodes.push(node);
+        }
+    }
+    return {
+        spec,
+        nodes,
+        fusions: { convPool, upsampleConcatConv }
+    };
+}
+function executableInputs(node) {
+    if (node.op === 'concat')
+        return node.inputs;
+    if (node.op === 'fusedUpsampleConcatConv2d') {
+        return node.inputs.map((input) => input.value);
+    }
+    return [node.input];
+}
+function sameSpatialShape(left, right) {
+    return left.width === right.width && left.height === right.height;
+}
+/** Resolve all runtime shapes and value lifetimes before allocating GPU data. */
+export function planModelExecution(validated, width, height, options) {
+    if (!Number.isInteger(width) || width <= 0 || !Number.isInteger(height) || height <= 0) {
+        throw new Error(`Invalid model input size ${width}x${height}`);
+    }
+    const graph = optimizeModelGraph(validated, options);
+    const inputShape = { width, height, channels: validated.inputChannels };
+    const valueShapes = new Map([
+        [validated.spec.input, inputShape]
+    ]);
+    const outputShapes = [];
+    const shapeOf = (value, nodeId) => {
+        const shape = valueShapes.get(value);
+        if (!shape)
+            throw new Error(`Planned node ${nodeId} reads missing value ${value}`);
+        return shape;
+    };
+    for (const node of graph.nodes) {
+        let outputShape;
+        if (node.op === 'conv2d') {
+            const input = shapeOf(node.input, node.id);
+            outputShape = {
+                width: input.width,
+                height: input.height,
+                channels: validated.convChannels.get(node.id).outputChannels
+            };
+        }
+        else if (node.op === 'maxPool2d') {
+            const input = shapeOf(node.input, node.id);
+            outputShape = {
+                width: Math.ceil(input.width / 2),
+                height: Math.ceil(input.height / 2),
+                channels: input.channels
+            };
+        }
+        else if (node.op === 'upsample2d') {
+            const input = shapeOf(node.input, node.id);
+            outputShape = {
+                width: input.width * 2,
+                height: input.height * 2,
+                channels: input.channels
+            };
+        }
+        else if (node.op === 'concat') {
+            const inputs = node.inputs.map((value) => shapeOf(value, node.id));
+            if (inputs.some((shape) => !sameSpatialShape(shape, inputs[0]))) {
+                throw new Error(`Concat ${node.id} has mismatched spatial shapes`);
+            }
+            outputShape = {
+                width: inputs[0].width,
+                height: inputs[0].height,
+                channels: inputs.reduce((sum, shape) => sum + shape.channels, 0)
+            };
+        }
+        else if (node.op === 'fusedConvReluMaxPool2d') {
+            const input = shapeOf(node.input, node.id);
+            outputShape = {
+                width: Math.ceil(input.width / 2),
+                height: Math.ceil(input.height / 2),
+                channels: validated.convChannels.get(node.conv.id).outputChannels
+            };
+        }
+        else {
+            const inputs = node.inputs.map((input) => {
+                const source = shapeOf(input.value, node.id);
+                return input.upsample
+                    ? { ...source, width: source.width * 2, height: source.height * 2 }
+                    : source;
+            });
+            if (inputs.some((shape) => !sameSpatialShape(shape, inputs[0]))) {
+                throw new Error(`Fused decoder ${node.id} has mismatched spatial shapes`);
+            }
+            outputShape = {
+                width: inputs[0].width,
+                height: inputs[0].height,
+                channels: validated.convChannels.get(node.conv.id).outputChannels
+            };
+        }
+        valueShapes.set(node.id, outputShape);
+        outputShapes.push(outputShape);
+    }
+    const lastUses = new Map();
+    graph.nodes.forEach((node, index) => {
+        for (const input of executableInputs(node))
+            lastUses.set(input, index);
+    });
+    lastUses.set(validated.spec.output, graph.nodes.length);
+    const plannedNodes = graph.nodes.map((node, index) => ({
+        node,
+        outputShape: outputShapes[index],
+        lastUse: lastUses.get(node.id) ?? index
+    }));
+    return { ...graph, inputShape, valueShapes, plannedNodes };
+}
+//# sourceMappingURL=graphOptimizer.js.map

package/lib/graphOptimizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graphOptimizer.js","sourceRoot":"","sources":["../src/graphOptimizer.ts"],"names":[],"mappings":"AA+CA,SAAS,UAAU,CAAC,IAAmB;IACrC,OAAO,IAAI,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,cAAc,CAAC,IAAmB;IACzC,MAAM,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;IACrD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YACxC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,YAAY,CACnB,SAA+C,EAC/C,KAAa,EACb,EAAK;IAEL,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAClC,IAAI,IAAI,EAAE,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE;QAAE,OAAO,SAAS,CAAC;IAC9D,OAAO,IAAI,CAAC,CAAC,CAAsC,CAAC;AACtD,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,kBAAkB,CAChC,SAAyB,EACzB,UAAoC,EAAE;IAEtC,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC;IAC5B,MAAM,SAAS,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IACrE,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,GAAG,EAA+B,CAAC;IACvD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAE3B,IAAI,OAAO,CAAC,YAAY,KAAK,KAAK,EAAE,CAAC;QACnC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,KAAK,MAAM;gBAAE,SAAS;YACjE,MAAM,IAAI,GAAG,YAAY,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAE5D,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACxB,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;gBACnB,EAAE,EAAE,wBAAwB;gBAC5B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI,EAAE,IAAI;gBACV,IAAI;aACL,CAAC,CAAC;YACH,QAAQ,EAAE,CAAC;QACb,CAAC;IACH,CAAC;IAED,IAAI,OAAO,CAAC,sBAAsB,KAAK,KAAK,EAAE,CAAC;QAC7C,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ;gBAAE,SAAS;YACnC,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzC,IAAI,MAAM,EAAE,EAAE,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YACpE,IAAI,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,KAAK,IAAI;gBAAE,SAAS;YACpE,mEAAmE;YACnE,uEAAuE;YACvE,MAAM,kBAAkB,GAAG,SAAS,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3E,IAAI,kBAAkB,KAAK,SAAS,IAAI,kBAAkB,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACrE,SAAS;YACX,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACzC,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACvC,IACE,SAAS,EAAE,EAAE,KAAK,YAAY;oBAC9B,SAAS,CAAC,KAAK,KAAK,CAAC;oBACrB,SAAS,CAAC,IAAI,KAAK,SAAS;oBAC5B,YAAY,CAAC,SAAS,EAAE,SAAS,CAAC,EAAE,EAAE,QAAQ,CAAC,KAAK,MAAM,EAC1D,CAAC;oBACD,OAAO,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;gBACzD,CAAC;gBACD,OAAO,EAAE,KAAK,EAAE,CAAC;YACnB,CAAC,CAAC,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC;YACtE,IAAI,aAAa,KAAK,CAAC;gBAAE,SAAS;YAElC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC1B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClC,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACvC,IAAI,SAAS,EAAE,EAAE,KAAK,YAAY;oBAAE,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YACnE,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;gBACnB,EAAE,EAAE,2BAA2B;gBAC/B,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,MAAM;gBACN,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YACH,kBAAkB,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAA0B,EAAE,CAAC;IACxC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,CAAC;aAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YACpC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED,OAAO;QACL,IAAI;QACJ,KAAK;QACL,OAAO,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE;KAC1C,CAAC;AACJ,CAAC;AAqBD,SAAS,gBAAgB,CAAC,IAAyB;IACjD,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,MAAM,CAAC;IAC7C,IAAI,IAAI,CAAC,EAAE,KAAK,2BAA2B,EAAE,CAAC;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,gBAAgB,CACvB,IAAqB,EACrB,KAAsB;IAEtB,OAAO,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,KAAK,KAAK,CAAC,MAAM,CAAC;AACpE,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,kBAAkB,CAChC,SAAyB,EACzB,KAAa,EACb,MAAc,EACd,OAAkC;IAElC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC,EAAE,CAAC;QACvF,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,IAAI,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACrD,MAAM,UAAU,GAAG,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC,aAAa,EAAE,CAAC;IACxE,MAAM,WAAW,GAAG,IAAI,GAAG,CAA0B;QACnD,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC;KACnC,CAAC,CAAC;IACH,MAAM,YAAY,GAAsB,EAAE,CAAC;IAE3C,MAAM,OAAO,GAAG,CAAC,KAAa,EAAE,MAAc,EAAE,EAAE;QAChD,MAAM,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,MAAM,wBAAwB,KAAK,EAAE,CAAC,CAAC;QACnF,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;IAEF,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAC/B,IAAI,WAA4B,CAAC;QACjC,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,WAAW,GAAG;gBACZ,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,QAAQ,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC,cAAc;aAC9D,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,EAAE,KAAK,WAAW,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,WAAW,GAAG;gBACZ,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;gBACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;gBACnC,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,EAAE,KAAK,YAAY,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,WAAW,GAAG;gBACZ,KAAK,EAAE,KAAK,CAAC,KAAK,GAAG,CAAC;gBACtB,MAAM,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC;gBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,EAAE,KAAK,QAAQ,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACnE,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,EAAE,gCAAgC,CAAC,CAAC;YACrE,CAAC;YACD,WAAW,GAAG;gBACZ,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK;gBACtB,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM;gBACxB,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;aACjE,CAAC;QACJ,CAAC;aAAM,IAAI,IAAI,CAAC,EAAE,KAAK,wBAAwB,EAAE,CAAC;YAChD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3C,WAAW,GAAG;gBACZ,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;gBACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;gBACnC,QAAQ,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC,cAAc;aACnE,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACvC,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC7C,OAAO,KAAK,CAAC,QAAQ;oBACnB,CAAC,CAAC,EAAE,GAAG,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,GAAG,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;oBACnE,CAAC,CAAC,MAAM,CAAC;YACb,CAAC,CAAC,CAAC;YACH,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,MAAM,IAAI,KAAK,CAAC,iBAAiB,IAAI,CAAC,EAAE,gCAAgC,CAAC,CAAC;YAC5E,CAAC;YACD,WAAW,GAAG;gBACZ,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK;gBACtB,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM;gBACxB,QAAQ,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAE,CAAC,cAAc;aACnE,CAAC;QACJ,CAAC;QAED,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;QACtC,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC3C,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAClC,KAAK,MAAM,KAAK,IAAI,gBAAgB,CAAC,IAAI,CAAC;YAAE,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IACH,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAExD,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QACrD,IAAI;QACJ,WAAW,EAAE,YAAY,CAAC,KAAK,CAAC;QAChC,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK;KACxC,CAAC,CAAC,CAAC;IAEJ,OAAO,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;AAC7D,CAAC"}

package/lib/hdrTransfer.d.ts

@@ -0,0 +1,14 @@
+/** HDR transfer functions used by the OIDN input and output processing passes. */
+export type HDRTransfer = 'pu' | 'log';
+export declare function hdrTransferFuncCPU({ data, channels, inputScale, transfer }: {
+    data: Float32Array;
+    channels: number;
+    inputScale: number;
+    transfer?: HDRTransfer;
+}): Float32Array<ArrayBuffer>;
+export declare function hdrTransferFuncInverseCPU({ data, channels, inputScale, transfer }: {
+    data: Float32Array;
+    channels: number;
+    inputScale: number;
+    transfer?: HDRTransfer;
+}): Float32Array<ArrayBuffer>;

package/lib/hdrTransfer.js

@@ -0,0 +1,61 @@
+const a = 1.41283765e3;
+const b = 1.64593172;
+const c = 4.31384981e-1;
+const d = -2.94139609e-3;
+const e = 1.92653254e-1;
+const f = 6.26026094e-3;
+const g = 9.98620152e-1;
+const y0 = 1.5794576e-6;
+const y1 = 3.22087631e-2;
+const x0 = 2.23151711e-3;
+const x1 = 3.70974749e-1;
+const yMax = 65504;
+const puXMax = puForward(yMax);
+const puNormScale = 1 / puXMax;
+const puRcpNormScale = puXMax;
+const logXMax = Math.log(yMax + 1);
+const logNormScale = 1 / logXMax;
+function puForward(y) {
+    if (y <= y0)
+        return a * y;
+    if (y <= y1)
+        return b * Math.pow(y, c) + d;
+    return e * Math.log(y + f) + g;
+}
+function puInverse(x) {
+    if (x <= x0)
+        return x / a;
+    if (x <= x1)
+        return Math.pow((x - d) / b, 1 / c);
+    return Math.exp((x - g) / e) - f;
+}
+function forward(y, transfer) {
+    return transfer === 'log'
+        ? Math.log(y + 1) * logNormScale
+        : puForward(y) * puNormScale;
+}
+function inverse(x, transfer) {
+    return transfer === 'log'
+        ? Math.exp(x * logXMax) - 1
+        : puInverse(x * puRcpNormScale);
+}
+export function hdrTransferFuncCPU({ data, channels, inputScale, transfer = 'pu' }) {
+    const newData = new Float32Array(data);
+    for (let i = 0; i < newData.length; i += channels) {
+        for (let channel = 0; channel < 3; channel++) {
+            newData[i + channel] = forward(newData[i + channel] * inputScale, transfer);
+        }
+    }
+    return newData;
+}
+export function hdrTransferFuncInverseCPU({ data, channels, inputScale, transfer = 'pu' }) {
+    const newData = new Float32Array(data);
+    const outputScale = 1 / inputScale;
+    for (let i = 0; i < newData.length; i += channels) {
+        for (let channel = 0; channel < 3; channel++) {
+            newData[i + channel] = inverse(newData[i + channel], transfer) * outputScale;
+        }
+    }
+    return newData;
+}
+//# sourceMappingURL=hdrTransfer.js.map

package/lib/hdrTransfer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hdrTransfer.js","sourceRoot":"","sources":["../src/hdrTransfer.ts"],"names":[],"mappings":"AAGA,MAAM,CAAC,GAAG,YAAY,CAAC;AACvB,MAAM,CAAC,GAAG,UAAU,CAAC;AACrB,MAAM,CAAC,GAAG,aAAa,CAAC;AACxB,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC;AACzB,MAAM,CAAC,GAAG,aAAa,CAAC;AACxB,MAAM,CAAC,GAAG,aAAa,CAAC;AACxB,MAAM,CAAC,GAAG,aAAa,CAAC;AACxB,MAAM,EAAE,GAAG,YAAY,CAAC;AACxB,MAAM,EAAE,GAAG,aAAa,CAAC;AACzB,MAAM,EAAE,GAAG,aAAa,CAAC;AACzB,MAAM,EAAE,GAAG,aAAa,CAAC;AACzB,MAAM,IAAI,GAAG,KAAK,CAAC;AACnB,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC;AAC/B,MAAM,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC;AAC/B,MAAM,cAAc,GAAG,MAAM,CAAC;AAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;AACnC,MAAM,YAAY,GAAG,CAAC,GAAG,OAAO,CAAC;AAEjC,SAAS,SAAS,CAAC,CAAS;IAC1B,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAC1B,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;IAC3C,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,SAAS,CAAC,CAAS;IAC1B,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAC1B,IAAI,CAAC,IAAI,EAAE;QAAE,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,OAAO,CAAC,CAAS,EAAE,QAAqB;IAC/C,OAAO,QAAQ,KAAK,KAAK;QACvB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,YAAY;QAChC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC;AACjC,CAAC;AAED,SAAS,OAAO,CAAC,CAAS,EAAE,QAAqB;IAC/C,OAAO,QAAQ,KAAK,KAAK;QACvB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC;QAC3B,CAAC,CAAC,SAAS,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC;AACpC,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,EACjC,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,QAAQ,GAAG,IAAI,EAMhB;IACC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC;QAClD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,OAAO,CAC5B,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,UAAU,EACjC,QAAQ,CACT,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,EACxC,IAAI,EACJ,QAAQ,EACR,UAAU,EACV,QAAQ,GAAG,IAAI,EAMhB;IACC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,WAAW,GAAG,CAAC,GAAG,UAAU,CAAC;IACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC;QAClD,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;YAC7C,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,OAAO,CAAC,EAAE,QAAQ,CAAC,GAAG,WAAW,CAAC;QAC/E,CAAC;IACH,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/lib/main.d.ts

@@ -1,19 +1,51 @@
 import { parseTZA } from './tza';
 import UNet from './UNet';
+import type { UNetEngineSetting } from './UNet';
+import type { DynamicTileSetting } from './tileScheduler';
+import type { UNetModelSpec } from './modelSpec';
+import type { NativeUNetGemmOptions, NativeUNetKernelSetting, NativeUNetPrecisionSetting } from './nativeUNet';
+import type { HDRTransfer } from './process';
 export { parseTZA, UNet };
-export declare function initUNetFromBuffer(tzaBuffer: ArrayBuffer, backendParams?: {
-    device: GPUDevice;
-    adapterInfo: GPUAdapterInfo;
-}, opts?: {
+export { planTileGrid } from './tileScheduler';
+export type { DynamicTileOptions, DynamicTileSetting, PlannedTile, TilePlan, TileRect } from './tileScheduler';
+export { detectUNetModelSpec, OIDN_UNET_LARGE_SPEC, OIDN_UNET_SMALL_SPEC, validateUNetModel } from './modelSpec';
+export type { ModelNodeSpec, UNetModelGraph, UNetModelSpec, ValidatedUNetModel } from './modelSpec';
+export { optimizeModelGraph, planModelExecution } from './graphOptimizer';
+export type { ExecutableModelNode, GraphOptimizationOptions, ModelExecutionPlan, ModelValueShape, OptimizedModelGraph } from './graphOptimizer';
+export { NativeUNetExecutor, resolveNativeUNetPrecision } from './nativeUNet';
+export type { NativeUNetGemmWorkgroup, NativeUNetGemmOptions, NativeUNetExecutionProfile, NativeUNetKernel, NativeUNetKernelSetting, NativeUNetLayerTiming, NativeUNetOptions, NativeUNetPrecision, NativeUNetPrecisionSetting } from './nativeUNet';
+export interface UNetOptions {
     aux?: boolean;
     hdr?: boolean;
+    /** HDR transfer function expected by the trained model. Defaults to PU. */
+    hdrTransfer?: HDRTransfer;
+    /** Hard upper bound for an output tile edge. Defaults to 512. */
     maxTileSize?: number;
-}): Promise<UNet>;
+    /** Adaptive GPU-time-based tile sizing. Enabled by default. */
+    dynamicTile?: DynamicTileSetting;
+    /** `auto` uses native WGSL; `webnn` opts into the WebNN backend. */
+    engine?: UNetEngineSetting;
+    /** `auto` selects FP16 when shader-f16 was enabled on the GPUDevice. */
+    precision?: NativeUNetPrecisionSetting;
+    /** `auto` uses implicit GEMM for FP16/FP32 convolutions, except the direct output layer. */
+    kernel?: NativeUNetKernelSetting;
+    /**
+     * Experimental implicit-GEMM tuning for native WGSL execution. These knobs
+     * exist for benchmarking and may change or be removed in any release.
+     * @experimental
+     */
+    gemm?: NativeUNetGemmOptions;
+    /** Versioned topology descriptor for future/custom OIDN TZA models. */
+    modelSpec?: UNetModelSpec;
+}
+export type { UNetEngineSetting, UNetExecutionStats } from './UNet';
+export type { HDRTransfer } from './process';
+export { WebNNUNetExecutor } from './webnnUNet';
+export type { WebNNRuntimeSupport, WebNNUNetOptions } from './webnnUNet';
+export type { OIDNResourceKind, OIDNResourceSnapshot, OIDNResourceStats } from './resourceTracker';
+export declare function initUNetFromBuffer(tzaBuffer: ArrayBuffer, backendParams?: {
+    device: GPUDevice;
+}, opts?: UNetOptions): Promise<UNet>;
 export declare function initUNetFromURL(modelPath: string, backendParams?: {
     device: GPUDevice;
-    adapterInfo: GPUAdapterInfo;
-}, opts?: {
-    aux?: boolean;
-    hdr?: boolean;
-    maxTileSize?: number;
-}): Promise<UNet>;
+}, opts?: UNetOptions): Promise<UNet>;

package/lib/main.js

@@ -1,13 +1,17 @@
 import { parseTZA } from './tza';
 import UNet from './UNet';
-import { initWebGPUBackend, initWebGPUBackendWithDevice } from './backend';
+import { initWebGPUBackend } from './backend';
 export { parseTZA, UNet };
+export { planTileGrid } from './tileScheduler';
+export { detectUNetModelSpec, OIDN_UNET_LARGE_SPEC, OIDN_UNET_SMALL_SPEC, validateUNetModel } from './modelSpec';
+export { optimizeModelGraph, planModelExecution } from './graphOptimizer';
+export { NativeUNetExecutor, resolveNativeUNetPrecision } from './nativeUNet';
+export { WebNNUNetExecutor } from './webnnUNet';
 export async function initUNetFromBuffer(tzaBuffer, backendParams, opts) {
-    const backend = await (backendParams
-        ? initWebGPUBackendWithDevice(backendParams.device, backendParams.adapterInfo)
-        : initWebGPUBackend());
+    const device = backendParams?.device ?? await initWebGPUBackend();
     const tensors = parseTZA(tzaBuffer);
-    const unet = new UNet(tensors, backend, opts);
+    const unet = new UNet(tensors, device, opts);
+    await unet.prepare();
     return unet;
 }
 export async function initUNetFromURL(modelPath, backendParams, opts) {

package/lib/main.js.map

@@ -1 +1 @@
-{"version":3,"file":"main.js","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjC,OAAO,IAAI,MAAM,QAAQ,CAAC;AAC1B,OAAO,EAAE,iBAAiB,EAAE,2BAA2B,EAAE,MAAM,WAAW,CAAC;AAE3E,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,SAAsB,EACtB,aAAkE,EAClE,IAIC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa;QAClC,CAAC,CAAC,2BAA2B,CACzB,aAAa,CAAC,MAAM,EACpB,aAAa,CAAC,WAAW,CAC1B;QACH,CAAC,CAAC,iBAAiB,EAAE,CAAC,CAAC;IACzB,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,OAAQ,EAAE,IAAI,CAAC,CAAC;IAC/C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,SAAiB,EACjB,aAAkE,EAClE,IAIC;IAED,OAAO,KAAK,CAAC,SAAS,CAAC;SACpB,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;SAChC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE;QACX,OAAO,kBAAkB,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACP,CAAC"}
+{"version":3,"file":"main.js","sourceRoot":"","sources":["../src/main.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjC,OAAO,IAAI,MAAM,QAAQ,CAAC;AAE1B,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAU9C,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAQ/C,OAAO,EACL,mBAAmB,EACnB,oBAAoB,EACpB,oBAAoB,EACpB,iBAAiB,EAClB,MAAM,aAAa,CAAC;AAOrB,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAQ1E,OAAO,EACL,kBAAkB,EAClB,0BAA0B,EAC3B,MAAM,cAAc,CAAC;AAwCtB,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAQhD,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,SAAsB,EACtB,aAAqC,EACrC,IAAkB;IAElB,MAAM,MAAM,GAAG,aAAa,EAAE,MAAM,IAAI,MAAM,iBAAiB,EAAE,CAAC;IAClE,MAAM,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IACpC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAC7C,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;IACrB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,SAAiB,EACjB,aAAqC,EACrC,IAAkB;IAElB,OAAO,KAAK,CAAC,SAAS,CAAC;SACpB,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC;SAChC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE;QACX,OAAO,kBAAkB,CAAC,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACP,CAAC"}

package/lib/modelSpec.d.ts

@@ -0,0 +1,80 @@
+import type { HostTensor } from './tza';
+/**
+ * Versioned, runtime-independent description of an OIDN network.
+ *
+ * TZA contains named tensors but no executable graph. Keeping the graph in a
+ * small declarative descriptor makes model upgrades independent from the WGSL
+ * kernels: a new OIDN topology only needs a new descriptor and validation
+ * fixture unless it introduces a genuinely new operation.
+ */
+export interface UNetModelSpec {
+    schemaVersion: 1;
+    id: string;
+    family: 'oidn-unet-small' | 'oidn-unet-large' | (string & {});
+    input: string;
+    output: string;
+    receptiveField: number;
+    nodes: readonly ModelNodeSpec[];
+    /** Reject unrecognised tensors so an upstream topology change is explicit. */
+    allowAdditionalTensors?: boolean;
+}
+export type ModelActivation = 'identity' | 'relu';
+export interface Conv2DNodeSpec {
+    op: 'conv2d';
+    id: string;
+    input: string;
+    weight: string;
+    bias: string;
+    activation: ModelActivation;
+    padding: 'same';
+}
+export interface MaxPool2DNodeSpec {
+    op: 'maxPool2d';
+    id: string;
+    input: string;
+    size: 2;
+    stride: 2;
+    padding: 'same';
+}
+export interface Upsample2DNodeSpec {
+    op: 'upsample2d';
+    id: string;
+    input: string;
+    scale: 2;
+    mode: 'nearest';
+}
+export interface ConcatNodeSpec {
+    op: 'concat';
+    id: string;
+    inputs: readonly string[];
+    axis: 'channels';
+}
+export type ModelNodeSpec = Conv2DNodeSpec | MaxPool2DNodeSpec | Upsample2DNodeSpec | ConcatNodeSpec;
+export interface ValidatedConvTensor {
+    weight: HostTensor;
+    bias: HostTensor;
+    inputChannels: number;
+    outputChannels: number;
+    kernelHeight: number;
+    kernelWidth: number;
+}
+export interface ModelConvChannels {
+    inputChannels: number;
+    outputChannels: number;
+}
+export interface UNetModelGraph {
+    spec: UNetModelSpec;
+    inputChannels: number;
+    outputChannels: number;
+    channelsByValue: ReadonlyMap<string, number>;
+    convChannels: ReadonlyMap<string, ModelConvChannels>;
+}
+export interface ValidatedUNetModel extends UNetModelGraph {
+    tensorDataType: HostTensor['desc']['dataType'];
+    convTensors: ReadonlyMap<string, ValidatedConvTensor>;
+}
+export declare const OIDN_UNET_SMALL_SPEC: UNetModelSpec;
+export declare const OIDN_UNET_LARGE_SPEC: UNetModelSpec;
+export declare function detectUNetModelSpec(tensors: ReadonlyMap<string, HostTensor>, specs?: readonly UNetModelSpec[]): UNetModelSpec;
+/** Validate tensor layout, shapes, dtypes, graph order, and channel flow. */
+export declare function validateUNetModel(tensors: ReadonlyMap<string, HostTensor>, spec?: UNetModelSpec): ValidatedUNetModel;