qairt-visualizer 0.5.1__py3-none-win_amd64.whl → 0.7.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 +44 -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/i18n/workspaces/en.json +3 -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-JPD5T7FP.js +120 -0
- qairt_visualizer/core/ui/dist/browser/chunk-L3QJ7DR2.js +1 -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/eaix.js +24 -0
- 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-EVI665MB.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-2Z5PB4WZ.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} +1406 -1115
- qairt_visualizer/core/ui/dist/browser/{assets/netron_plugin/source/weka.js → weka.js} +3 -4
- qairt_visualizer/core/ui/dist/browser/worker-EHYFMTM7.js +1 -0
- qairt_visualizer/core/ui/dist/browser/worker-KX5DA6GL.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.7.0.dist-info}/METADATA +1 -1
- qairt_visualizer-0.7.0.dist-info/RECORD +265 -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.7.0.dist-info}/LICENSE.pdf +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.7.0.dist-info}/WHEEL +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.7.0.dist-info}/entry_points.txt +0 -0
- {qairt_visualizer-0.5.1.dist-info → qairt_visualizer-0.7.0.dist-info}/top_level.txt +0 -0
|
@@ -1,39 +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
|
-
clean() {
|
|
12
|
-
echo "sentencepiece clean"
|
|
13
|
-
rm -rf "./third_party/source/sentencepiece"
|
|
14
|
-
}
|
|
15
|
-
|
|
16
|
-
sync() {
|
|
17
|
-
echo "sentencepiece sync"
|
|
18
|
-
mkdir -p "./third_party/source/sentencepiece/src"
|
|
19
|
-
curl --silent --location --output "./third_party/source/sentencepiece/src/sentencepiece_model.proto" "https://github.com/google/sentencepiece/raw/master/src/sentencepiece_model.proto"
|
|
20
|
-
curl --silent --location --output "./third_party/source/sentencepiece/src/sentencepiece.proto" "https://github.com/google/sentencepiece/raw/master/src/sentencepiece.proto"
|
|
21
|
-
}
|
|
22
|
-
|
|
23
|
-
schema() {
|
|
24
|
-
echo "sentencepiece schema"
|
|
25
|
-
[[ $(grep -U $'\x0D' ./source/sentencepiece-proto.js) ]] && crlf=1
|
|
26
|
-
node ./tools/protoc.js --binary --text --root sentencepiece --out ./source/sentencepiece-proto.js --path ./third_party/source/sentencepiece src/sentencepiece_model.proto
|
|
27
|
-
if [[ -n ${crlf} ]]; then
|
|
28
|
-
unix2dos --quiet --newfile ./source/sentencepiece-proto.js ./source/sentencepiece-proto.js
|
|
29
|
-
fi
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
while [ "$#" != 0 ]; do
|
|
33
|
-
command="$1" && shift
|
|
34
|
-
case "${command}" in
|
|
35
|
-
"clean") clean;;
|
|
36
|
-
"sync") sync;;
|
|
37
|
-
"schema") schema;;
|
|
38
|
-
esac
|
|
39
|
-
done
|
|
@@ -1,65 +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/scikit-learn
|
|
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 "sklearn clean"
|
|
23
|
-
rm -rf "./third_party/env/scikit-learn"
|
|
24
|
-
rm -rf "./third_party/source/scikit-learn"
|
|
25
|
-
}
|
|
26
|
-
|
|
27
|
-
sync() {
|
|
28
|
-
echo "sklearn sync"
|
|
29
|
-
[ -d "./third_party/source/scikit-learn" ] || git clone --quiet --recursive "https://github.com/scikit-learn/scikit-learn.git" "./third_party/source/scikit-learn"
|
|
30
|
-
pushd "./third_party/source/scikit-learn" > /dev/null
|
|
31
|
-
git pull --quiet --prune
|
|
32
|
-
git submodule sync --quiet
|
|
33
|
-
git submodule update --quiet --init --recursive
|
|
34
|
-
popd > /dev/null
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
install() {
|
|
38
|
-
echo "sklearn install"
|
|
39
|
-
venv
|
|
40
|
-
${python} -m pip install --quiet scipy
|
|
41
|
-
${python} -m pip install --quiet scikit-learn --pre --extra-index https://pypi.anaconda.org/scipy-wheels-nightly/simple
|
|
42
|
-
deactivate
|
|
43
|
-
}
|
|
44
|
-
|
|
45
|
-
metadata() {
|
|
46
|
-
echo "sklearn metadata"
|
|
47
|
-
[[ $(grep -U $'\x0D' ./source/sklearn-metadata.json) ]] && crlf=1
|
|
48
|
-
venv
|
|
49
|
-
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
|
|
50
|
-
${python} ./tools/sklearn_script.py
|
|
51
|
-
deactivate
|
|
52
|
-
if [[ -n ${crlf} ]]; then
|
|
53
|
-
unix2dos --quiet --newfile ./source/sklearn-metadata.json ./source/sklearn-metadata.json
|
|
54
|
-
fi
|
|
55
|
-
}
|
|
56
|
-
|
|
57
|
-
while [ "$#" != 0 ]; do
|
|
58
|
-
command="$1" && shift
|
|
59
|
-
case "${command}" in
|
|
60
|
-
"clean") clean;;
|
|
61
|
-
"sync") sync;;
|
|
62
|
-
"install") install;;
|
|
63
|
-
"metadata") metadata;;
|
|
64
|
-
esac
|
|
65
|
-
done
|
|
@@ -1,126 +0,0 @@
|
|
|
1
|
-
''' scikit-learn metadata script '''
|
|
2
|
-
|
|
3
|
-
import json
|
|
4
|
-
import os
|
|
5
|
-
import pydoc
|
|
6
|
-
import re
|
|
7
|
-
|
|
8
|
-
def _split_docstring(value):
|
|
9
|
-
headers = {}
|
|
10
|
-
current_header = ''
|
|
11
|
-
current_lines = []
|
|
12
|
-
lines = value.split('\n')
|
|
13
|
-
index = 0
|
|
14
|
-
while index < len(lines):
|
|
15
|
-
if index + 1 < len(lines) and len(lines[index + 1].strip(' ')) > 0 and \
|
|
16
|
-
len(lines[index + 1].strip(' ').strip('-')) == 0:
|
|
17
|
-
headers[current_header] = current_lines
|
|
18
|
-
current_header = lines[index].strip(' ')
|
|
19
|
-
current_lines = []
|
|
20
|
-
index = index + 1
|
|
21
|
-
else:
|
|
22
|
-
current_lines.append(lines[index])
|
|
23
|
-
index = index + 1
|
|
24
|
-
headers[current_header] = current_lines
|
|
25
|
-
return headers
|
|
26
|
-
|
|
27
|
-
def _update_description(schema, lines):
|
|
28
|
-
if len(''.join(lines).strip(' ')) > 0:
|
|
29
|
-
for i, value in enumerate(lines):
|
|
30
|
-
lines[i] = value.lstrip(' ')
|
|
31
|
-
schema['description'] = '\n'.join(lines)
|
|
32
|
-
|
|
33
|
-
def _attribute_value(attribute_type, attribute_value):
|
|
34
|
-
if attribute_value in ('None', 'np.finfo(float).eps'):
|
|
35
|
-
return None
|
|
36
|
-
if attribute_type in ('float32', 'int32', 'boolean', 'string'):
|
|
37
|
-
if attribute_value in ("'auto'", '"auto"') or attribute_type == 'string':
|
|
38
|
-
return attribute_value.strip("'").strip('"')
|
|
39
|
-
if attribute_type == 'float32':
|
|
40
|
-
return float(attribute_value)
|
|
41
|
-
if attribute_type == 'int32':
|
|
42
|
-
return int(attribute_value)
|
|
43
|
-
if attribute_type == 'boolean':
|
|
44
|
-
if attribute_value in ('True', 'False'):
|
|
45
|
-
return attribute_value == 'True'
|
|
46
|
-
raise ValueError("Unknown boolean default value '" + str(attribute_value) + "'.")
|
|
47
|
-
if attribute_type:
|
|
48
|
-
raise ValueError("Unknown default type '" + attribute_type + "'.")
|
|
49
|
-
return attribute_value.strip("'")
|
|
50
|
-
|
|
51
|
-
def _find_attribute(schema, name):
|
|
52
|
-
schema.setdefault('attributes', [])
|
|
53
|
-
attribute = next((_ for _ in schema['attributes'] if _['name'] == name), None)
|
|
54
|
-
if not attribute:
|
|
55
|
-
attribute = { 'name': name }
|
|
56
|
-
schema['attributes'].append(attribute)
|
|
57
|
-
return attribute
|
|
58
|
-
|
|
59
|
-
def _update_attributes(schema, lines):
|
|
60
|
-
while len(lines) > 0:
|
|
61
|
-
line = lines.pop(0)
|
|
62
|
-
match = re.match(r'\s*(\w*)\s*:\s*(.*)\s*', line)
|
|
63
|
-
if not match:
|
|
64
|
-
raise SyntaxError("Expected ':' in parameter.")
|
|
65
|
-
name = match.group(1)
|
|
66
|
-
line = match.group(2)
|
|
67
|
-
attribute = _find_attribute(schema, name)
|
|
68
|
-
match = re.match(r'(.*),\s*default=(.*)\s*', line)
|
|
69
|
-
default_value = None
|
|
70
|
-
if match:
|
|
71
|
-
line = match.group(1)
|
|
72
|
-
default_value = match.group(2)
|
|
73
|
-
attribute_types = {
|
|
74
|
-
'float': 'float32',
|
|
75
|
-
'boolean': 'boolean',
|
|
76
|
-
'bool': 'boolean',
|
|
77
|
-
'str': 'string',
|
|
78
|
-
'string': 'string',
|
|
79
|
-
'int': 'int32',
|
|
80
|
-
'integer': 'int32'
|
|
81
|
-
}
|
|
82
|
-
attribute_type = attribute_types.get(line, None)
|
|
83
|
-
if default_value:
|
|
84
|
-
attribute['default'] = _attribute_value(attribute_type, default_value)
|
|
85
|
-
description = []
|
|
86
|
-
while len(lines) > 0 and (len(lines[0].strip(' ')) == 0 or lines[0].startswith(' ')):
|
|
87
|
-
line = lines.pop(0).lstrip(' ')
|
|
88
|
-
description.append(line)
|
|
89
|
-
attribute['description'] = '\n'.join(description)
|
|
90
|
-
|
|
91
|
-
def _metadata():
|
|
92
|
-
json_file = os.path.join(os.path.dirname(__file__), '../source/sklearn-metadata.json')
|
|
93
|
-
with open(json_file, 'r', encoding='utf-8') as file:
|
|
94
|
-
json_root = json.loads(file.read())
|
|
95
|
-
|
|
96
|
-
for schema in json_root:
|
|
97
|
-
name = schema['name']
|
|
98
|
-
skip_modules = [
|
|
99
|
-
'lightgbm.',
|
|
100
|
-
'sklearn.svm.classes',
|
|
101
|
-
'sklearn.ensemble.forest.',
|
|
102
|
-
'sklearn.ensemble.weight_boosting.',
|
|
103
|
-
'sklearn.neural_network.multilayer_perceptron.',
|
|
104
|
-
'sklearn.tree.tree.'
|
|
105
|
-
]
|
|
106
|
-
if not any(name.startswith(module) for module in skip_modules):
|
|
107
|
-
class_definition = pydoc.locate(name)
|
|
108
|
-
if not class_definition:
|
|
109
|
-
raise KeyError('\'' + name + '\' not found.')
|
|
110
|
-
docstring = class_definition.__doc__
|
|
111
|
-
if not docstring:
|
|
112
|
-
raise Exception('\'' + name + '\' missing __doc__.') # pylint: disable=broad-exception-raised
|
|
113
|
-
headers = _split_docstring(docstring)
|
|
114
|
-
if '' in headers:
|
|
115
|
-
_update_description(schema, headers[''])
|
|
116
|
-
if 'Parameters' in headers:
|
|
117
|
-
_update_attributes(schema, headers['Parameters'])
|
|
118
|
-
|
|
119
|
-
with open(json_file, 'w', encoding='utf-8') as file:
|
|
120
|
-
file.write(json.dumps(json_root, sort_keys=False, indent=2))
|
|
121
|
-
|
|
122
|
-
def main(): # pylint: disable=missing-function-docstring
|
|
123
|
-
_metadata()
|
|
124
|
-
|
|
125
|
-
if __name__ == '__main__':
|
|
126
|
-
main()
|
|
@@ -1,129 +0,0 @@
|
|
|
1
|
-
''' scikit-learn metadata script '''
|
|
2
|
-
|
|
3
|
-
import json
|
|
4
|
-
import os
|
|
5
|
-
import pydoc
|
|
6
|
-
import re
|
|
7
|
-
import sys
|
|
8
|
-
|
|
9
|
-
def _split_docstring(value):
|
|
10
|
-
headers = {}
|
|
11
|
-
current_header = ''
|
|
12
|
-
current_lines = []
|
|
13
|
-
lines = value.split('\n')
|
|
14
|
-
index = 0
|
|
15
|
-
while index < len(lines):
|
|
16
|
-
if index + 1 < len(lines) and len(lines[index + 1].strip(' ')) > 0 and \
|
|
17
|
-
len(lines[index + 1].strip(' ').strip('-')) == 0:
|
|
18
|
-
headers[current_header] = current_lines
|
|
19
|
-
current_header = lines[index].strip(' ')
|
|
20
|
-
current_lines = []
|
|
21
|
-
index = index + 1
|
|
22
|
-
else:
|
|
23
|
-
current_lines.append(lines[index])
|
|
24
|
-
index = index + 1
|
|
25
|
-
headers[current_header] = current_lines
|
|
26
|
-
return headers
|
|
27
|
-
|
|
28
|
-
def _update_description(schema, lines):
|
|
29
|
-
if len(''.join(lines).strip(' ')) > 0:
|
|
30
|
-
for i, value in enumerate(lines):
|
|
31
|
-
lines[i] = value.lstrip(' ')
|
|
32
|
-
schema['description'] = '\n'.join(lines)
|
|
33
|
-
|
|
34
|
-
def _attribute_value(attribute_type, attribute_value):
|
|
35
|
-
if attribute_value in ('None', 'np.finfo(float).eps'):
|
|
36
|
-
return None
|
|
37
|
-
if attribute_type in ('float32', 'int32', 'boolean', 'string'):
|
|
38
|
-
if attribute_value in ("'auto'", '"auto"') or attribute_type == 'string':
|
|
39
|
-
return attribute_value.strip("'").strip('"')
|
|
40
|
-
if attribute_type == 'float32':
|
|
41
|
-
return float(attribute_value)
|
|
42
|
-
if attribute_type == 'int32':
|
|
43
|
-
return int(attribute_value)
|
|
44
|
-
if attribute_type == 'boolean':
|
|
45
|
-
if attribute_value in ('True', 'False'):
|
|
46
|
-
return attribute_value == 'True'
|
|
47
|
-
raise ValueError("Unknown boolean default value '" + str(attribute_value) + "'.")
|
|
48
|
-
if attribute_type:
|
|
49
|
-
raise ValueError("Unknown default type '" + attribute_type + "'.")
|
|
50
|
-
return attribute_value.strip("'")
|
|
51
|
-
|
|
52
|
-
def _find_attribute(schema, name):
|
|
53
|
-
schema.setdefault('attributes', [])
|
|
54
|
-
attribute = next((_ for _ in schema['attributes'] if _['name'] == name), None)
|
|
55
|
-
if not attribute:
|
|
56
|
-
attribute = { 'name': name }
|
|
57
|
-
schema['attributes'].append(attribute)
|
|
58
|
-
return attribute
|
|
59
|
-
|
|
60
|
-
def _update_attributes(schema, lines):
|
|
61
|
-
doc_indent = ' ' if sys.version_info[:2] >= (3, 13) else ' '
|
|
62
|
-
while len(lines) > 0:
|
|
63
|
-
line = lines.pop(0)
|
|
64
|
-
match = re.match(r'\s*(\w*)\s*:\s*(.*)\s*', line)
|
|
65
|
-
if not match:
|
|
66
|
-
raise SyntaxError("Expected ':' in parameter.")
|
|
67
|
-
name = match.group(1)
|
|
68
|
-
line = match.group(2)
|
|
69
|
-
attribute = _find_attribute(schema, name)
|
|
70
|
-
match = re.match(r'(.*),\s*default=(.*)\s*', line)
|
|
71
|
-
default_value = None
|
|
72
|
-
if match:
|
|
73
|
-
line = match.group(1)
|
|
74
|
-
default_value = match.group(2)
|
|
75
|
-
attribute_types = {
|
|
76
|
-
'float': 'float32',
|
|
77
|
-
'boolean': 'boolean',
|
|
78
|
-
'bool': 'boolean',
|
|
79
|
-
'str': 'string',
|
|
80
|
-
'string': 'string',
|
|
81
|
-
'int': 'int32',
|
|
82
|
-
'integer': 'int32'
|
|
83
|
-
}
|
|
84
|
-
attribute_type = attribute_types.get(line, None)
|
|
85
|
-
if default_value:
|
|
86
|
-
attribute['default'] = _attribute_value(attribute_type, default_value)
|
|
87
|
-
description = []
|
|
88
|
-
while len(lines) > 0 and (len(lines[0].strip(' ')) == 0 or lines[0].startswith(doc_indent)):
|
|
89
|
-
line = lines.pop(0).lstrip(' ')
|
|
90
|
-
description.append(line)
|
|
91
|
-
attribute['description'] = '\n'.join(description)
|
|
92
|
-
|
|
93
|
-
def _metadata():
|
|
94
|
-
root_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
95
|
-
json_file = os.path.join(root_dir, 'source', 'sklearn-metadata.json')
|
|
96
|
-
with open(json_file, 'r', encoding='utf-8') as file:
|
|
97
|
-
json_root = json.loads(file.read())
|
|
98
|
-
|
|
99
|
-
for schema in json_root:
|
|
100
|
-
name = schema['name']
|
|
101
|
-
skip_modules = [
|
|
102
|
-
'lightgbm.',
|
|
103
|
-
'sklearn.svm.classes',
|
|
104
|
-
'sklearn.ensemble.forest.',
|
|
105
|
-
'sklearn.ensemble.weight_boosting.',
|
|
106
|
-
'sklearn.neural_network.multilayer_perceptron.',
|
|
107
|
-
'sklearn.tree.tree.'
|
|
108
|
-
]
|
|
109
|
-
if not any(name.startswith(module) for module in skip_modules):
|
|
110
|
-
class_definition = pydoc.locate(name)
|
|
111
|
-
if not class_definition:
|
|
112
|
-
raise KeyError('\'' + name + '\' not found.')
|
|
113
|
-
docstring = class_definition.__doc__
|
|
114
|
-
if not docstring:
|
|
115
|
-
raise Exception('\'' + name + '\' missing __doc__.') # pylint: disable=broad-exception-raised
|
|
116
|
-
headers = _split_docstring(docstring)
|
|
117
|
-
if '' in headers:
|
|
118
|
-
_update_description(schema, headers[''])
|
|
119
|
-
if 'Parameters' in headers:
|
|
120
|
-
_update_attributes(schema, headers['Parameters'])
|
|
121
|
-
|
|
122
|
-
with open(json_file, 'w', encoding='utf-8') as file:
|
|
123
|
-
file.write(json.dumps(json_root, sort_keys=False, indent=2))
|
|
124
|
-
|
|
125
|
-
def main(): # pylint: disable=missing-function-docstring
|
|
126
|
-
_metadata()
|
|
127
|
-
|
|
128
|
-
if __name__ == '__main__':
|
|
129
|
-
main()
|
|
@@ -1,97 +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/tensorflow
|
|
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 requests
|
|
19
|
-
}
|
|
20
|
-
|
|
21
|
-
clean() {
|
|
22
|
-
echo "tf clean"
|
|
23
|
-
rm -rf "./third_party/env/tensorflow"
|
|
24
|
-
rm -rf "./third_party/source/tensorflow"
|
|
25
|
-
rm -rf "./third_party/source/tflite-support"
|
|
26
|
-
}
|
|
27
|
-
|
|
28
|
-
sync() {
|
|
29
|
-
echo "tf sync"
|
|
30
|
-
[ -d "./third_party/source/tensorflow" ] || git clone --quiet https://github.com/tensorflow/tensorflow.git "./third_party/source/tensorflow"
|
|
31
|
-
git -C "./third_party/source/tensorflow" pull --quiet --prune
|
|
32
|
-
mkdir -p "./third_party/source/tflite-support/tensorflow_lite_support/metadata"
|
|
33
|
-
curl --silent --location --output "./third_party/source/tflite-support/tensorflow_lite_support/metadata/metadata_schema.fbs" "https://github.com/tensorflow/tflite-support/raw/master/tensorflow_lite_support/metadata/metadata_schema.fbs"
|
|
34
|
-
}
|
|
35
|
-
|
|
36
|
-
install() {
|
|
37
|
-
echo "tf install"
|
|
38
|
-
venv
|
|
39
|
-
${python} -m pip install --quiet --upgrade wheel
|
|
40
|
-
${python} -m pip install --quiet --upgrade tf-nightly > /dev/null 2>&1 || true
|
|
41
|
-
deactivate
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
schema() {
|
|
45
|
-
echo "tf schema"
|
|
46
|
-
[[ $(grep -U $'\x0D' ./source/tf-proto.js) ]] && crlf=1
|
|
47
|
-
node ./tools/protoc.js --binary --text --json --root tf --out ./source/tf-proto.js --path ./third_party/source/tensorflow --path ./third_party/source/tensorflow/third_party/xla tensorflow/core/protobuf/saved_model.proto tensorflow/core/protobuf/tensor_bundle.proto tensorflow/core/util/saved_tensor_slice.proto tensorflow/core/util/event.proto tensorflow/core/protobuf/config.proto tensorflow/core/util/memmapped_file_system.proto tensorflow/core/protobuf/fingerprint.proto
|
|
48
|
-
if [[ -n ${crlf} ]]; then
|
|
49
|
-
unix2dos --quiet --newfile ./source/tf-proto.js ./source/tf-proto.js
|
|
50
|
-
fi
|
|
51
|
-
echo "tflite schema"
|
|
52
|
-
[[ $(grep -U $'\x0D' ./source/tflite-schema.js) ]] && crlf=1
|
|
53
|
-
temp1=$(mktemp -d)
|
|
54
|
-
temp2=$(mktemp -d)
|
|
55
|
-
node ./tools/flatc.js --text --root tflite --out ./source/tflite-schema.js ./third_party/source/tensorflow/tensorflow/compiler/mlir/lite/schema/schema.fbs ./third_party/source/tflite-support/tensorflow_lite_support/metadata/metadata_schema.fbs
|
|
56
|
-
if [[ -n ${crlf} ]]; then
|
|
57
|
-
unix2dos --quiet --newfile ./source/tflite-schema.js ./source/tflite-schema.js
|
|
58
|
-
fi
|
|
59
|
-
echo "keras schema"
|
|
60
|
-
[[ $(grep -U $'\x0D' ./source/keras-proto.js) ]] && crlf=1
|
|
61
|
-
node ./tools/protoc.js --binary --text --root tf --out ./source/keras-proto.js --path ./third_party/source/tensorflow --path ./third_party/source/tensorflow/third_party/xla/third_party/tsl tensorflow/python/keras/protobuf/saved_metadata.proto
|
|
62
|
-
if [[ -n ${crlf} ]]; then
|
|
63
|
-
unix2dos --quiet --newfile ./source/keras-proto.js ./source/keras-proto.js
|
|
64
|
-
fi
|
|
65
|
-
}
|
|
66
|
-
|
|
67
|
-
metadata() {
|
|
68
|
-
echo "tflite metadata"
|
|
69
|
-
export TF_CPP_MIN_LOG_LEVEL=2
|
|
70
|
-
if [[ $(grep -U $'\x0D' ./source/tflite-metadata.json) ]]; then crlf=1; else crlf=; fi
|
|
71
|
-
node ./tools/tflite-script.js
|
|
72
|
-
if [[ -n ${crlf} ]]; then
|
|
73
|
-
unix2dos --quiet --newfile ./source/tflite-metadata.json ./source/tflite-metadata.json
|
|
74
|
-
fi
|
|
75
|
-
echo "tf metadata"
|
|
76
|
-
venv
|
|
77
|
-
if [[ $(grep -U $'\x0D' ./source/tf-metadata.json) ]]; then crlf=1; else crlf=; fi
|
|
78
|
-
export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
|
|
79
|
-
if pip show "tf-nightly" > /dev/null 2>&1; then
|
|
80
|
-
${python} ./tools/tf_script.py
|
|
81
|
-
fi
|
|
82
|
-
if [[ -n ${crlf} ]]; then
|
|
83
|
-
unix2dos --quiet --newfile ./source/tf-metadata.json ./source/tf-metadata.json
|
|
84
|
-
fi
|
|
85
|
-
deactivate
|
|
86
|
-
}
|
|
87
|
-
|
|
88
|
-
while [ "$#" != 0 ]; do
|
|
89
|
-
command="$1" && shift
|
|
90
|
-
case "${command}" in
|
|
91
|
-
"clean") clean;;
|
|
92
|
-
"sync") sync;;
|
|
93
|
-
"install") install;;
|
|
94
|
-
"schema") schema;;
|
|
95
|
-
"metadata") metadata;;
|
|
96
|
-
esac
|
|
97
|
-
done
|