qairt-visualizer 0.5.1__py3-none-win_amd64.whl → 0.6.0__py3-none-win_amd64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- qairt_visualizer/core/helpers/version.py +23 -0
- qairt_visualizer/core/launchers/base_ui_launcher_context.py +28 -7
- qairt_visualizer/core/launchers/electron_launcher_context.py +72 -3
- qairt_visualizer/core/parsers/dlc_parser/dlc_parser.py +105 -33
- qairt_visualizer/core/parsers/dlc_parser/get_source_topology.py +47 -0
- qairt_visualizer/core/parsers/dlc_parser/libDlModelToolsPy.pyd +0 -0
- qairt_visualizer/core/parsers/dlc_parser/libDlModelToolsPy.so +0 -0
- qairt_visualizer/core/parsers/dlc_parser/libPyIrGraph.pyd +0 -0
- qairt_visualizer/core/parsers/dlc_parser/libPyIrGraph.so +0 -0
- qairt_visualizer/core/parsers/dlc_parser/models/op_tensor_mappings.py +8 -4
- qairt_visualizer/core/ui/dist/browser/__init__.py +47 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/acuity.js → acuity.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/app.js → app.js} +63 -57
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/armnn-schema.js → armnn-schema.js} +590 -5
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/armnn.js → armnn.js} +3 -5
- qairt_visualizer/core/ui/dist/browser/assets/i18n/common/en.json +8 -2
- qairt_visualizer/core/ui/dist/browser/assets/i18n/graph/en.json +32 -36
- qairt_visualizer/core/ui/dist/browser/assets/i18n/panels/en.json +9 -3
- qairt_visualizer/core/ui/dist/browser/assets/i18n/panels/model-diff/en.json +6 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/barracuda.js → barracuda.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/base.js → base.js} +20 -16
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/bigdl.js → bigdl.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/browser.js → browser.js} +59 -49
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe.js → caffe.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe2-proto.js → caffe2-proto.js} +26 -26
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe2.js → caffe2.js} +113 -114
- qairt_visualizer/core/ui/dist/browser/chunk-3SMGFPTN.js +11 -0
- qairt_visualizer/core/ui/dist/browser/chunk-43LQFSHA.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-447NKSFT.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-5MAR72QP.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-7PTY53DS.js +5 -0
- qairt_visualizer/core/ui/dist/browser/chunk-GAUD627J.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-H7TLLCSI.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-HISUA6LZ.js +4 -0
- qairt_visualizer/core/ui/dist/browser/chunk-HXB7IAZI.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-L3QJ7DR2.js +1 -0
- qairt_visualizer/core/ui/dist/browser/chunk-M7SD7HRU.js +129 -0
- qairt_visualizer/core/ui/dist/browser/{chunk-HBP6TEQ6.js → chunk-MCR4N53U.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/chunk-QD7PA2I4.js +19 -0
- qairt_visualizer/core/ui/dist/browser/chunk-TKZG6FLW.js +12 -0
- qairt_visualizer/core/ui/dist/browser/chunk-WFAVWONI.js +5 -0
- qairt_visualizer/core/ui/dist/browser/chunk-WRMDGTCT.js +35 -0
- qairt_visualizer/core/ui/dist/browser/chunk-X25J6H7V.js +1 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/circle-schema.js → circle-schema.js} +17 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/circle.js → circle.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/cntk.js → cntk.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/coreml-proto.js → coreml-proto.js} +1 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/coreml.js → coreml.js} +24 -21
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dagre.js → dagre.js} +5 -5
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/darknet-metadata.json → darknet-metadata.json} +0 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/darknet.js → darknet.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/electron.mjs → desktop.mjs} +44 -131
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dl4j.js → dl4j.js} +5 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dlc-schema.js → dlc-schema.js} +8 -8
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dlc.js → dlc.js} +110 -29
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dnn.js → dnn.js} +2 -4
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dot.js → dot.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/espresso.js → espresso.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/executorch-schema.js → executorch-schema.js} +363 -42
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/executorch.js → executorch.js} +108 -40
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/flatbuffers.js → flatbuffers.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/flax.js → flax.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/flux.js → flux.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/gguf.js → gguf.js} +44 -24
- qairt_visualizer/core/ui/dist/browser/grapher.css +145 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/grapher.js → grapher.js} +24 -48
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/hailo.js → hailo.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/hickle.js → hickle.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/imgdnn.js → imgdnn.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/index.html +15 -16
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/index.js → index.js} +2 -24
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/json.js → json.js} +28 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/kann.js → kann.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/keras-metadata.json → keras-metadata.json} +15 -7
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/keras.js → keras.js} +13 -7
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/kmodel.js → kmodel.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/lasagne.js → lasagne.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/lightgbm.js → lightgbm.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/main-VGWU2273.js +1 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mediapipe.js → mediapipe.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/megengine.js → megengine.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/message.js → message.js} +3 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mlir.js → mlir.js} +53 -17
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mlnet.js → mlnet.js} +2 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mnn-schema.js → mnn-schema.js} +9 -6
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mnn.js → mnn.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mslite.js → mslite.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mxnet.js → mxnet.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/ncnn.js → ncnn.js} +3 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/nnabla.js → nnabla.js} +4 -4
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/nnef.js → nnef.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/node.js +102 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/numpy.js → numpy.js} +9 -9
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/om.js → om.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onednn.js → onednn.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onnx-metadata.json → onnx-metadata.json} +3500 -233
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onnx-proto.js → onnx-proto.js} +436 -4
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onnx.js → onnx.js} +284 -199
- qairt_visualizer/core/ui/dist/browser/onnx.py +231 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/openvino.js → openvino.js} +36 -31
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/paddle-proto.js → paddle-proto.js} +54 -54
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/paddle.js → paddle.js} +22 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/pickle.js → pickle.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/python.js → python.js} +743 -325
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/pytorch-metadata.json → pytorch-metadata.json} +3217 -2934
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/pytorch-proto.js → pytorch-proto.js} +4 -4
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/pytorch.js → pytorch.js} +172 -86
- qairt_visualizer/core/ui/dist/browser/pytorch.py +304 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/qnn.js → qnn.js} +6 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/rknn.js → rknn.js} +3 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/safetensors.js → safetensors.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/sentencepiece.js → sentencepiece.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/server.py → server.py} +95 -103
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/sklearn-metadata.json → sklearn-metadata.json} +15 -10
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/sklearn.js → sklearn.js} +5 -5
- qairt_visualizer/core/ui/dist/browser/{styles-MRR6DOT2.css → styles-52VQCFP5.css} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tar.js → tar.js} +18 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tengine.js → tengine.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tensorrt.js → tensorrt.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tf-metadata.json → tf-metadata.json} +1255 -163
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tf-proto.js → tf-proto.js} +46 -12
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tf.js → tf.js} +36 -13
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tflite.js → tflite.js} +18 -21
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tnn.js → tnn.js} +1 -3
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/torch.js → torch.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/transformers.js +171 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tvm.js → tvm.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/uff.js → uff.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/view.js → view.js} +1405 -1115
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/weka.js → weka.js} +3 -4
- qairt_visualizer/core/ui/dist/browser/worker-CECUQ4FG.js +1 -0
- qairt_visualizer/core/ui/dist/browser/worker-IGSZ74QJ.js +4 -0
- qairt_visualizer/core/ui/dist/browser/xgboost.js +59 -0
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/xml.js → xml.js} +2 -2
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/xmodel.js → xmodel.js} +1 -1
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/zip.js → zip.js} +38 -41
- qairt_visualizer/core/ui/dist/qairt_visualizer.exe +0 -0
- qairt_visualizer/core/ui/dist/resources/app.asar +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.6.0.dist-info}/METADATA +1 -1
- qairt_visualizer-0.6.0.dist-info/RECORD +264 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/CITATION.cff +0 -10
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/CONTRIBUTING.md +0 -29
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/LICENSE +0 -21
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/README.md +0 -36
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/jenkins/Jenkinsfile +0 -69
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/package-lock.json +0 -5743
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/package.js +0 -689
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/package.json +0 -53
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/package.py +0 -87
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/background.png +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/background.svg +0 -5
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/background@2x.png +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/electron-builder.json +0 -109
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/eslint.config.js +0 -231
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/forge.config.js +0 -87
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.html +0 -27
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.icns +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.icns.svg +0 -175
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.ico +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.png +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/icon.svg +0 -179
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/notarize.cjs +0 -24
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/pyproject.toml +0 -63
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/publish/setup.py +0 -43
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/resolve_upgrade_merge.sh +0 -10
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/__init__.py +0 -41
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/cambricon.js +0 -32
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/favicon.ico +0 -0
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/grapher.css +0 -602
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/index.html +0 -2222
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/modular.js +0 -81
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/onnx.py +0 -227
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/pytorch.py +0 -298
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/qais/qcontrol.js +0 -207
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/qais/qgraph-toolkit.js +0 -123
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/qais/qsubgraph.js +0 -226
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/qais/qsummary-panel.js +0 -89
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/source/server.js +0 -250
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/test/backend.py +0 -80
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/test/measures.py +0 -27
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/test/models.js +0 -359
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/test/models.json +0 -8523
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/test/worker.js +0 -859
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/armnn +0 -33
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/bigdl +0 -33
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/caffe +0 -37
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/caffe-script.js +0 -12
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/circle +0 -43
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/circle-script.js +0 -87
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/circle_script.js +0 -73
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/cntk +0 -33
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/coreml +0 -49
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/dlc +0 -20
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/dlc.fbs +0 -165
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/dnn +0 -20
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/dnn.proto +0 -79
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/executorch +0 -41
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/flatc.js +0 -1315
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/gguf +0 -22
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/kann +0 -33
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/keras +0 -62
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/keras_metadata.py +0 -205
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/keras_script.py +0 -204
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/mediapipe +0 -72
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/megengine +0 -47
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/megengine-script.js +0 -113
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/megengine_script.js +0 -112
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/mnn +0 -40
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/mslite +0 -45
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/mslite-script.js +0 -83
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/mslite_metadata.js +0 -82
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/nnabla +0 -79
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/nnabla_script.py +0 -145
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/om +0 -26
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/om.proto +0 -204
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/onnx +0 -135
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/onnx_metadata.py +0 -233
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/onnx_script.py +0 -332
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/paddle +0 -38
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/protoc.js +0 -1609
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/pytorch +0 -85
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/pytorch_metadata.py +0 -166
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/pytorch_script.py +0 -404
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/rknn +0 -20
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/rknn.fbs +0 -78
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/sentencepiece +0 -39
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/sklearn +0 -65
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/sklearn_metadata.py +0 -126
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/sklearn_script.py +0 -129
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/tf +0 -97
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/tf_metadata.py +0 -403
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/tf_script.py +0 -422
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/tflite-script.js +0 -87
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/tflite_metadata.js +0 -73
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/uff +0 -20
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/uff.proto +0 -127
- qairt_visualizer/core/ui/dist/browser/assets/netron_plugin/tools/xmodel +0 -36
- qairt_visualizer/core/ui/dist/browser/chunk-3NAQCYEU.js +0 -1
- qairt_visualizer/core/ui/dist/browser/chunk-NFKTPX3Z.js +0 -11
- qairt_visualizer/core/ui/dist/browser/chunk-OMI6HJ5Y.js +0 -19
- qairt_visualizer/core/ui/dist/browser/chunk-TTZXMRJ5.js +0 -34
- qairt_visualizer/core/ui/dist/browser/main-NXPVOKMG.js +0 -1
- qairt_visualizer/core/ui/dist/browser/worker-5VPKG2GV.js +0 -4
- qairt_visualizer/core/ui/dist/browser/worker-UVGWDZ5K.js +0 -1
- qairt_visualizer-0.5.1.dist-info/RECORD +0 -337
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/acuity-metadata.json → acuity-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/armnn-metadata.json → armnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/bigdl-metadata.json → bigdl-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/bigdl-proto.js → bigdl-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe-metadata.json → caffe-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe-proto.js → caffe-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/caffe2-metadata.json → caffe2-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/catboost.js → catboost.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/circle-metadata.json → circle-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/cntk-metadata.json → cntk-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/cntk-proto.js → cntk-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/coreml-metadata.json → coreml-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dl4j-metadata.json → dl4j-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dlc-metadata.json → dlc-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dnn-metadata.json → dnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/dnn-proto.js → dnn-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/espresso-metadata.json → espresso-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/flexbuffers.js → flexbuffers.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/flux-metadata.json → flux-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/hailo-metadata.json → hailo-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/hdf5.js → hdf5.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/icon.png → icon.png} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/kann-metadata.json → kann-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/kann-schema.js → kann-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/keras-proto.js → keras-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/lasagne-metadata.json → lasagne-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/megengine-metadata.json → megengine-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/megengine-schema.js → megengine-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mlnet-metadata.json → mlnet-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mnn-metadata.json → mnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mslite-metadata.json → mslite-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mslite-schema.js → mslite-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/mxnet-metadata.json → mxnet-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/ncnn-metadata.json → ncnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/nnabla-metadata.json → nnabla-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/nnabla-proto.js → nnabla-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/nnc.js → nnc.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/om-metadata.json → om-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/om-proto.js → om-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onednn-metadata.json → onednn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/onnx-schema.js → onnx-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/openvino-metadata.json → openvino-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/paddle-metadata.json → paddle-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/paddle-schema.js → paddle-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/protobuf.js → protobuf.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/pytorch-schema.js → pytorch-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/qnn-metadata.json → qnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/rknn-metadata.json → rknn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/rknn-schema.js → rknn-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/sentencepiece-proto.js → sentencepiece-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tengine-metadata.json → tengine-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/text.js → text.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tflite-metadata.json → tflite-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tflite-schema.js → tflite-schema.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/tnn-metadata.json → tnn-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/torch-metadata.json → torch-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/uff-metadata.json → uff-metadata.json} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/uff-proto.js → uff-proto.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/worker.js → worker.js} +0 -0
- /qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/xmodel-proto.js → xmodel-proto.js} +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.6.0.dist-info}/LICENSE.pdf +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.6.0.dist-info}/WHEEL +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.6.0.dist-info}/entry_points.txt +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.6.0.dist-info}/top_level.txt +0 -0
|
@@ -1,145 +0,0 @@
|
|
|
1
|
-
''' NNabla metadata script '''
|
|
2
|
-
|
|
3
|
-
import json
|
|
4
|
-
import sys
|
|
5
|
-
import os
|
|
6
|
-
import yaml # pylint: disable=import-error
|
|
7
|
-
import mako.template # pylint: disable=import-error
|
|
8
|
-
|
|
9
|
-
def _write(path, content):
|
|
10
|
-
with open(path, 'w', encoding='utf-8') as file:
|
|
11
|
-
file.write(content)
|
|
12
|
-
|
|
13
|
-
def _read_yaml(path):
|
|
14
|
-
with open(path, 'r', encoding='utf-8') as file:
|
|
15
|
-
return yaml.safe_load(file)
|
|
16
|
-
|
|
17
|
-
def _metadata():
|
|
18
|
-
def parse_functions(function_info):
|
|
19
|
-
functions = []
|
|
20
|
-
for category_name, category in function_info.items():
|
|
21
|
-
for function_name, function_value in category.items():
|
|
22
|
-
function = {
|
|
23
|
-
'name': function_name,
|
|
24
|
-
'description': function_value['doc'].strip()
|
|
25
|
-
}
|
|
26
|
-
for input_name, input_value in function_value.get('inputs', {}).items():
|
|
27
|
-
function.setdefault('inputs', []).append({
|
|
28
|
-
'name': input_name,
|
|
29
|
-
'type': 'nnabla.Variable',
|
|
30
|
-
'option': 'optional' if input_value.get('optional', False) else None,
|
|
31
|
-
'list': input_value.get('variadic', False),
|
|
32
|
-
'description': input_value['doc'].strip()
|
|
33
|
-
})
|
|
34
|
-
for arg_name, arg_value in function_value.get('arguments', {}).items():
|
|
35
|
-
attribute = _attribute(arg_name, arg_value)
|
|
36
|
-
function.setdefault('attributes', []).append(attribute)
|
|
37
|
-
for output_name, output_value in function_value.get('outputs', {}).items():
|
|
38
|
-
function.setdefault('outputs', []).append({
|
|
39
|
-
'name': output_name,
|
|
40
|
-
'type': 'nnabla.Variable',
|
|
41
|
-
'list': output_value.get('variadic', False),
|
|
42
|
-
'description': output_value['doc'].strip()
|
|
43
|
-
})
|
|
44
|
-
if 'Pooling' in function_name:
|
|
45
|
-
function['category'] = 'Pool'
|
|
46
|
-
elif category_name == 'Neural Network Layer':
|
|
47
|
-
function['category'] = 'Layer'
|
|
48
|
-
elif category_name == 'Neural Network Activation Functions':
|
|
49
|
-
function['category'] = 'Activation'
|
|
50
|
-
elif category_name == 'Normalization':
|
|
51
|
-
function['category'] = 'Normalization'
|
|
52
|
-
elif category_name == 'Logical':
|
|
53
|
-
function['category'] = 'Logic'
|
|
54
|
-
elif category_name == 'Array Manipulation':
|
|
55
|
-
function['category'] = 'Shape'
|
|
56
|
-
functions.append(function)
|
|
57
|
-
return functions
|
|
58
|
-
def cleanup_functions(functions):
|
|
59
|
-
for function in functions:
|
|
60
|
-
for inp in function.get('inputs', []):
|
|
61
|
-
if inp['option'] is None:
|
|
62
|
-
inp.pop('option', None)
|
|
63
|
-
if not inp['list']:
|
|
64
|
-
inp.pop('list', None)
|
|
65
|
-
for output in function.get('outputs', []):
|
|
66
|
-
if not output['list']:
|
|
67
|
-
output.pop('list', None)
|
|
68
|
-
root = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
|
|
69
|
-
functions_yaml_path = os.path.join(root, \
|
|
70
|
-
'third_party', 'source', 'nnabla', 'build-tools', 'code_generator', 'functions.yaml')
|
|
71
|
-
function_info = _read_yaml(functions_yaml_path)
|
|
72
|
-
functions = parse_functions(function_info)
|
|
73
|
-
cleanup_functions(functions)
|
|
74
|
-
_write(os.path.join(root, 'source', 'nnabla-metadata.json'), json.dumps(functions, indent=2))
|
|
75
|
-
|
|
76
|
-
def _schema():
|
|
77
|
-
root = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
|
|
78
|
-
third_party_dir = os.path.join(root, 'third_party', 'source', 'nnabla')
|
|
79
|
-
tmpl_file = os.path.join(third_party_dir, 'src/nbla/proto/nnabla.proto.tmpl')
|
|
80
|
-
yaml_functions_path = os.path.join(third_party_dir, 'build-tools/code_generator/functions.yaml')
|
|
81
|
-
yaml_solvers_path = os.path.join(third_party_dir, 'build-tools/code_generator/solvers.yaml')
|
|
82
|
-
functions = _read_yaml(yaml_functions_path)
|
|
83
|
-
function_info = {k: v for _, category in functions.items() for k, v in category.items()}
|
|
84
|
-
solver_info = _read_yaml(yaml_solvers_path)
|
|
85
|
-
path = tmpl_file.replace('.tmpl', '')
|
|
86
|
-
template = mako.template.Template(text=None, filename=tmpl_file, preprocessor=None)
|
|
87
|
-
content = template.render(function_info=function_info, solver_info=solver_info)
|
|
88
|
-
content = content.replace('\r\n', '\n').replace('\r', '\n')
|
|
89
|
-
_write(path, content)
|
|
90
|
-
|
|
91
|
-
def _attribute(name, value): # pylint: disable=too-many-branches
|
|
92
|
-
attribute = {}
|
|
93
|
-
attribute['name'] = name
|
|
94
|
-
default = 'default' in value
|
|
95
|
-
if not default:
|
|
96
|
-
attribute['required'] = True
|
|
97
|
-
if value['type'] == 'float':
|
|
98
|
-
attribute['type'] = 'float32'
|
|
99
|
-
if default:
|
|
100
|
-
attribute['default'] = float(value['default'])
|
|
101
|
-
elif value['type'] == 'double':
|
|
102
|
-
attribute['type'] = 'float64'
|
|
103
|
-
if default:
|
|
104
|
-
attribute['default'] = float(value['default'])
|
|
105
|
-
elif value['type'] == 'bool':
|
|
106
|
-
attribute['type'] = 'boolean'
|
|
107
|
-
if default:
|
|
108
|
-
_ = value['default']
|
|
109
|
-
if isinstance(_, bool):
|
|
110
|
-
attribute['default'] = _
|
|
111
|
-
elif _ == 'True':
|
|
112
|
-
attribute['default'] = True
|
|
113
|
-
elif _ == 'False':
|
|
114
|
-
attribute['default'] = False
|
|
115
|
-
elif value['type'] == 'string':
|
|
116
|
-
attribute['type'] = 'string'
|
|
117
|
-
if default:
|
|
118
|
-
_ = value['default']
|
|
119
|
-
attribute['default'] = _.strip("'")
|
|
120
|
-
elif value['type'] == 'int64':
|
|
121
|
-
attribute['type'] = 'int64'
|
|
122
|
-
if default:
|
|
123
|
-
_ = value['default']
|
|
124
|
-
if isinstance(_, str) and not _.startswith('len') and _ != 'None':
|
|
125
|
-
attribute['default'] = int(_)
|
|
126
|
-
else:
|
|
127
|
-
attribute['default'] = _
|
|
128
|
-
elif value['type'] == 'repeated int64':
|
|
129
|
-
attribute['type'] = 'int64[]'
|
|
130
|
-
elif value['type'] == 'repeated float':
|
|
131
|
-
attribute['type'] = 'float32[]'
|
|
132
|
-
elif value['type'] == 'Shape':
|
|
133
|
-
attribute['type'] = 'shape'
|
|
134
|
-
if default and 'default' not in attribute:
|
|
135
|
-
attribute['default'] = value['default']
|
|
136
|
-
attribute['description'] = value['doc'].strip()
|
|
137
|
-
return attribute
|
|
138
|
-
|
|
139
|
-
def main(): # pylint: disable=missing-function-docstring
|
|
140
|
-
table = { 'metadata': _metadata, 'schema': _schema }
|
|
141
|
-
for command in sys.argv[1:]:
|
|
142
|
-
table[command]()
|
|
143
|
-
|
|
144
|
-
if __name__ == '__main__':
|
|
145
|
-
main()
|
|
@@ -1,26 +0,0 @@
|
|
|
1
|
-
#!/bin/bash
|
|
2
|
-
|
|
3
|
-
set -e
|
|
4
|
-
pushd $(cd $(dirname ${0})/..; pwd) > /dev/null
|
|
5
|
-
|
|
6
|
-
case "${OSTYPE}" in
|
|
7
|
-
msys*) python="winpty python";;
|
|
8
|
-
*) python="python3";;
|
|
9
|
-
esac
|
|
10
|
-
|
|
11
|
-
schema() {
|
|
12
|
-
echo "om schema"
|
|
13
|
-
[[ $(grep -U $'\x0D' ./source/om-proto.js) ]] && crlf=1
|
|
14
|
-
# https://github.com/Ascend/parser/blob/development/parser/proto/ge_ir.proto
|
|
15
|
-
node ./tools/protoc.js --binary --text --root om --out ./source/om-proto.js ./tools/om.proto
|
|
16
|
-
if [[ -n ${crlf} ]]; then
|
|
17
|
-
unix2dos --quiet --newfile ./source/om-proto.js ./source/om-proto.js
|
|
18
|
-
fi
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
while [ "$#" != 0 ]; do
|
|
22
|
-
command="$1" && shift
|
|
23
|
-
case "${command}" in
|
|
24
|
-
"schema") schema;;
|
|
25
|
-
esac
|
|
26
|
-
done
|
|
@@ -1,204 +0,0 @@
|
|
|
1
|
-
syntax = "proto3";
|
|
2
|
-
|
|
3
|
-
package ge.proto;
|
|
4
|
-
|
|
5
|
-
enum DataType
|
|
6
|
-
{
|
|
7
|
-
DT_UNDEFINED = 0; // Used to indicate a DataType field has not been set.
|
|
8
|
-
DT_FLOAT = 1; // float type
|
|
9
|
-
DT_FLOAT16 = 2; // fp16 type
|
|
10
|
-
DT_INT8 = 3; // int8 type
|
|
11
|
-
DT_UINT8 = 4; // uint8 type
|
|
12
|
-
DT_INT16 = 5; // int16 type
|
|
13
|
-
DT_UINT16 = 6; // uint16 type
|
|
14
|
-
DT_INT32 = 7; //
|
|
15
|
-
DT_INT64 = 8; // int64 type
|
|
16
|
-
DT_UINT32 = 9; // unsigned int32
|
|
17
|
-
DT_UINT64 = 10; // unsigned int64
|
|
18
|
-
DT_BOOL = 11; // bool type
|
|
19
|
-
DT_DOUBLE = 12; // double type
|
|
20
|
-
DT_STRING = 13; // string type
|
|
21
|
-
DT_DUAL_SUB_INT8 = 14; /**< dual output int8 type */
|
|
22
|
-
DT_DUAL_SUB_UINT8 = 15; /**< dual output uint8 type */
|
|
23
|
-
DT_COMPLEX64 = 16; // complex64 type
|
|
24
|
-
DT_COMPLEX128 = 17; // complex128 type
|
|
25
|
-
DT_QINT8 = 18; // qint8 type
|
|
26
|
-
DT_QINT16 = 19; // qint16 type
|
|
27
|
-
DT_QINT32 = 20; // qint32 type
|
|
28
|
-
DT_QUINT8 = 21; // quint8 type
|
|
29
|
-
DT_QUINT16 = 22; // quint16 type
|
|
30
|
-
DT_RESOURCE = 23; // resource type
|
|
31
|
-
DT_STRING_REF = 24; // string_ref type
|
|
32
|
-
DT_DUAL = 25; /**< dual output type */
|
|
33
|
-
DT_VARIANT = 26; // variant type
|
|
34
|
-
DT_BF16 = 27; // bf16 type
|
|
35
|
-
DT_INT4 = 28; // int4 type
|
|
36
|
-
DT_UINT1 = 29; // uint1 type
|
|
37
|
-
DT_INT2 = 30; // int2 type
|
|
38
|
-
DT_UINT2 = 31; // uint2 type
|
|
39
|
-
}
|
|
40
|
-
|
|
41
|
-
message AttrDef
|
|
42
|
-
{
|
|
43
|
-
message ListValue
|
|
44
|
-
{
|
|
45
|
-
enum ListValueType{
|
|
46
|
-
VT_LIST_NONE = 0;
|
|
47
|
-
VT_LIST_STRING = 1;
|
|
48
|
-
VT_LIST_INT = 2;
|
|
49
|
-
VT_LIST_FLOAT = 3;
|
|
50
|
-
VT_LIST_BOOL = 4;
|
|
51
|
-
VT_LIST_BYTES = 5;
|
|
52
|
-
VT_LIST_TENSOR_DESC = 6;
|
|
53
|
-
VT_LIST_TENSOR = 7;
|
|
54
|
-
VT_LIST_GRAPH = 8;
|
|
55
|
-
VT_LIST_NAMED_ATTRS = 9;
|
|
56
|
-
VT_LIST_DATA_TYPE = 10;
|
|
57
|
-
}
|
|
58
|
-
repeated bytes s = 2; // "list(string)"
|
|
59
|
-
repeated int64 i = 3; // "list(int)"
|
|
60
|
-
repeated float f = 4; // "list(float)"
|
|
61
|
-
repeated bool b = 5; // "list(bool)"
|
|
62
|
-
repeated bytes bt = 7;
|
|
63
|
-
repeated TensorDescriptor td = 8;
|
|
64
|
-
repeated TensorDef t = 9;
|
|
65
|
-
repeated GraphDef g = 10;
|
|
66
|
-
repeated NamedAttrs na = 11;
|
|
67
|
-
repeated int64 dt = 12; // list ge::DataType
|
|
68
|
-
|
|
69
|
-
ListValueType val_type = 20;
|
|
70
|
-
}
|
|
71
|
-
|
|
72
|
-
message ListListInt{
|
|
73
|
-
message ListInt{
|
|
74
|
-
repeated int64 list_i = 1; // list int
|
|
75
|
-
}
|
|
76
|
-
repeated ListInt list_list_i = 1; // list list int
|
|
77
|
-
}
|
|
78
|
-
|
|
79
|
-
message ListListFloat{
|
|
80
|
-
message ListFloat{
|
|
81
|
-
repeated float list_f = 1; // list float
|
|
82
|
-
}
|
|
83
|
-
repeated ListFloat list_list_f = 1; // list list float
|
|
84
|
-
}
|
|
85
|
-
|
|
86
|
-
oneof value
|
|
87
|
-
{
|
|
88
|
-
bytes s = 2; // "string"
|
|
89
|
-
int64 i = 3; // "int"
|
|
90
|
-
float f = 4; // "float"
|
|
91
|
-
bool b = 5; // "bool"
|
|
92
|
-
bytes bt = 7;
|
|
93
|
-
ListValue list = 1; // any "list(...)"
|
|
94
|
-
NamedAttrs func = 10; // Used to support attr nesting
|
|
95
|
-
TensorDescriptor td = 11; // GeTensorDesc type
|
|
96
|
-
TensorDef t = 12; // GeTensor type
|
|
97
|
-
GraphDef g = 13; // Graph type
|
|
98
|
-
ListListInt list_list_int = 14; // List List Int type
|
|
99
|
-
int64 dt = 15; // ge::DataType
|
|
100
|
-
ListListFloat list_list_float = 16; // List List Float type
|
|
101
|
-
}
|
|
102
|
-
}
|
|
103
|
-
|
|
104
|
-
// A list of attr names and their values. The whole list is attached
|
|
105
|
-
// with a string name. E.g., MatMul[T=float].
|
|
106
|
-
message NamedAttrs
|
|
107
|
-
{
|
|
108
|
-
string name = 1;
|
|
109
|
-
map<string, AttrDef> attr = 2;
|
|
110
|
-
}
|
|
111
|
-
|
|
112
|
-
// Shape / dimension description, using row-major order
|
|
113
|
-
message ShapeDef
|
|
114
|
-
{
|
|
115
|
-
repeated int64 dim = 1; // Size of each dimension
|
|
116
|
-
}
|
|
117
|
-
|
|
118
|
-
// Multidimensional data description
|
|
119
|
-
message TensorDescriptor
|
|
120
|
-
{
|
|
121
|
-
string name = 1; // Optional parameter, tensor name
|
|
122
|
-
|
|
123
|
-
DataType dtype = 2; // tensor datatype
|
|
124
|
-
ShapeDef shape = 3; // Shape / dimension
|
|
125
|
-
string layout = 4; // Tensor format, eg: "NCHW", "NHWC", "CHW", "ND"
|
|
126
|
-
|
|
127
|
-
bool has_out_attr = 9;
|
|
128
|
-
int64 size = 10;
|
|
129
|
-
int64 weight_size = 11;
|
|
130
|
-
bool reuse_input = 12;
|
|
131
|
-
bool output_tensor = 13;
|
|
132
|
-
string device_type = 14;
|
|
133
|
-
bool input_tensor =15;
|
|
134
|
-
int64 real_dim_cnt = 16;
|
|
135
|
-
int64 reuse_input_index = 17;
|
|
136
|
-
int64 data_offset = 18;
|
|
137
|
-
int64 cmps_size = 19;
|
|
138
|
-
string cmps_tab = 20;
|
|
139
|
-
int64 cmps_tab_offset = 21;
|
|
140
|
-
|
|
141
|
-
map<string, AttrDef> attr = 5; // Set of extra parameter fields
|
|
142
|
-
}
|
|
143
|
-
|
|
144
|
-
// GeTensor definition
|
|
145
|
-
message TensorDef
|
|
146
|
-
{
|
|
147
|
-
TensorDescriptor desc = 1; // Tensor description
|
|
148
|
-
bytes data = 2; // Tensor data
|
|
149
|
-
}
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
// Operator description
|
|
153
|
-
message OpDef
|
|
154
|
-
{
|
|
155
|
-
string name = 1; // name
|
|
156
|
-
string type = 2; // type
|
|
157
|
-
|
|
158
|
-
repeated string input = 5; // input original op name + outgoing index. op_name:index
|
|
159
|
-
|
|
160
|
-
map<string, AttrDef> attr = 10; // Set of operator parameter fields
|
|
161
|
-
|
|
162
|
-
bool has_out_attr = 20;
|
|
163
|
-
int64 id = 21;
|
|
164
|
-
int64 stream_id =22;
|
|
165
|
-
repeated string input_name = 23;
|
|
166
|
-
repeated string src_name = 24;
|
|
167
|
-
repeated int64 src_index = 25;
|
|
168
|
-
repeated string dst_name = 26;
|
|
169
|
-
repeated int64 dst_index = 27;
|
|
170
|
-
repeated int64 input_i = 28;
|
|
171
|
-
repeated int64 output_i = 29;
|
|
172
|
-
repeated int64 workspace = 30;
|
|
173
|
-
repeated int64 workspace_bytes = 31;
|
|
174
|
-
repeated bool is_input_const = 32;
|
|
175
|
-
repeated TensorDescriptor input_desc = 33;
|
|
176
|
-
repeated TensorDescriptor output_desc = 34;
|
|
177
|
-
repeated string subgraph_name = 35;
|
|
178
|
-
}
|
|
179
|
-
|
|
180
|
-
// Graph definition
|
|
181
|
-
message GraphDef
|
|
182
|
-
{
|
|
183
|
-
string name = 1; // name
|
|
184
|
-
|
|
185
|
-
repeated string input = 4; // Graph input
|
|
186
|
-
repeated string output = 5; // Graph output
|
|
187
|
-
|
|
188
|
-
repeated OpDef op = 6; // List of operators
|
|
189
|
-
|
|
190
|
-
map<string, AttrDef> attr = 11; // Extended field
|
|
191
|
-
}
|
|
192
|
-
|
|
193
|
-
// model definition
|
|
194
|
-
message ModelDef
|
|
195
|
-
{
|
|
196
|
-
string name = 1; // name
|
|
197
|
-
uint32 version = 2; // IR Proto verion
|
|
198
|
-
string custom_version = 3; // User model version number, passed in by user
|
|
199
|
-
|
|
200
|
-
repeated GraphDef graph = 7; // Graph definition,graph[0] represents the main diagram in modeldef
|
|
201
|
-
|
|
202
|
-
map<string, AttrDef> attr = 11; // Extended field
|
|
203
|
-
}
|
|
204
|
-
|
|
@@ -1,135 +0,0 @@
|
|
|
1
|
-
#!/bin/bash
|
|
2
|
-
|
|
3
|
-
set -e
|
|
4
|
-
pushd $(cd $(dirname ${0})/..; pwd) > /dev/null
|
|
5
|
-
|
|
6
|
-
case "${OSTYPE}" in
|
|
7
|
-
msys*) python="winpty python";;
|
|
8
|
-
*) python="python";;
|
|
9
|
-
esac
|
|
10
|
-
|
|
11
|
-
venv() {
|
|
12
|
-
env_dir=./third_party/env/onnx
|
|
13
|
-
[ -d "${env_dir}" ] || ${python} -m venv ${env_dir}
|
|
14
|
-
case "${OSTYPE}" in
|
|
15
|
-
msys*) source ${env_dir}/Scripts/activate;;
|
|
16
|
-
*) source ${env_dir}/bin/activate;;
|
|
17
|
-
esac
|
|
18
|
-
${python} -m pip install --quiet --upgrade pip setuptools wheel
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
clean() {
|
|
22
|
-
echo "onnx clean"
|
|
23
|
-
rm -rf "./third_party/env/onnx"
|
|
24
|
-
rm -rf "./third_party/source/onnx"
|
|
25
|
-
rm -rf "./third_party/source/onnxruntime"
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
sync() {
|
|
29
|
-
echo "onnx sync"
|
|
30
|
-
[ -d "./third_party/source/onnx" ] || git clone --quiet --recursive https://github.com/onnx/onnx.git "./third_party/source/onnx"
|
|
31
|
-
pushd "./third_party/source/onnx" > /dev/null
|
|
32
|
-
git pull --quiet --prune
|
|
33
|
-
git submodule sync --quiet
|
|
34
|
-
git submodule update --quiet --init --recursive
|
|
35
|
-
popd > /dev/null
|
|
36
|
-
mkdir -p "./third_party/source/onnxruntime/core/flatbuffers/schema"
|
|
37
|
-
curl --silent --location --output "./third_party/source/onnxruntime/core/flatbuffers/schema/ort.fbs" "https://github.com/microsoft/onnxruntime/raw/main/onnxruntime/core/flatbuffers/schema/ort.fbs"
|
|
38
|
-
}
|
|
39
|
-
|
|
40
|
-
install() {
|
|
41
|
-
echo "onnx install"
|
|
42
|
-
venv
|
|
43
|
-
protobuf_version=$(grep "protobuf" ./third_party/source/onnx/requirements.txt | cut -d'=' -f2)
|
|
44
|
-
protoc_dir="$(pwd)/third_party/bin/protobuf/v${protobuf_version}"
|
|
45
|
-
case "${OSTYPE}" in
|
|
46
|
-
linux*)
|
|
47
|
-
[ -x "$(command -v cmake)" ] || sudo apt install -y cmake
|
|
48
|
-
if [ ! -f "${protoc_dir}/bin/protoc" ]; then
|
|
49
|
-
rm -rf ${protoc_dir}
|
|
50
|
-
git clone --quiet -c advice.detachedHead=false --quiet --recursive --depth=1 --branch v${protobuf_version} https://github.com/protocolbuffers/protobuf.git ${protoc_dir}/src
|
|
51
|
-
pushd "${protoc_dir}/src/cmake" > /dev/null
|
|
52
|
-
cmake ../cmake -DCMAKE_CXX_FLAGS="-w" -Dprotobuf_BUILD_SHARED_LIBS=OFF -DCMAKE_INSTALL_PREFIX="..\.." -DCMAKE_INSTALL_SYSCONFDIR=/etc -DCMAKE_POSITION_INDEPENDENT_CODE=ON -Dprotobuf_BUILD_TESTS=OFF -DCMAKE_BUILD_TYPE=Release > /dev/null
|
|
53
|
-
make -j$(nproc) > /dev/null
|
|
54
|
-
make install > /dev/null
|
|
55
|
-
popd > /dev/null
|
|
56
|
-
fi
|
|
57
|
-
export PATH="${protoc_dir}\bin":${PATH}
|
|
58
|
-
;;
|
|
59
|
-
# darwin*)
|
|
60
|
-
# [ -x "$(command -v cmake)" ] || brew install cmake > /dev/null
|
|
61
|
-
# if [ ! -f "${protoc_dir}/bin/protoc" ]; then
|
|
62
|
-
# rm -rf ${protoc_dir}
|
|
63
|
-
# git clone --quiet -c advice.detachedHead=false --quiet --recursive --depth=1 --branch v${protobuf_version} https://github.com/protocolbuffers/protobuf.git ${protoc_dir}/src
|
|
64
|
-
# pushd "${protoc_dir}/src/cmake" > /dev/null
|
|
65
|
-
# cmake ../cmake -DCMAKE_CXX_FLAGS="-w" -Dprotobuf_BUILD_SHARED_LIBS=OFF -DCMAKE_INSTALL_PREFIX="..\.." -DCMAKE_INSTALL_SYSCONFDIR=/etc -DCMAKE_POSITION_INDEPENDENT_CODE=ON -Dprotobuf_BUILD_TESTS=OFF -DCMAKE_BUILD_TYPE=Release > /dev/null
|
|
66
|
-
# make > /dev/null
|
|
67
|
-
# make install > /dev/null
|
|
68
|
-
# popd > /dev/null
|
|
69
|
-
# fi
|
|
70
|
-
# export PATH="${protoc_dir}\bin":${PATH}
|
|
71
|
-
# ;;
|
|
72
|
-
msys*)
|
|
73
|
-
# winget install -e --id=Microsoft.VisualStudio.2019.BuildTools --override "--passive --wait --add Microsoft.VisualStudio.Workload.VCTools;includeRecommended"
|
|
74
|
-
programfiles_x86_dir=$(env | grep "^ProgramFiles(x86)=" | cut -d '=' -f 2)
|
|
75
|
-
build_tools_dir="${programfiles_x86_dir}\Microsoft Visual Studio\2019\BuildTools"
|
|
76
|
-
cmake_dir="${build_tools_dir}\Common7\IDE\CommonExtensions\Microsoft\CMake\CMake\bin"
|
|
77
|
-
protobuf="protobuf==${protobuf_version}"
|
|
78
|
-
${python} -m pip install --quiet --upgrade ${protobuf}
|
|
79
|
-
export PATH="$(cygpath -u "${cmake_dir}")":${PATH}
|
|
80
|
-
msbuild_dir="${build_tools_dir}\MSBuild\Current\Bin"
|
|
81
|
-
if [ ! -f "${protoc_dir}/bin/protoc.exe" ]; then
|
|
82
|
-
rm -rf ${protoc_dir}
|
|
83
|
-
git clone --quiet -c advice.detachedHead=false -b v${protobuf_version} https://github.com/protocolbuffers/protobuf.git ${protoc_dir}/src
|
|
84
|
-
pushd "${protoc_dir}/src/cmake" > /dev/null
|
|
85
|
-
"${cmake_dir}\cmake.exe" -G "Visual Studio 16 2019" -A x64 -Dprotobuf_MSVC_STATIC_RUNTIME=OFF -Dprotobuf_BUILD_SHARED_LIBS=OFF -Dprotobuf_BUILD_TESTS=OFF -Dprotobuf_BUILD_EXAMPLES=OFF -DCMAKE_INSTALL_PREFIX="..\..\.." . > /dev/null
|
|
86
|
-
"${msbuild_dir}\MSBuild.exe" protobuf.sln //m //p:Configuration=Release > /dev/null
|
|
87
|
-
"${msbuild_dir}\MSBuild.exe" INSTALL.vcxproj //p:Configuration=Release > /dev/null
|
|
88
|
-
popd > /dev/null
|
|
89
|
-
fi
|
|
90
|
-
export PATH="${protoc_dir}\bin":${PATH}
|
|
91
|
-
export USE_MSVC_STATIC_RUNTIME=0
|
|
92
|
-
;;
|
|
93
|
-
esac
|
|
94
|
-
${python} -m pip install --quiet "./third_party/source/onnx"
|
|
95
|
-
${python} -m pip install --quiet --upgrade packaging flatbuffers protobuf numpy sympy
|
|
96
|
-
${python} -m pip install --quiet --upgrade onnxruntime --pre --extra-index https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/
|
|
97
|
-
deactivate
|
|
98
|
-
}
|
|
99
|
-
|
|
100
|
-
schema() {
|
|
101
|
-
echo "onnx schema"
|
|
102
|
-
[[ $(grep -U $'\x0D' ./source/onnx-proto.js) ]] && crlf=1
|
|
103
|
-
node ./tools/protoc.js --binary --text --json --root onnx --out ./source/onnx-proto.js --path ./third_party/source/onnx onnx/onnx-ml.proto onnx/onnx-operators-ml.proto
|
|
104
|
-
if [[ -n ${crlf} ]]; then
|
|
105
|
-
unix2dos --quiet --newfile ./source/onnx-proto.js ./source/onnx-proto.js
|
|
106
|
-
fi
|
|
107
|
-
[[ $(grep -U $'\x0D' ./source/onnx-schema.js) ]] && crlf=1
|
|
108
|
-
node ./tools/flatc.js --root ort --out ./source/onnx-schema.js ./third_party/source/onnxruntime/core/flatbuffers/schema/ort.fbs
|
|
109
|
-
if [[ -n ${crlf} ]]; then
|
|
110
|
-
unix2dos --quiet --newfile ./source/onnx-schema.js ./source/onnx-schema.js
|
|
111
|
-
fi
|
|
112
|
-
}
|
|
113
|
-
|
|
114
|
-
metadata() {
|
|
115
|
-
echo "onnx metadata"
|
|
116
|
-
[[ $(grep -U $'\x0D' ./source/onnx-metadata.json) ]] && crlf=1
|
|
117
|
-
venv
|
|
118
|
-
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
|
|
119
|
-
${python} ./tools/onnx_script.py
|
|
120
|
-
deactivate
|
|
121
|
-
if [[ -n ${crlf} ]]; then
|
|
122
|
-
unix2dos --quiet --newfile ./source/onnx-metadata.json ./source/onnx-metadata.json
|
|
123
|
-
fi
|
|
124
|
-
}
|
|
125
|
-
|
|
126
|
-
while [ "$#" != 0 ]; do
|
|
127
|
-
command="$1" && shift
|
|
128
|
-
case "${command}" in
|
|
129
|
-
"clean") clean;;
|
|
130
|
-
"sync") sync;;
|
|
131
|
-
"install") install;;
|
|
132
|
-
"schema") schema;;
|
|
133
|
-
"metadata") metadata;;
|
|
134
|
-
esac
|
|
135
|
-
done
|