@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
31 lines (30 loc) • 970 B
TypeScript
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;
}