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; }; export interface OnnxModelI { getNodes(): { [id: number]: OnnxNode; }; getNodeWithInput(output: string): number; }