Function wrap

packages/transformers/src/ops/registry.js:15–32

(session_bytes, session_options, names)

 * The wrapper function for running the ONNX inference session.
 */
const wrap = async (session_bytes, session_options, names) => {
    const session = await createInferenceSession(new Uint8Array(session_bytes), session_options);

    return /** @type {any} */ (
        async (/** @type {Record<string, Tensor>} */ inputs) => {
            const proxied = isONNXProxy();
            const ortFeed = Object.fromEntries(
                Object.entries(inputs).map(([k, v]) => [k, (proxied ? v.clone() : v).ort_tensor]),
            );
            const outputs = await runInferenceSession(session, ortFeed);
            if (Array.isArray(names)) {
                return names.map((n) => new Tensor(outputs[n]));
            } else {
                return new Tensor(outputs[/** @type {string} */ (names)]);
            }
        }
    );
};

// In-memory registry of initialized ONNX operators
export class TensorOpRegistry {