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