@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
18 lines (16 loc) • 394 B
JavaScript
import { onnx } from '../onnx_importer.js'
/**
* Handle output node
* @module HandleONNXOutputNode
*/
export default {
/**
* Import from onnx object.
* @param {onnx.ModelProto} model Model object
* @param {onnx.ValueInfoProto} node Node object
* @returns {object[]} Objects represented a layer
*/
import(model, node) {
return [{ type: 'output', input: node.getName() }]
},
}