UNPKG

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

Version:

Data analysis model package without any dependencies

27 lines (24 loc) 586 B
import { onnx } from '../onnx_exporter.js' /** * Handle min layer */ export default { /** * Export to onnx object. * @param {onnx.ModelProto} model Model object * @param {import("../../graph").LayerObject & {type: 'min'}} obj Node object */ export(model, obj) { if (!Array.isArray(obj.input)) { throw new Error(`Invalid attribute 'input' value ${obj.input}.`) } const node = new onnx.NodeProto() node.setOpType('Min') for (const i of obj.input) { node.addInput(i) } node.addOutput(obj.name) const graph = model.getGraph() graph.addNode(node) }, }