UNPKG

@ai-on-browser/data-analysis-models

Version:

Data analysis model package without any dependencies

31 lines (30 loc) 970 B
export { default as onnx } from "./onnx_pb.js"; /** * ONNX exporter */ export default class ONNXExporter { /** * Create onnx model proto. * @param {object} [config] object * @param {object} [config.opset] Config for operator set * @param {number} [config.opset.version] Version of operator set * @returns {onnx.ModelProto} Model proto */ static createONNXModel({ opset }?: { opset?: { version?: number; }; }): onnx.ModelProto; /** * Dump layer object. * @param {import("../graph").LayerObject["type"]} type The layer type * @returns {*} Layer exporter */ static getLayerExporter(type: import("../graph").LayerObject["type"]): any; /** * Dump onnx model. * @param {import("../graph").LayerObject[]} nodes represented the graph * @returns {Uint8Array} Protocol buffer data */ static dump(nodes: import("../graph").LayerObject[]): Uint8Array; }