UNPKG

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

Version:

Data analysis model package without any dependencies

29 lines (28 loc) 916 B
declare namespace _default { /** * Export to onnx object. * @param {onnx.ModelProto} model Model object * @param {import("../../graph").LayerObject & {type: 'reshape'}} obj Node object * @param {{[key: string]: {type: onnx.TensorProto.DataType; size: number[]}}} info Output informatino of other layers * @returns {{type: onnx.TensorProto.DataType; size: number[]}} Output information of this layer */ function _export(model: onnx.ModelProto, obj: { type: "reshape"; size: string | number[]; } & { input?: string | string[]; name?: string; } & { type: "reshape"; }, info: { [key: string]: { type: onnx.TensorProto.DataType; size: number[]; }; }): { type: onnx.TensorProto.DataType; size: number[]; }; export { _export as export }; } export default _default;