@ai-on-browser/data-analysis-models
Version:
Data analysis model package without any dependencies
29 lines (28 loc) • 821 B
TypeScript
declare namespace _default {
/**
* Export to onnx object.
* @param {onnx.ModelProto} model Model object
* @param {import("../../graph.js").LayerObject & {type: 'lrn'}} obj Node object
* @param {{[key: string]: {type: onnx.TensorProto.DataType; size: number[]}}} info Output informatino of other layers
*/
function _export(model: onnx.ModelProto, obj: {
type: "lrn";
alpha?: number;
beta?: number;
k?: number;
n: number;
channel_dim?: number;
} & {
input?: string | string[];
name?: string;
} & {
type: "lrn";
}, info: {
[key: string]: {
type: onnx.TensorProto.DataType;
size: number[];
};
}): void;
export { _export as export };
}
export default _default;