@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
27 lines (24 loc) • 586 B
JavaScript
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)
},
}