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,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