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,422 +0,0 @@
|
|
|
1
|
-
''' TensorFlow Metadata Script '''
|
|
2
|
-
|
|
3
|
-
import json
|
|
4
|
-
import logging
|
|
5
|
-
import os
|
|
6
|
-
import re
|
|
7
|
-
import sys
|
|
8
|
-
import google.protobuf # pylint: disable=import-error
|
|
9
|
-
|
|
10
|
-
logging.getLogger('tensorflow').setLevel(logging.ERROR)
|
|
11
|
-
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
|
|
12
|
-
|
|
13
|
-
dup_stderr = os.dup(sys.stderr.fileno())
|
|
14
|
-
null = os.open(os.devnull, os.O_WRONLY)
|
|
15
|
-
os.dup2(null, sys.stderr.fileno())
|
|
16
|
-
os.close(null)
|
|
17
|
-
|
|
18
|
-
from tensorflow.core.framework import api_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position
|
|
19
|
-
from tensorflow.core.framework import op_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position
|
|
20
|
-
from tensorflow.core.framework import types_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position
|
|
21
|
-
|
|
22
|
-
os.dup2(dup_stderr, sys.stderr.fileno())
|
|
23
|
-
os.close(dup_stderr)
|
|
24
|
-
|
|
25
|
-
def _read(path):
|
|
26
|
-
with open(path, 'r', encoding='utf-8') as file:
|
|
27
|
-
return file.read()
|
|
28
|
-
|
|
29
|
-
def _write(path, content):
|
|
30
|
-
with open(path, 'w', encoding='utf-8') as file:
|
|
31
|
-
file.write(content)
|
|
32
|
-
|
|
33
|
-
def _find_multiline(line, colon):
|
|
34
|
-
if colon == -1:
|
|
35
|
-
return None
|
|
36
|
-
line = line[colon+1:]
|
|
37
|
-
while line.startswith(' '):
|
|
38
|
-
line = line[1:]
|
|
39
|
-
if line.startswith('<<'):
|
|
40
|
-
line = line[2:]
|
|
41
|
-
return line
|
|
42
|
-
return None
|
|
43
|
-
|
|
44
|
-
def _str_escape(text):
|
|
45
|
-
result = ''
|
|
46
|
-
for value in text:
|
|
47
|
-
if value == '\n':
|
|
48
|
-
result += '\\n'
|
|
49
|
-
elif value == '\r':
|
|
50
|
-
result += "\\r"
|
|
51
|
-
elif value == '\t':
|
|
52
|
-
result += "\\t"
|
|
53
|
-
elif value == '\"':
|
|
54
|
-
result += "\\\""
|
|
55
|
-
elif value == '\'':
|
|
56
|
-
result += "\\'"
|
|
57
|
-
elif value == '\\':
|
|
58
|
-
result += "\\\\"
|
|
59
|
-
else:
|
|
60
|
-
result += value
|
|
61
|
-
return result
|
|
62
|
-
|
|
63
|
-
def _pbtxt_from_multiline(multiline_pbtxt):
|
|
64
|
-
pbtxt = ''
|
|
65
|
-
while len(multiline_pbtxt) > 0:
|
|
66
|
-
index = multiline_pbtxt.find('\n')
|
|
67
|
-
if index == -1:
|
|
68
|
-
pbtxt = pbtxt + multiline_pbtxt
|
|
69
|
-
multiline_pbtxt = ''
|
|
70
|
-
break
|
|
71
|
-
line = multiline_pbtxt[0:index]
|
|
72
|
-
multiline_pbtxt = multiline_pbtxt[index+1:]
|
|
73
|
-
colon = line.find(':')
|
|
74
|
-
end = _find_multiline(line, colon)
|
|
75
|
-
if end is None:
|
|
76
|
-
pbtxt = pbtxt + line + '\n'
|
|
77
|
-
continue
|
|
78
|
-
pbtxt = pbtxt + line[0:colon+1]
|
|
79
|
-
unescaped = ''
|
|
80
|
-
newline = False
|
|
81
|
-
line = ''
|
|
82
|
-
while len(multiline_pbtxt) > 0:
|
|
83
|
-
index = multiline_pbtxt.find('\n')
|
|
84
|
-
line = multiline_pbtxt[0:index]
|
|
85
|
-
multiline_pbtxt = multiline_pbtxt[index+1:]
|
|
86
|
-
if line.startswith(end):
|
|
87
|
-
line = line[len(end):]
|
|
88
|
-
break
|
|
89
|
-
if newline:
|
|
90
|
-
unescaped = unescaped + '\n'
|
|
91
|
-
newline = True
|
|
92
|
-
unescaped = unescaped + line
|
|
93
|
-
line = ''
|
|
94
|
-
pbtxt = pbtxt + '\"' + _str_escape(unescaped) + '\"' + line + '\n'
|
|
95
|
-
return pbtxt
|
|
96
|
-
|
|
97
|
-
def _read_op_list(file):
|
|
98
|
-
op_list = op_def_pb2.OpList() # pylint: disable=no-member
|
|
99
|
-
content = _read(file)
|
|
100
|
-
content = re.sub(r'^go/[a-z]+\s*', '', content)
|
|
101
|
-
google.protobuf.text_format.Merge(content, op_list)
|
|
102
|
-
return op_list
|
|
103
|
-
|
|
104
|
-
def _read_api_def_map(folder):
|
|
105
|
-
api_def_map = {}
|
|
106
|
-
for filename in sorted(os.listdir(folder)):
|
|
107
|
-
if filename.endswith('.pbtxt'):
|
|
108
|
-
api_defs = api_def_pb2.ApiDefs() # pylint: disable=no-member
|
|
109
|
-
filename = folder + '/' + filename
|
|
110
|
-
with open(filename, 'r', encoding='utf-8') as file:
|
|
111
|
-
multiline_pbtxt = file.read()
|
|
112
|
-
pbtxt = _pbtxt_from_multiline(multiline_pbtxt)
|
|
113
|
-
google.protobuf.text_format.Merge(pbtxt, api_defs)
|
|
114
|
-
for api_def in api_defs.op:
|
|
115
|
-
api_def_map[api_def.graph_op_name] = api_def
|
|
116
|
-
return api_def_map
|
|
117
|
-
|
|
118
|
-
def _convert_type(value):
|
|
119
|
-
return { 'type': 'type', 'value': value }
|
|
120
|
-
|
|
121
|
-
def _convert_tensor(tensor): # pylint: disable=unused-argument
|
|
122
|
-
return { 'type': 'tensor', 'value': '?' }
|
|
123
|
-
|
|
124
|
-
def _convert_shape(shape): # pylint: disable=unused-argument
|
|
125
|
-
return { 'type': 'shape', 'value': '?' }
|
|
126
|
-
|
|
127
|
-
def _convert_number(number):
|
|
128
|
-
if number == float('inf'):
|
|
129
|
-
return 'NaN'
|
|
130
|
-
if number == float('-inf'):
|
|
131
|
-
return '-NaN'
|
|
132
|
-
return number
|
|
133
|
-
|
|
134
|
-
attr_type_table = {
|
|
135
|
-
'type': 'type', 'list(type)': 'type[]',
|
|
136
|
-
'bool': 'boolean',
|
|
137
|
-
'int': 'int64', 'list(int)': 'int64[]',
|
|
138
|
-
'float': 'float32', 'list(float)': 'float32[]',
|
|
139
|
-
'string': 'string', 'list(string)': 'string[]',
|
|
140
|
-
'shape': 'shape', 'list(shape)': 'shape[]',
|
|
141
|
-
'tensor': 'tensor',
|
|
142
|
-
'func': 'function', 'list(func)': 'function[]'
|
|
143
|
-
}
|
|
144
|
-
|
|
145
|
-
def _convert_attr_type(attr_type):
|
|
146
|
-
if attr_type in attr_type_table:
|
|
147
|
-
return attr_type_table[attr_type]
|
|
148
|
-
print(attr_type)
|
|
149
|
-
return attr_type
|
|
150
|
-
|
|
151
|
-
def _convert_attr_list(attr_value):
|
|
152
|
-
result = []
|
|
153
|
-
attr_value_list = attr_value.list
|
|
154
|
-
if len(attr_value_list.s) > 0:
|
|
155
|
-
for value in attr_value_list.s:
|
|
156
|
-
result.append(value.decode('utf8'))
|
|
157
|
-
if len(attr_value_list.i) > 0:
|
|
158
|
-
for i in attr_value_list.i:
|
|
159
|
-
result.append(i)
|
|
160
|
-
if len(attr_value_list.f) > 0:
|
|
161
|
-
for value in attr_value_list.f:
|
|
162
|
-
result.append(_convert_number(value))
|
|
163
|
-
if len(attr_value_list.type) > 0:
|
|
164
|
-
for value in attr_value_list.type:
|
|
165
|
-
result.append(_convert_type(value))
|
|
166
|
-
if len(result) == 0:
|
|
167
|
-
for _, value in attr_value_list.ListFields():
|
|
168
|
-
if len(value) > 0:
|
|
169
|
-
raise NotImplementedError()
|
|
170
|
-
return result
|
|
171
|
-
|
|
172
|
-
def _convert_attr_value(attr_value):
|
|
173
|
-
if attr_value.HasField('list'):
|
|
174
|
-
value = _convert_attr_list(attr_value)
|
|
175
|
-
elif attr_value.HasField('s'):
|
|
176
|
-
value = attr_value.s.decode('utf8')
|
|
177
|
-
elif attr_value.HasField('i'):
|
|
178
|
-
value = attr_value.i
|
|
179
|
-
elif attr_value.HasField('f'):
|
|
180
|
-
value = _convert_number(attr_value.f)
|
|
181
|
-
elif attr_value.HasField('b'):
|
|
182
|
-
value = attr_value.b
|
|
183
|
-
elif attr_value.HasField('type'):
|
|
184
|
-
value = _convert_type(attr_value.type)
|
|
185
|
-
elif attr_value.HasField('tensor'):
|
|
186
|
-
value = _convert_tensor(attr_value.tensor)
|
|
187
|
-
elif attr_value.HasField('shape'):
|
|
188
|
-
value = _convert_shape(attr_value.shape)
|
|
189
|
-
else:
|
|
190
|
-
raise NotImplementedError()
|
|
191
|
-
return value
|
|
192
|
-
|
|
193
|
-
DataType = types_pb2.DataType # pylint: disable=no-member
|
|
194
|
-
|
|
195
|
-
type_to_string_map = {
|
|
196
|
-
DataType.DT_HALF: "float16",
|
|
197
|
-
DataType.DT_FLOAT: "float32",
|
|
198
|
-
DataType.DT_DOUBLE: "float64",
|
|
199
|
-
DataType.DT_INT32: "int32",
|
|
200
|
-
DataType.DT_UINT8: "uint8",
|
|
201
|
-
DataType.DT_UINT16: "uint16",
|
|
202
|
-
DataType.DT_UINT32: "uint32",
|
|
203
|
-
DataType.DT_UINT64: "uint64",
|
|
204
|
-
DataType.DT_INT16: "int16",
|
|
205
|
-
DataType.DT_INT8: "int8",
|
|
206
|
-
DataType.DT_STRING: "string",
|
|
207
|
-
DataType.DT_COMPLEX64: "complex64",
|
|
208
|
-
DataType.DT_COMPLEX128: "complex128",
|
|
209
|
-
DataType.DT_INT64: "int64",
|
|
210
|
-
DataType.DT_BOOL: "bool",
|
|
211
|
-
DataType.DT_QINT8: "qint8",
|
|
212
|
-
DataType.DT_QUINT8: "quint8",
|
|
213
|
-
DataType.DT_QINT16: "qint16",
|
|
214
|
-
DataType.DT_QUINT16: "quint16",
|
|
215
|
-
DataType.DT_QINT32: "qint32",
|
|
216
|
-
DataType.DT_BFLOAT16: "bfloat16",
|
|
217
|
-
DataType.DT_RESOURCE: "resource",
|
|
218
|
-
DataType.DT_VARIANT: "variant",
|
|
219
|
-
DataType.DT_HALF_REF: "float16_ref",
|
|
220
|
-
DataType.DT_FLOAT_REF: "float32_ref",
|
|
221
|
-
DataType.DT_DOUBLE_REF: "float64_ref",
|
|
222
|
-
DataType.DT_INT32_REF: "int32_ref",
|
|
223
|
-
DataType.DT_UINT32_REF: "uint32_ref",
|
|
224
|
-
DataType.DT_UINT8_REF: "uint8_ref",
|
|
225
|
-
DataType.DT_UINT16_REF: "uint16_ref",
|
|
226
|
-
DataType.DT_INT16_REF: "int16_ref",
|
|
227
|
-
DataType.DT_INT8_REF: "int8_ref",
|
|
228
|
-
DataType.DT_STRING_REF: "string_ref",
|
|
229
|
-
DataType.DT_COMPLEX64_REF: "complex64_ref",
|
|
230
|
-
DataType.DT_COMPLEX128_REF: "complex128_ref",
|
|
231
|
-
DataType.DT_INT64_REF: "int64_ref",
|
|
232
|
-
DataType.DT_UINT64_REF: "uint64_ref",
|
|
233
|
-
DataType.DT_BOOL_REF: "bool_ref",
|
|
234
|
-
DataType.DT_QINT8_REF: "qint8_ref",
|
|
235
|
-
DataType.DT_QUINT8_REF: "quint8_ref",
|
|
236
|
-
DataType.DT_QINT16_REF: "qint16_ref",
|
|
237
|
-
DataType.DT_QUINT16_REF: "quint16_ref",
|
|
238
|
-
DataType.DT_QINT32_REF: "qint32_ref",
|
|
239
|
-
DataType.DT_BFLOAT16_REF: "bfloat16_ref",
|
|
240
|
-
DataType.DT_RESOURCE_REF: "resource_ref",
|
|
241
|
-
DataType.DT_VARIANT_REF: "variant_ref",
|
|
242
|
-
}
|
|
243
|
-
|
|
244
|
-
def _format_data_type(data_type):
|
|
245
|
-
if data_type in type_to_string_map:
|
|
246
|
-
return type_to_string_map[data_type]
|
|
247
|
-
raise KeyError()
|
|
248
|
-
|
|
249
|
-
def _format_attribute_value(value):
|
|
250
|
-
if isinstance(value, dict) and \
|
|
251
|
-
'type' in value and 'value' in value and value['type'] == 'type':
|
|
252
|
-
return _format_data_type(value['value'])
|
|
253
|
-
if isinstance(value, str):
|
|
254
|
-
return value
|
|
255
|
-
if value is True:
|
|
256
|
-
return 'true'
|
|
257
|
-
if value is False:
|
|
258
|
-
return 'false'
|
|
259
|
-
raise NotImplementedError()
|
|
260
|
-
|
|
261
|
-
def _update_attributes(json_schema, operator, api_def):
|
|
262
|
-
api_def_attr_map = {}
|
|
263
|
-
for attr in api_def.attr:
|
|
264
|
-
api_def_attr_map[attr.name] = attr
|
|
265
|
-
for attr in operator.attr:
|
|
266
|
-
if 'attributes' not in json_schema:
|
|
267
|
-
json_schema['attributes'] = []
|
|
268
|
-
json_attribute = {}
|
|
269
|
-
json_attribute['name'] = attr.name
|
|
270
|
-
attr_type = _convert_attr_type(attr.type)
|
|
271
|
-
if attr_type:
|
|
272
|
-
json_attribute['type'] = attr_type
|
|
273
|
-
else:
|
|
274
|
-
del json_attribute['type']
|
|
275
|
-
if attr.name in api_def_attr_map:
|
|
276
|
-
api_def_attr = api_def_attr_map[attr.name]
|
|
277
|
-
if api_def_attr.description:
|
|
278
|
-
json_attribute['description'] = api_def_attr.description
|
|
279
|
-
if attr.has_minimum:
|
|
280
|
-
json_attribute['minimum'] = attr.minimum
|
|
281
|
-
if attr.HasField('allowed_values'):
|
|
282
|
-
allowed_values = _convert_attr_value(attr.allowed_values)
|
|
283
|
-
description = json_attribute['description'] + \
|
|
284
|
-
' ' if 'description' in json_attribute else ''
|
|
285
|
-
allowed_values = list( \
|
|
286
|
-
map(lambda x: "`" + _format_attribute_value(x) + "`", \
|
|
287
|
-
allowed_values))
|
|
288
|
-
description = description + \
|
|
289
|
-
'Must be one of the following: ' + ', '.join(allowed_values) + '.'
|
|
290
|
-
json_attribute['description'] = description
|
|
291
|
-
if attr.HasField('default_value'):
|
|
292
|
-
default_value = _convert_attr_value(attr.default_value)
|
|
293
|
-
json_attribute['default'] = default_value
|
|
294
|
-
json_schema['attributes'].append(json_attribute)
|
|
295
|
-
|
|
296
|
-
def _update_inputs(json_schema, operator, api_def):
|
|
297
|
-
api_def_in_arg_map = {}
|
|
298
|
-
for in_arg in api_def.in_arg:
|
|
299
|
-
api_def_in_arg_map[in_arg.name] = in_arg
|
|
300
|
-
for input_arg in operator.input_arg:
|
|
301
|
-
if 'inputs' not in json_schema:
|
|
302
|
-
json_schema['inputs'] = []
|
|
303
|
-
json_input = {}
|
|
304
|
-
json_input['name'] = input_arg.name
|
|
305
|
-
if input_arg.name in api_def_in_arg_map:
|
|
306
|
-
api_def_in_arg = api_def_in_arg_map[input_arg.name]
|
|
307
|
-
if api_def_in_arg.description:
|
|
308
|
-
json_input['description'] = api_def_in_arg.description
|
|
309
|
-
if input_arg.number_attr:
|
|
310
|
-
json_input['numberAttr'] = input_arg.number_attr
|
|
311
|
-
if input_arg.type:
|
|
312
|
-
json_input['type'] = input_arg.type
|
|
313
|
-
if input_arg.type_attr:
|
|
314
|
-
json_input['typeAttr'] = input_arg.type_attr
|
|
315
|
-
if input_arg.type_list_attr:
|
|
316
|
-
json_input['typeListAttr'] = input_arg.type_list_attr
|
|
317
|
-
if input_arg.is_ref:
|
|
318
|
-
json_input['isRef'] = True
|
|
319
|
-
json_schema['inputs'].append(json_input)
|
|
320
|
-
|
|
321
|
-
def _update_outputs(json_schema, operator, api_def):
|
|
322
|
-
api_def_out_arg_map = {}
|
|
323
|
-
for out_arg in api_def.out_arg:
|
|
324
|
-
api_def_out_arg_map[out_arg.name] = out_arg
|
|
325
|
-
for output_arg in operator.output_arg:
|
|
326
|
-
if 'outputs' not in json_schema:
|
|
327
|
-
json_schema['outputs'] = []
|
|
328
|
-
json_output = {}
|
|
329
|
-
json_output['name'] = output_arg.name
|
|
330
|
-
if output_arg.name in api_def_out_arg_map:
|
|
331
|
-
api_def_out_arg = api_def_out_arg_map[output_arg.name]
|
|
332
|
-
if api_def_out_arg.description:
|
|
333
|
-
json_output['description'] = api_def_out_arg.description
|
|
334
|
-
if output_arg.number_attr:
|
|
335
|
-
json_output['numberAttr'] = output_arg.number_attr
|
|
336
|
-
if output_arg.type:
|
|
337
|
-
json_output['type'] = output_arg.type
|
|
338
|
-
elif output_arg.type_attr:
|
|
339
|
-
json_output['typeAttr'] = output_arg.type_attr
|
|
340
|
-
elif output_arg.type_list_attr:
|
|
341
|
-
json_output['typeListAttr'] = output_arg.type_list_attr
|
|
342
|
-
if output_arg.is_ref:
|
|
343
|
-
json_output['isRef'] = True
|
|
344
|
-
json_schema['outputs'].append(json_output)
|
|
345
|
-
|
|
346
|
-
categories = {
|
|
347
|
-
'Assign': 'Control',
|
|
348
|
-
'AvgPool': 'Pool',
|
|
349
|
-
'BatchNormWithGlobalNormalization': 'Normalization',
|
|
350
|
-
'BiasAdd': 'Layer',
|
|
351
|
-
'Concat': 'Tensor',
|
|
352
|
-
'ConcatV2': 'Tensor',
|
|
353
|
-
'Const': 'Constant',
|
|
354
|
-
'Conv2D': 'Layer',
|
|
355
|
-
'DepthwiseConv2dNative': 'Layer',
|
|
356
|
-
'Dequantize': 'Quantization',
|
|
357
|
-
'Elu': 'Activation',
|
|
358
|
-
'FusedBatchNorm': 'Normalization',
|
|
359
|
-
'FusedBatchNormV2': 'Normalization',
|
|
360
|
-
'FusedBatchNormV3': 'Normalization',
|
|
361
|
-
'Gather': 'Transform',
|
|
362
|
-
'Identity': 'Control',
|
|
363
|
-
'LeakyRelu': 'Activation',
|
|
364
|
-
'LRN': 'Normalization',
|
|
365
|
-
'LSTMBlockCell': 'Layer',
|
|
366
|
-
'MaxPool': 'Pool',
|
|
367
|
-
'MaxPoolV2': 'Pool',
|
|
368
|
-
'MaxPoolWithArgmax': 'Pool',
|
|
369
|
-
'Pad': 'Tensor',
|
|
370
|
-
'QuantizeAndDequantize': 'Quantization',
|
|
371
|
-
'QuantizeAndDequantizeV2': 'Quantization',
|
|
372
|
-
'QuantizeAndDequantizeV3': 'Quantization',
|
|
373
|
-
'QuantizeAndDequantizeV4': 'Quantization',
|
|
374
|
-
'QuantizeAndDequantizeV4Grad': 'Quantization',
|
|
375
|
-
'QuantizeDownAndShrinkRange': 'Quantization',
|
|
376
|
-
'QuantizeV2': 'Quantization',
|
|
377
|
-
'Relu': 'Activation',
|
|
378
|
-
'Relu6': 'Activation',
|
|
379
|
-
'Reshape': 'Shape',
|
|
380
|
-
'Sigmoid': 'Activation',
|
|
381
|
-
'Slice': 'Tensor',
|
|
382
|
-
'Softmax': 'Activation',
|
|
383
|
-
'Split': 'Tensor',
|
|
384
|
-
'Squeeze': 'Transform',
|
|
385
|
-
'StridedSlice': 'Tensor',
|
|
386
|
-
'swish_f32': 'Activation',
|
|
387
|
-
'Transpose': 'Transform',
|
|
388
|
-
'Variable': 'Control',
|
|
389
|
-
'VariableV2': 'Control',
|
|
390
|
-
}
|
|
391
|
-
|
|
392
|
-
def _metadata():
|
|
393
|
-
root_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
|
|
394
|
-
core_dir = os.path.join(root_dir, 'third_party', 'source', 'tensorflow', 'tensorflow', 'core')
|
|
395
|
-
api_def_map = _read_api_def_map(os.path.join(core_dir, 'api_def' , 'base_api'))
|
|
396
|
-
ops_list = _read_op_list(os.path.join(core_dir, 'ops', 'ops.pbtxt'))
|
|
397
|
-
|
|
398
|
-
json_root = []
|
|
399
|
-
for operator in ops_list.op:
|
|
400
|
-
json_schema = {}
|
|
401
|
-
json_schema['name'] = operator.name
|
|
402
|
-
if operator.name in categories:
|
|
403
|
-
json_schema['category'] = categories[operator.name]
|
|
404
|
-
api_def = api_def_pb2.ApiDef() # pylint: disable=no-member
|
|
405
|
-
if operator.name in api_def_map:
|
|
406
|
-
api_def = api_def_map[operator.name]
|
|
407
|
-
if api_def.summary:
|
|
408
|
-
json_schema['summary'] = api_def.summary
|
|
409
|
-
if api_def.description:
|
|
410
|
-
json_schema['description'] = api_def.description
|
|
411
|
-
_update_attributes(json_schema, operator, api_def)
|
|
412
|
-
_update_inputs(json_schema, operator, api_def)
|
|
413
|
-
_update_outputs(json_schema, operator, api_def)
|
|
414
|
-
json_root.append(json_schema)
|
|
415
|
-
json_file = os.path.join(root_dir, 'source', 'tf-metadata.json')
|
|
416
|
-
_write(json_file, json.dumps(json_root, sort_keys=False, indent=2))
|
|
417
|
-
|
|
418
|
-
def main(): # pylint: disable=missing-function-docstring
|
|
419
|
-
_metadata()
|
|
420
|
-
|
|
421
|
-
if __name__ == '__main__':
|
|
422
|
-
main()
|
|
@@ -1,87 +0,0 @@
|
|
|
1
|
-
|
|
2
|
-
import * as flatc from './flatc.js';
|
|
3
|
-
import * as fs from 'fs/promises';
|
|
4
|
-
import * as path from 'path';
|
|
5
|
-
import * as url from 'url';
|
|
6
|
-
|
|
7
|
-
/* eslint-disable no-extend-native */
|
|
8
|
-
|
|
9
|
-
BigInt.prototype.toNumber = function() {
|
|
10
|
-
if (this > Number.MAX_SAFE_INTEGER || this < Number.MIN_SAFE_INTEGER) {
|
|
11
|
-
throw new Error('64-bit value exceeds safe integer.');
|
|
12
|
-
}
|
|
13
|
-
return Number(this);
|
|
14
|
-
};
|
|
15
|
-
|
|
16
|
-
/* eslint-enable no-extend-native */
|
|
17
|
-
|
|
18
|
-
const main = async () => {
|
|
19
|
-
const dirname = path.dirname(url.fileURLToPath(import.meta.url));
|
|
20
|
-
const schema = path.join(dirname, '..', 'third_party', 'source', 'tensorflow', 'tensorflow', 'compiler', 'mlir', 'lite', 'schema', 'schema.fbs');
|
|
21
|
-
const file = path.join(dirname, '..', 'source', 'tflite-metadata.json');
|
|
22
|
-
const input = await fs.readFile(file, 'utf-8');
|
|
23
|
-
const json = JSON.parse(input);
|
|
24
|
-
const operators = new Map();
|
|
25
|
-
const attributes = new Map();
|
|
26
|
-
for (const operator of json) {
|
|
27
|
-
if (operators.has(operator.name)) {
|
|
28
|
-
throw new Error(`Duplicate operator '${operator.name}'.`);
|
|
29
|
-
}
|
|
30
|
-
operators.set(operator.name, operator);
|
|
31
|
-
if (operator && operator.attributes) {
|
|
32
|
-
for (const attribute of operator.attributes) {
|
|
33
|
-
const name = `${operator.name}:${attribute.name}`;
|
|
34
|
-
attributes.set(name, attribute);
|
|
35
|
-
}
|
|
36
|
-
}
|
|
37
|
-
}
|
|
38
|
-
const root = new flatc.Root('tflite');
|
|
39
|
-
await root.load([], [schema]);
|
|
40
|
-
const namespace = root.find('tflite', flatc.Namespace);
|
|
41
|
-
const builtOperator = namespace.find('tflite.BuiltinOperator', flatc.Type);
|
|
42
|
-
const upperCase = new Set(['2D', 'LSH', 'SVDF', 'RNN', 'L2', 'LSTM']);
|
|
43
|
-
for (const op of builtOperator.values.keys()) {
|
|
44
|
-
let op_key = op === 'BATCH_MATMUL' ? 'BATCH_MAT_MUL' : op;
|
|
45
|
-
op_key = op_key.split('_').map((s) => (s.length < 1 || upperCase.has(s)) ? s : s[0] + s.substring(1).toLowerCase()).join('');
|
|
46
|
-
const table = namespace.find(`tflite.${op_key}Options`, flatc.Type);
|
|
47
|
-
if (table && table.fields.size > 0) {
|
|
48
|
-
if (!operators.has(op_key)) {
|
|
49
|
-
const operator = { name: op_key };
|
|
50
|
-
operators.set(op_key, operator);
|
|
51
|
-
json.push(operator);
|
|
52
|
-
}
|
|
53
|
-
const operator = operators.get(op_key);
|
|
54
|
-
operator.attributes = operator.attributes || [];
|
|
55
|
-
for (const field of table.fields.values()) {
|
|
56
|
-
const attr_key = `${op_key}:${field.name}`;
|
|
57
|
-
if (!attributes.has(attr_key)) {
|
|
58
|
-
const attribute = { name: field.name };
|
|
59
|
-
attributes.set(attr_key, attribute);
|
|
60
|
-
operator.attributes.push(attribute);
|
|
61
|
-
}
|
|
62
|
-
const attribute = attributes.get(attr_key);
|
|
63
|
-
const type = field.type;
|
|
64
|
-
let defaultValue = field.defaultValue;
|
|
65
|
-
if (typeof defaultValue === 'bigint') {
|
|
66
|
-
defaultValue = defaultValue.toNumber();
|
|
67
|
-
}
|
|
68
|
-
if (type instanceof flatc.Enum) {
|
|
69
|
-
if (!type.keys.has(defaultValue)) {
|
|
70
|
-
throw new Error(`Invalid '${type.name}' default value '${defaultValue}'.`);
|
|
71
|
-
}
|
|
72
|
-
defaultValue = type.keys.get(defaultValue);
|
|
73
|
-
}
|
|
74
|
-
attribute.type = type.name === 'bool' ? 'boolean' : type.name + (field.repeated ? '[]' : '');
|
|
75
|
-
attribute.default = defaultValue;
|
|
76
|
-
}
|
|
77
|
-
}
|
|
78
|
-
}
|
|
79
|
-
json.sort((a, b) => a.name.localeCompare(b.name));
|
|
80
|
-
let output = JSON.stringify(json, null, 2);
|
|
81
|
-
output = output.replace(/\s {8}/g, ' ');
|
|
82
|
-
output = output.replace(/,\s {8}/g, ', ');
|
|
83
|
-
output = output.replace(/\s {6}}/g, ' }');
|
|
84
|
-
await fs.writeFile(file, output, 'utf-8');
|
|
85
|
-
};
|
|
86
|
-
|
|
87
|
-
await main();
|
|
@@ -1,73 +0,0 @@
|
|
|
1
|
-
|
|
2
|
-
import * as path from 'path';
|
|
3
|
-
import * as fs from 'fs/promises';
|
|
4
|
-
import * as url from 'url';
|
|
5
|
-
import * as flatc from './flatc.js';
|
|
6
|
-
|
|
7
|
-
const main = async () => {
|
|
8
|
-
const dirname = path.dirname(url.fileURLToPath(import.meta.url));
|
|
9
|
-
const schema = path.join(dirname, '..', 'third_party', 'source', 'tensorflow', 'tensorflow', 'lite', 'schema', 'schema.fbs');
|
|
10
|
-
const file = path.join(dirname, '..', 'source', 'tflite-metadata.json');
|
|
11
|
-
const input = await fs.readFile(file, 'utf-8');
|
|
12
|
-
const json = JSON.parse(input);
|
|
13
|
-
const operators = new Map();
|
|
14
|
-
const attributes = new Map();
|
|
15
|
-
for (const operator of json) {
|
|
16
|
-
if (operators.has(operator.name)) {
|
|
17
|
-
throw new Error(`Duplicate operator '${operator.name}'.`);
|
|
18
|
-
}
|
|
19
|
-
operators.set(operator.name, operator);
|
|
20
|
-
if (operator && operator.attributes) {
|
|
21
|
-
for (const attribute of operator.attributes) {
|
|
22
|
-
const name = `${operator.name}:${attribute.name}`;
|
|
23
|
-
attributes.set(name, attribute);
|
|
24
|
-
}
|
|
25
|
-
}
|
|
26
|
-
}
|
|
27
|
-
const root = new flatc.Root('tflite');
|
|
28
|
-
await root.load([], [ schema ]);
|
|
29
|
-
const namespace = root.find('tflite', flatc.Namespace);
|
|
30
|
-
const builtOperator = namespace.find('tflite.BuiltinOperator', flatc.Type);
|
|
31
|
-
const upperCase = new Set([ '2D', 'LSH', 'SVDF', 'RNN', 'L2', 'LSTM' ]);
|
|
32
|
-
for (const op of builtOperator.values.keys()) {
|
|
33
|
-
let op_key = op === 'BATCH_MATMUL' ? 'BATCH_MAT_MUL' : op;
|
|
34
|
-
op_key = op_key.split('_').map((s) => (s.length < 1 || upperCase.has(s)) ? s : s[0] + s.substring(1).toLowerCase()).join('');
|
|
35
|
-
const table = namespace.find(`tflite.${op_key}Options`, flatc.Type);
|
|
36
|
-
if (table && table.fields.size > 0) {
|
|
37
|
-
if (!operators.has(op_key)) {
|
|
38
|
-
const operator = { name: op_key };
|
|
39
|
-
operators.set(op_key, operator);
|
|
40
|
-
json.push(operator);
|
|
41
|
-
}
|
|
42
|
-
const operator = operators.get(op_key);
|
|
43
|
-
operator.attributes = operator.attributes || [];
|
|
44
|
-
for (const field of table.fields.values()) {
|
|
45
|
-
const attr_key = `${op_key}:${field.name}`;
|
|
46
|
-
if (!attributes.has(attr_key)) {
|
|
47
|
-
const attribute = { name: field.name };
|
|
48
|
-
attributes.set(attr_key, attribute);
|
|
49
|
-
operator.attributes.push(attribute);
|
|
50
|
-
}
|
|
51
|
-
const attribute = attributes.get(attr_key);
|
|
52
|
-
const type = field.type;
|
|
53
|
-
let defaultValue = field.defaultValue;
|
|
54
|
-
if (type instanceof flatc.Enum) {
|
|
55
|
-
if (!type.keys.has(defaultValue)) {
|
|
56
|
-
throw new Error(`Invalid '${type.name}' default value '${defaultValue}'.`);
|
|
57
|
-
}
|
|
58
|
-
defaultValue = type.keys.get(defaultValue);
|
|
59
|
-
}
|
|
60
|
-
attribute.type = type.name === 'bool' ? 'boolean' : type.name + (field.repeated ? '[]' : '');
|
|
61
|
-
attribute.default = defaultValue;
|
|
62
|
-
}
|
|
63
|
-
}
|
|
64
|
-
}
|
|
65
|
-
json.sort((a, b) => a.name.localeCompare(b.name));
|
|
66
|
-
let output = JSON.stringify(json, null, 2);
|
|
67
|
-
output = output.replace(/\s {8}/g, ' ');
|
|
68
|
-
output = output.replace(/,\s {8}/g, ', ');
|
|
69
|
-
output = output.replace(/\s {6}}/g, ' }');
|
|
70
|
-
await fs.writeFile(file, output, 'utf-8');
|
|
71
|
-
};
|
|
72
|
-
|
|
73
|
-
main();
|
|
@@ -1,20 +0,0 @@
|
|
|
1
|
-
#!/bin/bash
|
|
2
|
-
|
|
3
|
-
set -e
|
|
4
|
-
pushd $(cd $(dirname ${0})/..; pwd) > /dev/null
|
|
5
|
-
|
|
6
|
-
schema() {
|
|
7
|
-
echo "uff schema"
|
|
8
|
-
[[ $(grep -U $'\x0D' ./source/uff-proto.js) ]] && crlf=1
|
|
9
|
-
node ./tools/protoc.js --binary --text --root uff --out ./source/uff-proto.js ./tools/uff.proto
|
|
10
|
-
if [[ -n ${crlf} ]]; then
|
|
11
|
-
unix2dos --quiet --newfile ./source/uff-proto.js ./source/uff-proto.js
|
|
12
|
-
fi
|
|
13
|
-
}
|
|
14
|
-
|
|
15
|
-
while [ "$#" != 0 ]; do
|
|
16
|
-
command="$1" && shift
|
|
17
|
-
case "${command}" in
|
|
18
|
-
"schema") schema;;
|
|
19
|
-
esac
|
|
20
|
-
done
|