@hoff97/tensor-js/dist/lib/onnx/types.d.ts

PyTorch like deep learning inferrence library

import { onnx } from 'onnx-proto';
import { Mode } from '../model/module';
import Tensor from '../types';
import { OnnxNode } from './node';
export declare type Attributes = onnx.IAttributeProto[];
export declare type NodeConstructor = (attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode) => OnnxNode;
export declare type Constants = {
    [name: string]: Tensor<any>;
};
export interface OnnxModelI {
    getNodes(): {
        [id: number]: OnnxNode;
    };
    getNodeWithInput(output: string): number;
}