onnx 1.15.0__cp311-cp311-win_amd64.whl → 1.16.1__cp311-cp311-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.

Potentially problematic release.


This version of onnx might be problematic. Click here for more details.

Files changed (584) hide show
  1. onnx/__init__.py +10 -10
  2. onnx/backend/base.py +13 -14
  3. onnx/backend/sample/ops/abs.py +1 -1
  4. onnx/backend/test/case/model/__init__.py +0 -1
  5. onnx/backend/test/case/node/ai_onnx_ml/tree_ensemble.py +122 -0
  6. onnx/backend/test/case/node/averagepool.py +15 -30
  7. onnx/backend/test/case/node/cast.py +88 -11
  8. onnx/backend/test/case/node/dequantizelinear.py +155 -0
  9. onnx/backend/test/case/node/groupnormalization.py +13 -9
  10. onnx/backend/test/case/node/gru.py +2 -2
  11. onnx/backend/test/case/node/isinf.py +4 -4
  12. onnx/backend/test/case/node/isnan.py +2 -2
  13. onnx/backend/test/case/node/lppool.py +8 -16
  14. onnx/backend/test/case/node/lstm.py +1 -1
  15. onnx/backend/test/case/node/maxpool.py +40 -34
  16. onnx/backend/test/case/node/pow.py +1 -1
  17. onnx/backend/test/case/node/qlinearmatmul.py +143 -109
  18. onnx/backend/test/case/node/quantizelinear.py +298 -7
  19. onnx/backend/test/case/node/reducemax.py +26 -0
  20. onnx/backend/test/case/node/rnn.py +1 -1
  21. onnx/backend/test/case/node/scan.py +6 -2
  22. onnx/backend/test/case/node/scatterelements.py +1 -1
  23. onnx/backend/test/case/node/topk.py +1 -1
  24. onnx/backend/test/case/utils.py +1 -3
  25. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/model.onnx +0 -0
  26. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/test_data_set_0/input_0.pb +0 -0
  27. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/test_data_set_0/output_0.pb +0 -0
  28. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/model.onnx +0 -0
  29. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/test_data_set_0/input_0.pb +1 -0
  30. onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/test_data_set_0/output_0.pb +0 -0
  31. onnx/backend/test/data/node/test_cast_BFLOAT16_to_FLOAT/model.onnx +0 -0
  32. onnx/backend/test/data/node/test_cast_DOUBLE_to_FLOAT/model.onnx +0 -0
  33. onnx/backend/test/data/node/test_cast_DOUBLE_to_FLOAT16/model.onnx +0 -0
  34. onnx/backend/test/data/node/test_cast_FLOAT16_to_DOUBLE/model.onnx +0 -0
  35. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT/model.onnx +0 -0
  36. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/model.onnx +0 -0
  37. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +2 -2
  38. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
  39. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
  40. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +2 -2
  41. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
  42. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/model.onnx +0 -0
  43. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/test_data_set_0/input_0.pb +2 -2
  44. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
  45. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
  46. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +2 -2
  47. onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
  48. onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/model.onnx +0 -0
  49. onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/test_data_set_0/input_0.pb +0 -0
  50. onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/test_data_set_0/output_0.pb +1 -0
  51. onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/model.onnx +0 -0
  52. onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/test_data_set_0/input_0.pb +0 -0
  53. onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/test_data_set_0/output_0.pb +0 -0
  54. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/model.onnx +0 -0
  55. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/test_data_set_0/input_0.pb +0 -0
  56. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  57. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/model.onnx +0 -0
  58. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
  59. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  60. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/model.onnx +0 -0
  61. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/test_data_set_0/input_0.pb +0 -0
  62. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  63. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/model.onnx +0 -0
  64. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
  65. onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  66. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/model.onnx +0 -0
  67. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/test_data_set_0/input_0.pb +0 -0
  68. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  69. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/model.onnx +0 -0
  70. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
  71. onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  72. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/model.onnx +0 -0
  73. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/test_data_set_0/input_0.pb +0 -0
  74. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  75. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/model.onnx +0 -0
  76. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
  77. onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  78. onnx/backend/test/data/node/test_cast_FLOAT_to_BFLOAT16/model.onnx +0 -0
  79. onnx/backend/test/data/node/test_cast_FLOAT_to_DOUBLE/model.onnx +0 -0
  80. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT16/model.onnx +0 -0
  81. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
  82. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +0 -0
  83. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
  84. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
  85. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +0 -0
  86. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
  87. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
  88. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/test_data_set_0/input_0.pb +0 -0
  89. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
  90. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
  91. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +0 -0
  92. onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
  93. onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/model.onnx +0 -0
  94. onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/test_data_set_0/input_0.pb +0 -0
  95. onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/test_data_set_0/output_0.pb +1 -0
  96. onnx/backend/test/data/node/test_cast_FLOAT_to_STRING/model.onnx +0 -0
  97. onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/model.onnx +0 -0
  98. onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/test_data_set_0/input_0.pb +0 -0
  99. onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/test_data_set_0/output_0.pb +0 -0
  100. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/model.onnx +0 -0
  101. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/test_data_set_0/input_0.pb +1 -0
  102. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  103. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/model.onnx +0 -0
  104. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/test_data_set_0/input_0.pb +1 -0
  105. onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  106. onnx/backend/test/data/node/test_cast_INT4_to_INT8/model.onnx +0 -0
  107. onnx/backend/test/data/node/test_cast_INT4_to_INT8/test_data_set_0/input_0.pb +1 -0
  108. onnx/backend/test/data/node/test_cast_INT4_to_INT8/test_data_set_0/output_0.pb +0 -0
  109. onnx/backend/test/data/node/test_cast_STRING_to_FLOAT/model.onnx +0 -0
  110. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/model.onnx +0 -0
  111. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/test_data_set_0/input_0.pb +0 -0
  112. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/test_data_set_0/output_0.pb +0 -0
  113. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/model.onnx +0 -0
  114. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
  115. onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
  116. onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/model.onnx +0 -0
  117. onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/test_data_set_0/input_0.pb +0 -0
  118. onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/test_data_set_0/output_0.pb +0 -0
  119. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/model.onnx +0 -0
  120. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +2 -2
  121. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
  122. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
  123. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +2 -2
  124. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
  125. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/model.onnx +0 -0
  126. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/test_data_set_0/input_0.pb +2 -2
  127. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
  128. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
  129. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +2 -2
  130. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
  131. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
  132. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +0 -0
  133. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
  134. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
  135. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +0 -0
  136. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
  137. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
  138. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/test_data_set_0/input_0.pb +0 -0
  139. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
  140. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
  141. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +0 -0
  142. onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
  143. onnx/backend/test/data/node/test_castlike_BFLOAT16_to_FLOAT/model.onnx +0 -0
  144. onnx/backend/test/data/node/test_castlike_BFLOAT16_to_FLOAT_expanded/model.onnx +0 -0
  145. onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT/model.onnx +0 -0
  146. onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT16/model.onnx +0 -0
  147. onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT16_expanded/model.onnx +0 -0
  148. onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT_expanded/model.onnx +0 -0
  149. onnx/backend/test/data/node/test_castlike_FLOAT16_to_DOUBLE/model.onnx +0 -0
  150. onnx/backend/test/data/node/test_castlike_FLOAT16_to_DOUBLE_expanded/model.onnx +0 -0
  151. onnx/backend/test/data/node/test_castlike_FLOAT16_to_FLOAT/model.onnx +0 -0
  152. onnx/backend/test/data/node/test_castlike_FLOAT16_to_FLOAT_expanded/model.onnx +0 -0
  153. onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT/model.onnx +0 -0
  154. onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT_expanded/model.onnx +0 -0
  155. onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FN_to_FLOAT/model.onnx +0 -0
  156. onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FN_to_FLOAT_expanded/model.onnx +0 -0
  157. onnx/backend/test/data/node/test_castlike_FLOAT8E5M2FNUZ_to_FLOAT/model.onnx +0 -0
  158. onnx/backend/test/data/node/test_castlike_FLOAT8E5M2FNUZ_to_FLOAT_expanded/model.onnx +0 -0
  159. onnx/backend/test/data/node/test_castlike_FLOAT8E5M2_to_FLOAT/model.onnx +0 -0
  160. onnx/backend/test/data/node/test_castlike_FLOAT8E5M2_to_FLOAT_expanded/model.onnx +0 -0
  161. onnx/backend/test/data/node/test_castlike_FLOAT_to_BFLOAT16/model.onnx +0 -0
  162. onnx/backend/test/data/node/test_castlike_FLOAT_to_BFLOAT16_expanded/model.onnx +0 -0
  163. onnx/backend/test/data/node/test_castlike_FLOAT_to_DOUBLE/model.onnx +0 -0
  164. onnx/backend/test/data/node/test_castlike_FLOAT_to_DOUBLE_expanded/model.onnx +0 -0
  165. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT16/model.onnx +0 -0
  166. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT16_expanded/model.onnx +0 -0
  167. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
  168. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
  169. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FNUZ_expanded/model.onnx +0 -0
  170. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FN_expanded/model.onnx +0 -0
  171. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
  172. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
  173. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ_expanded/model.onnx +0 -0
  174. onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2_expanded/model.onnx +0 -0
  175. onnx/backend/test/data/node/test_castlike_FLOAT_to_STRING/model.onnx +0 -0
  176. onnx/backend/test/data/node/test_castlike_FLOAT_to_STRING_expanded/model.onnx +0 -0
  177. onnx/backend/test/data/node/test_castlike_STRING_to_FLOAT/model.onnx +0 -0
  178. onnx/backend/test/data/node/test_castlike_STRING_to_FLOAT_expanded/model.onnx +0 -0
  179. onnx/backend/test/data/node/test_constant/model.onnx +0 -0
  180. onnx/backend/test/data/node/test_constant_pad/model.onnx +0 -0
  181. onnx/backend/test/data/node/test_constant_pad_axes/model.onnx +0 -0
  182. onnx/backend/test/data/node/test_constant_pad_negative_axes/model.onnx +0 -0
  183. onnx/backend/test/data/node/test_constantofshape_float_ones/model.onnx +0 -0
  184. onnx/backend/test/data/node/test_constantofshape_int_shape_zero/model.onnx +0 -0
  185. onnx/backend/test/data/node/test_constantofshape_int_zeros/model.onnx +0 -0
  186. onnx/backend/test/data/node/test_dequantizelinear/model.onnx +0 -0
  187. onnx/backend/test/data/node/test_dequantizelinear_axis/model.onnx +0 -0
  188. onnx/backend/test/data/node/test_dequantizelinear_blocked/model.onnx +0 -0
  189. onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_0.pb +1 -0
  190. onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_1.pb +0 -0
  191. onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_2.pb +0 -0
  192. onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/output_0.pb +0 -0
  193. onnx/backend/test/data/node/test_dequantizelinear_e4m3fn/model.onnx +0 -0
  194. onnx/backend/test/data/node/test_dequantizelinear_e4m3fn_float16/model.onnx +0 -0
  195. onnx/backend/test/data/node/test_dequantizelinear_e4m3fn_zero_point/model.onnx +0 -0
  196. onnx/backend/test/data/node/test_dequantizelinear_e5m2/model.onnx +0 -0
  197. onnx/backend/test/data/node/test_dequantizelinear_int16/model.onnx +0 -0
  198. onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_0.pb +1 -0
  199. onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_1.pb +0 -0
  200. onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_2.pb +0 -0
  201. onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/output_0.pb +0 -0
  202. onnx/backend/test/data/node/test_dequantizelinear_int4/model.onnx +0 -0
  203. onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_0.pb +1 -0
  204. onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_1.pb +0 -0
  205. onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_2.pb +1 -0
  206. onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/output_0.pb +0 -0
  207. onnx/backend/test/data/node/test_dequantizelinear_uint16/model.onnx +0 -0
  208. onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_0.pb +0 -0
  209. onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_1.pb +0 -0
  210. onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_2.pb +1 -0
  211. onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/output_0.pb +0 -0
  212. onnx/backend/test/data/node/test_dequantizelinear_uint4/model.onnx +0 -0
  213. onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_0.pb +1 -0
  214. onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_1.pb +0 -0
  215. onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_2.pb +1 -0
  216. onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/output_0.pb +0 -0
  217. onnx/backend/test/data/node/test_edge_pad/model.onnx +0 -0
  218. onnx/backend/test/data/node/test_flatten_axis0/model.onnx +0 -0
  219. onnx/backend/test/data/node/test_flatten_axis1/model.onnx +0 -0
  220. onnx/backend/test/data/node/test_flatten_axis2/model.onnx +0 -0
  221. onnx/backend/test/data/node/test_flatten_axis3/model.onnx +0 -0
  222. onnx/backend/test/data/node/test_flatten_default_axis/model.onnx +0 -0
  223. onnx/backend/test/data/node/test_flatten_negative_axis1/model.onnx +0 -0
  224. onnx/backend/test/data/node/test_flatten_negative_axis2/model.onnx +0 -0
  225. onnx/backend/test/data/node/test_flatten_negative_axis3/model.onnx +0 -0
  226. onnx/backend/test/data/node/test_flatten_negative_axis4/model.onnx +0 -0
  227. onnx/backend/test/data/node/test_group_normalization_epsilon/model.onnx +0 -0
  228. onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_0.pb +1 -1
  229. onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_1.pb +1 -1
  230. onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_2.pb +1 -1
  231. onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/output_0.pb +0 -0
  232. onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/model.onnx +0 -0
  233. onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_0.pb +1 -1
  234. onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_1.pb +1 -1
  235. onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_2.pb +1 -1
  236. onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/output_0.pb +0 -0
  237. onnx/backend/test/data/node/test_group_normalization_example/model.onnx +0 -0
  238. onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/input_1.pb +1 -1
  239. onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/input_2.pb +1 -1
  240. onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/output_0.pb +0 -0
  241. onnx/backend/test/data/node/test_group_normalization_example_expanded/model.onnx +0 -0
  242. onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/input_1.pb +1 -1
  243. onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/input_2.pb +1 -1
  244. onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/output_0.pb +0 -0
  245. onnx/backend/test/data/node/test_identity/model.onnx +0 -0
  246. onnx/backend/test/data/node/test_identity_sequence/model.onnx +0 -0
  247. onnx/backend/test/data/node/test_lrn_default/test_data_set_0/output_0.pb +0 -0
  248. onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/model.onnx +0 -0
  249. onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/test_data_set_0/input_0.pb +0 -0
  250. onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/test_data_set_0/output_0.pb +0 -0
  251. onnx/backend/test/data/node/test_mvn/test_data_set_0/output_0.pb +1 -1
  252. onnx/backend/test/data/node/test_mvn_expanded/test_data_set_0/output_0.pb +1 -1
  253. onnx/backend/test/data/node/test_mvn_expanded_ver18/test_data_set_0/output_0.pb +1 -1
  254. onnx/backend/test/data/node/test_pow/test_data_set_0/output_0.pb +0 -0
  255. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/model.onnx +0 -0
  256. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_0.pb +1 -0
  257. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_1.pb +2 -0
  258. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_2.pb +1 -0
  259. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_3.pb +0 -0
  260. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_4.pb +2 -0
  261. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_5.pb +1 -0
  262. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_6.pb +2 -0
  263. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_7.pb +1 -0
  264. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/output_0.pb +1 -0
  265. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/model.onnx +0 -0
  266. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_0.pb +1 -0
  267. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_2.pb +1 -0
  268. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_3.pb +0 -0
  269. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_5.pb +1 -0
  270. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_7.pb +1 -0
  271. onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/output_0.pb +1 -0
  272. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/model.onnx +0 -0
  273. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_1.pb +2 -0
  274. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_4.pb +2 -0
  275. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_6.pb +2 -0
  276. onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float32}/model.onnx +0 -0
  277. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/input_0.pb +0 -0
  278. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/input_3.pb +0 -0
  279. onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/output_0.pb +1 -0
  280. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/model.onnx +0 -0
  281. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_0.pb +1 -0
  282. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_1.pb +2 -0
  283. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_2.pb +1 -0
  284. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_3.pb +0 -0
  285. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_4.pb +2 -0
  286. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_5.pb +1 -0
  287. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_6.pb +2 -0
  288. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_7.pb +1 -0
  289. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/output_0.pb +1 -0
  290. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/model.onnx +0 -0
  291. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_0.pb +1 -0
  292. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_1.pb +1 -0
  293. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_2.pb +1 -0
  294. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_3.pb +0 -0
  295. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_4.pb +1 -0
  296. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_5.pb +1 -0
  297. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_6.pb +1 -0
  298. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_7.pb +1 -0
  299. onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/output_0.pb +1 -0
  300. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/model.onnx +0 -0
  301. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_1.pb +2 -0
  302. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_2.pb +1 -0
  303. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_4.pb +2 -0
  304. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_5.pb +1 -0
  305. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_6.pb +2 -0
  306. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_7.pb +1 -0
  307. onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float32}/model.onnx +0 -0
  308. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_0.pb +0 -0
  309. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_1.pb +1 -0
  310. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_2.pb +1 -0
  311. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_3.pb +0 -0
  312. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_4.pb +1 -0
  313. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_5.pb +1 -0
  314. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_6.pb +1 -0
  315. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_7.pb +1 -0
  316. onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/output_0.pb +1 -0
  317. onnx/backend/test/data/node/test_quantizelinear/model.onnx +0 -0
  318. onnx/backend/test/data/node/test_quantizelinear_axis/model.onnx +0 -0
  319. onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/model.onnx +0 -0
  320. onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_0.pb +0 -0
  321. onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_1.pb +0 -0
  322. onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_2.pb +0 -0
  323. onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/output_0.pb +1 -0
  324. onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/model.onnx +0 -0
  325. onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/input_0.pb +0 -0
  326. onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/input_1.pb +0 -0
  327. onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/output_0.pb +0 -0
  328. onnx/backend/test/data/node/test_quantizelinear_e4m3fn/model.onnx +0 -0
  329. onnx/backend/test/data/node/test_quantizelinear_e4m3fn/test_data_set_0/input_2.pb +0 -0
  330. onnx/backend/test/data/node/test_quantizelinear_e4m3fn/test_data_set_0/output_0.pb +0 -0
  331. onnx/backend/test/data/node/test_quantizelinear_e5m2/model.onnx +0 -0
  332. onnx/backend/test/data/node/test_quantizelinear_e5m2/test_data_set_0/input_2.pb +0 -0
  333. onnx/backend/test/data/node/test_quantizelinear_e5m2/test_data_set_0/output_0.pb +0 -0
  334. onnx/backend/test/data/node/test_quantizelinear_int16/model.onnx +0 -0
  335. onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_0.pb +0 -0
  336. onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_1.pb +0 -0
  337. onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_2.pb +0 -0
  338. onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/output_0.pb +0 -0
  339. onnx/backend/test/data/node/test_quantizelinear_int4/model.onnx +0 -0
  340. onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_0.pb +0 -0
  341. onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_1.pb +0 -0
  342. onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_2.pb +1 -0
  343. onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/output_0.pb +1 -0
  344. onnx/backend/test/data/node/test_quantizelinear_uint16/model.onnx +0 -0
  345. onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_0.pb +0 -0
  346. onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_1.pb +0 -0
  347. onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_2.pb +1 -0
  348. onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/output_0.pb +0 -0
  349. onnx/backend/test/data/node/test_quantizelinear_uint4/model.onnx +0 -0
  350. onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_0.pb +0 -0
  351. onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_1.pb +0 -0
  352. onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_2.pb +1 -0
  353. onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/output_0.pb +0 -0
  354. onnx/backend/test/data/node/test_reflect_pad/model.onnx +0 -0
  355. onnx/backend/test/data/node/test_reshape_allowzero_reordered/model.onnx +0 -0
  356. onnx/backend/test/data/node/test_reshape_extended_dims/model.onnx +0 -0
  357. onnx/backend/test/data/node/test_reshape_negative_dim/model.onnx +0 -0
  358. onnx/backend/test/data/node/test_reshape_negative_extended_dims/model.onnx +0 -0
  359. onnx/backend/test/data/node/test_reshape_one_dim/model.onnx +0 -0
  360. onnx/backend/test/data/node/test_reshape_reduced_dims/model.onnx +0 -0
  361. onnx/backend/test/data/node/test_reshape_reordered_all_dims/model.onnx +0 -0
  362. onnx/backend/test/data/node/test_reshape_reordered_last_dims/model.onnx +0 -0
  363. onnx/backend/test/data/node/test_reshape_zero_and_negative_dim/model.onnx +0 -0
  364. onnx/backend/test/data/node/test_reshape_zero_dim/model.onnx +0 -0
  365. onnx/backend/test/data/node/test_shape/model.onnx +0 -0
  366. onnx/backend/test/data/node/test_shape_clip_end/model.onnx +0 -0
  367. onnx/backend/test/data/node/test_shape_clip_start/model.onnx +0 -0
  368. onnx/backend/test/data/node/test_shape_end_1/model.onnx +0 -0
  369. onnx/backend/test/data/node/test_shape_end_negative_1/model.onnx +0 -0
  370. onnx/backend/test/data/node/test_shape_example/model.onnx +0 -0
  371. onnx/backend/test/data/node/test_shape_start_1/model.onnx +0 -0
  372. onnx/backend/test/data/node/test_shape_start_1_end_2/model.onnx +0 -0
  373. onnx/backend/test/data/node/test_shape_start_1_end_negative_1/model.onnx +0 -0
  374. onnx/backend/test/data/node/test_shape_start_negative_1/model.onnx +0 -0
  375. onnx/backend/test/data/node/test_size/model.onnx +0 -0
  376. onnx/backend/test/data/node/test_size_example/model.onnx +0 -0
  377. onnx/backend/test/data/node/test_squeeze/model.onnx +0 -0
  378. onnx/backend/test/data/node/test_squeeze_negative_axes/model.onnx +0 -0
  379. onnx/backend/test/data/node/test_transpose_all_permutations_0/model.onnx +0 -0
  380. onnx/backend/test/data/node/test_transpose_all_permutations_1/model.onnx +0 -0
  381. onnx/backend/test/data/node/test_transpose_all_permutations_2/model.onnx +0 -0
  382. onnx/backend/test/data/node/test_transpose_all_permutations_3/model.onnx +0 -0
  383. onnx/backend/test/data/node/test_transpose_all_permutations_4/model.onnx +0 -0
  384. onnx/backend/test/data/node/test_transpose_all_permutations_5/model.onnx +0 -0
  385. onnx/backend/test/data/node/test_transpose_default/model.onnx +0 -0
  386. onnx/backend/test/data/node/test_unsqueeze_axis_0/model.onnx +0 -0
  387. onnx/backend/test/data/node/test_unsqueeze_axis_1/model.onnx +0 -0
  388. onnx/backend/test/data/node/test_unsqueeze_axis_2/model.onnx +0 -0
  389. onnx/backend/test/data/node/test_unsqueeze_negative_axes/model.onnx +0 -0
  390. onnx/backend/test/data/node/test_unsqueeze_three_axes/model.onnx +0 -0
  391. onnx/backend/test/data/node/test_unsqueeze_two_axes/model.onnx +0 -0
  392. onnx/backend/test/data/node/test_unsqueeze_unsorted_axes/model.onnx +0 -0
  393. onnx/backend/test/data/node/test_wrap_pad/model.onnx +0 -0
  394. onnx/backend/test/loader/__init__.py +0 -1
  395. onnx/backend/test/runner/__init__.py +43 -15
  396. onnx/checker.cc +104 -99
  397. onnx/checker.h +23 -3
  398. onnx/checker.py +56 -20
  399. onnx/common/assertions.cc +10 -5
  400. onnx/common/common.h +19 -0
  401. onnx/common/file_utils.h +3 -1
  402. onnx/common/interned_strings.h +7 -1
  403. onnx/common/ir.h +30 -7
  404. onnx/common/ir_pb_converter.cc +6 -0
  405. onnx/common/path.h +18 -2
  406. onnx/common/proto_util.h +43 -0
  407. onnx/common/version.h +1 -1
  408. onnx/cpp2py_export.cc +88 -56
  409. onnx/defs/__init__.py +29 -8
  410. onnx/defs/controlflow/defs.cc +16 -16
  411. onnx/defs/controlflow/old.cc +177 -0
  412. onnx/defs/data_propagators.h +2 -0
  413. onnx/defs/data_type_utils.cc +2 -0
  414. onnx/defs/generator/defs.cc +6 -4
  415. onnx/defs/generator/old.cc +115 -0
  416. onnx/defs/math/defs.cc +37 -142
  417. onnx/defs/math/old.cc +96 -12
  418. onnx/defs/math/utils.cc +127 -0
  419. onnx/defs/math/utils.h +8 -0
  420. onnx/defs/nn/defs.cc +72 -59
  421. onnx/defs/nn/old.cc +181 -2
  422. onnx/defs/object_detection/defs.cc +2 -2
  423. onnx/defs/object_detection/old.cc +2 -2
  424. onnx/defs/operator_sets.h +51 -0
  425. onnx/defs/operator_sets_ml.h +14 -0
  426. onnx/defs/parser.cc +112 -54
  427. onnx/defs/parser.h +14 -2
  428. onnx/defs/printer.cc +14 -7
  429. onnx/defs/quantization/defs.cc +111 -44
  430. onnx/defs/quantization/old.cc +130 -1
  431. onnx/defs/schema.cc +62 -18
  432. onnx/defs/schema.h +194 -48
  433. onnx/defs/shape_inference.cc +28 -19
  434. onnx/defs/shape_inference.h +2 -0
  435. onnx/defs/tensor/defs.cc +54 -96
  436. onnx/defs/tensor/old.cc +939 -34
  437. onnx/defs/tensor/utils.cc +6 -3
  438. onnx/defs/tensor/utils.h +5 -1
  439. onnx/defs/tensor_proto_util.cc +2 -0
  440. onnx/defs/tensor_util.cc +2 -0
  441. onnx/defs/traditionalml/defs.cc +273 -117
  442. onnx/defs/traditionalml/old.cc +329 -14
  443. onnx/defs/traditionalml/utils.h +27 -0
  444. onnx/external_data_helper.py +12 -26
  445. onnx/helper.py +242 -169
  446. onnx/hub.py +104 -70
  447. onnx/inliner/inliner.cc +89 -31
  448. onnx/inliner/inliner.h +5 -0
  449. onnx/inliner.py +2 -0
  450. onnx/mapping.py +9 -0
  451. onnx/model_container.py +346 -0
  452. onnx/numpy_helper.py +100 -38
  453. onnx/onnx-ml.proto +50 -13
  454. onnx/onnx.in.proto +50 -13
  455. onnx/onnx.proto +50 -13
  456. onnx/onnx_cpp2py_export/__init__.pyi +5 -0
  457. onnx/onnx_cpp2py_export/checker.pyi +21 -0
  458. onnx/onnx_cpp2py_export/defs.pyi +202 -0
  459. onnx/onnx_cpp2py_export/inliner.pyi +19 -0
  460. onnx/onnx_cpp2py_export/parser.pyi +32 -0
  461. onnx/onnx_cpp2py_export/printer.pyi +3 -0
  462. onnx/onnx_cpp2py_export/shape_inference.pyi +16 -0
  463. onnx/onnx_cpp2py_export/version_converter.pyi +4 -0
  464. onnx/onnx_cpp2py_export.cp311-win_amd64.pyd +0 -0
  465. onnx/onnx_data_pb2.pyi +146 -0
  466. onnx/onnx_ml_pb2.py +52 -52
  467. onnx/onnx_ml_pb2.pyi +663 -0
  468. onnx/onnx_operators_ml_pb2.pyi +67 -0
  469. onnx/reference/__init__.py +2 -0
  470. onnx/reference/custom_element_types.py +2 -0
  471. onnx/reference/op_run.py +166 -121
  472. onnx/reference/ops/_op.py +27 -50
  473. onnx/reference/ops/_op_list.py +36 -24
  474. onnx/reference/ops/aionnx_preview_training/_op_list.py +15 -8
  475. onnx/reference/ops/aionnxml/_common_classifier.py +3 -5
  476. onnx/reference/ops/aionnxml/_op_list.py +16 -8
  477. onnx/reference/ops/aionnxml/op_array_feature_extractor.py +4 -6
  478. onnx/reference/ops/aionnxml/op_linear_classifier.py +1 -2
  479. onnx/reference/ops/aionnxml/op_normalizer.py +3 -3
  480. onnx/reference/ops/aionnxml/op_svm_helper.py +1 -3
  481. onnx/reference/ops/aionnxml/op_svm_regressor.py +1 -3
  482. onnx/reference/ops/aionnxml/op_tree_ensemble.py +257 -0
  483. onnx/reference/ops/aionnxml/op_tree_ensemble_helper.py +2 -6
  484. onnx/reference/ops/aionnxml/op_tree_ensemble_regressor.py +4 -4
  485. onnx/reference/ops/experimental/_op_list.py +15 -8
  486. onnx/reference/ops/op_blackman_window.py +5 -6
  487. onnx/reference/ops/op_cast.py +22 -0
  488. onnx/reference/ops/op_cast_like.py +6 -0
  489. onnx/reference/ops/op_clip.py +5 -8
  490. onnx/reference/ops/op_col2im.py +1 -3
  491. onnx/reference/ops/op_constant.py +7 -1
  492. onnx/reference/ops/op_dequantize_linear.py +43 -40
  493. onnx/reference/ops/op_det.py +1 -1
  494. onnx/reference/ops/op_dynamic_quantize_linear.py +2 -2
  495. onnx/reference/ops/op_grid_sample.py +2 -4
  496. onnx/reference/ops/op_hamming_window.py +3 -6
  497. onnx/reference/ops/op_hann_window.py +3 -6
  498. onnx/reference/ops/op_if.py +4 -3
  499. onnx/reference/ops/op_loop.py +7 -9
  500. onnx/reference/ops/op_matmul.py +1 -2
  501. onnx/reference/ops/op_max_pool.py +5 -0
  502. onnx/reference/ops/op_optional.py +1 -1
  503. onnx/reference/ops/op_pool_common.py +3 -6
  504. onnx/reference/ops/op_qlinear_matmul.py +2 -2
  505. onnx/reference/ops/op_quantize_linear.py +166 -71
  506. onnx/reference/ops/op_resize.py +25 -21
  507. onnx/reference/ops/op_rnn.py +20 -12
  508. onnx/reference/ops/op_scan.py +23 -15
  509. onnx/reference/ops/op_scatter_elements.py +7 -6
  510. onnx/reference/ops/op_stft.py +3 -5
  511. onnx/reference/ops/op_string_normalizer.py +7 -7
  512. onnx/reference/ops/op_tfidf_vectorizer.py +7 -8
  513. onnx/reference/ops/op_topk.py +9 -11
  514. onnx/reference/ops/op_unique.py +1 -1
  515. onnx/reference/reference_evaluator.py +119 -63
  516. onnx/shape_inference/implementation.cc +160 -127
  517. onnx/shape_inference.py +11 -10
  518. onnx/subbyte.py +72 -0
  519. onnx/test/__init__.pyi +6 -0
  520. onnx/test/checker_test.py +21 -1
  521. onnx/test/compose_test.py +26 -74
  522. onnx/test/cpp/inliner_test.cc +76 -1
  523. onnx/test/cpp/ir_test.cc +60 -0
  524. onnx/test/cpp/parser_test.cc +106 -0
  525. onnx/test/function_test.py +1 -3
  526. onnx/test/helper_test.py +64 -4
  527. onnx/test/model_container_refeval_test.py +139 -0
  528. onnx/test/model_container_test.py +136 -0
  529. onnx/test/model_inference_test.py +44 -0
  530. onnx/test/reference_evaluator_ml_test.py +448 -47
  531. onnx/test/reference_evaluator_model_test.py +130 -0
  532. onnx/test/reference_evaluator_test.py +901 -14
  533. onnx/test/schema_test.py +166 -1
  534. onnx/test/shape_inference_test.py +285 -6
  535. onnx/test/symbolic_shape_test.py +3 -8
  536. onnx/test/test_backend_onnxruntime.py +238 -224
  537. onnx/test/test_backend_reference.py +11 -0
  538. onnx/test/test_external_data.py +51 -2
  539. onnx/test/version_converter/automatic_conversion_test_base.py +2 -1
  540. onnx/test/version_converter/automatic_upgrade_test.py +12 -10
  541. onnx/test/version_converter_test.py +166 -0
  542. onnx/tools/replace_constants.py +23 -26
  543. onnx/tools/update_model_dims.py +1 -2
  544. onnx/version.py +2 -2
  545. onnx/version_converter/adapters/group_normalization_20_21.h +128 -0
  546. onnx/version_converter/adapters/q_dq_21_20.h +77 -0
  547. onnx/version_converter/convert.h +67 -2
  548. onnx/version_converter.py +6 -142
  549. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/METADATA +18 -15
  550. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/RECORD +572 -406
  551. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/WHEEL +1 -1
  552. onnx/examples/Protobufs.ipynb +0 -639
  553. onnx/examples/check_model.ipynb +0 -128
  554. onnx/examples/load_model.ipynb +0 -116
  555. onnx/examples/make_model.ipynb +0 -176
  556. onnx/examples/np_array_tensorproto.ipynb +0 -136
  557. onnx/examples/resources/single_relu.onnx +0 -12
  558. onnx/examples/resources/single_relu_new.onnx +0 -12
  559. onnx/examples/resources/tensor.pb +0 -0
  560. onnx/examples/resources/two_transposes.onnx +0 -0
  561. onnx/examples/save_model.ipynb +0 -56
  562. onnx/examples/shape_inference.ipynb +0 -111
  563. onnx/test/reference_evaluator_backend_test.py +0 -876
  564. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_1.pb +0 -0
  565. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_4.pb +0 -0
  566. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_6.pb +0 -0
  567. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_0.pb +0 -0
  568. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_2.pb +0 -0
  569. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_3.pb +0 -0
  570. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_5.pb +0 -0
  571. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_7.pb +0 -0
  572. /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/output_0.pb +0 -0
  573. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_1.pb +0 -0
  574. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_2.pb +0 -0
  575. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_4.pb +0 -0
  576. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_5.pb +0 -0
  577. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_6.pb +0 -0
  578. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_7.pb +0 -0
  579. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/input_0.pb +0 -0
  580. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/input_3.pb +0 -0
  581. /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/output_0.pb +0 -0
  582. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/LICENSE +0 -0
  583. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/entry_points.txt +0 -0
  584. {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/top_level.txt +0 -0
onnx/onnx_ml_pb2.pyi ADDED
@@ -0,0 +1,663 @@
1
+ # @generated by protoc-gen-mypy.py. Do not edit!
2
+ # mypy: disable-error-code=override
3
+ from typing import (
4
+ Iterable,
5
+ List,
6
+ Optional as OptionalType,
7
+ Tuple,
8
+ cast,
9
+ )
10
+
11
+ from google.protobuf.message import ( # type: ignore
12
+ Message,
13
+ )
14
+
15
+ from google.protobuf.internal.containers import ( # type: ignore
16
+ RepeatedCompositeFieldContainer,
17
+ RepeatedScalarFieldContainer,
18
+ )
19
+
20
+ class Version(int):
21
+ @classmethod
22
+ def Name(cls, number: int) -> str: ...
23
+ @classmethod
24
+ def Value(cls, name: str) -> int: ...
25
+ @classmethod
26
+ def keys(cls) -> List[str]: ...
27
+ @classmethod
28
+ def values(cls) -> List[int]: ...
29
+ @classmethod
30
+ def items(cls) -> List[Tuple[str, int]]: ...
31
+ _START_VERSION = cast(Version, 0)
32
+ IR_VERSION_2017_10_10 = cast(Version, 1)
33
+ IR_VERSION_2017_10_30 = cast(Version, 2)
34
+ IR_VERSION_2017_11_3 = cast(Version, 3)
35
+ IR_VERSION_2019_1_22 = cast(Version, 4)
36
+ IR_VERSION_2019_3_18 = cast(Version, 5)
37
+ IR_VERSION_2019_9_19 = cast(Version, 6)
38
+ IR_VERSION_2020_5_8 = cast(Version, 7)
39
+ IR_VERSION_2021_7_30 = cast(Version, 8)
40
+ IR_VERSION_2023_5_5 = cast(Version, 9)
41
+ IR_VERSION = cast(Version, 10)
42
+
43
+ class OperatorStatus(int):
44
+ @classmethod
45
+ def Name(cls, number: int) -> str: ...
46
+ @classmethod
47
+ def Value(cls, name: str) -> int: ...
48
+ @classmethod
49
+ def keys(cls) -> List[str]: ...
50
+ @classmethod
51
+ def values(cls) -> List[int]: ...
52
+ @classmethod
53
+ def items(cls) -> List[Tuple[str, int]]: ...
54
+ EXPERIMENTAL = cast(OperatorStatus, 0)
55
+ STABLE = cast(OperatorStatus, 1)
56
+
57
+ class AttributeProto(Message):
58
+ class AttributeType(int):
59
+ @classmethod
60
+ def Name(cls, number: int) -> str: ...
61
+ @classmethod
62
+ def Value(cls, name: str) -> int: ...
63
+ @classmethod
64
+ def keys(cls) -> List[str]: ...
65
+ @classmethod
66
+ def values(cls) -> List[int]: ...
67
+ @classmethod
68
+ def items(cls) -> List[Tuple[str, int]]: ...
69
+ UNDEFINED = cast(AttributeType, 0)
70
+ FLOAT = cast(AttributeType, 1)
71
+ INT = cast(AttributeType, 2)
72
+ STRING = cast(AttributeType, 3)
73
+ TENSOR = cast(AttributeType, 4)
74
+ GRAPH = cast(AttributeType, 5)
75
+ SPARSE_TENSOR = cast(AttributeType, 11)
76
+ TYPE_PROTO = cast(AttributeType, 13)
77
+ FLOATS = cast(AttributeType, 6)
78
+ INTS = cast(AttributeType, 7)
79
+ STRINGS = cast(AttributeType, 8)
80
+ TENSORS = cast(AttributeType, 9)
81
+ GRAPHS = cast(AttributeType, 10)
82
+ SPARSE_TENSORS = cast(AttributeType, 12)
83
+ TYPE_PROTOS = cast(AttributeType, 14)
84
+
85
+ name = ... # type: str
86
+ ref_attr_name = ... # type: str
87
+ doc_string = ... # type: str
88
+ type = ... # type: AttributeProto.AttributeType
89
+ f = ... # type: float
90
+ i = ... # type: int
91
+ s = ... # type: bytes
92
+ floats = ... # type: RepeatedScalarFieldContainer[float]
93
+ ints = ... # type: RepeatedScalarFieldContainer[int]
94
+ strings = ... # type: RepeatedScalarFieldContainer[bytes]
95
+
96
+ @property
97
+ def t(self) -> TensorProto: ...
98
+
99
+ @property
100
+ def g(self) -> GraphProto: ...
101
+
102
+ @property
103
+ def sparse_tensor(self) -> SparseTensorProto: ...
104
+
105
+ @property
106
+ def tp(self) -> TypeProto: ...
107
+
108
+ @property
109
+ def tensors(self) -> RepeatedCompositeFieldContainer[TensorProto]: ...
110
+
111
+ @property
112
+ def graphs(self) -> RepeatedCompositeFieldContainer[GraphProto]: ...
113
+
114
+ @property
115
+ def sparse_tensors(self) -> RepeatedCompositeFieldContainer[SparseTensorProto]: ...
116
+
117
+ @property
118
+ def type_protos(self) -> RepeatedCompositeFieldContainer[TypeProto]: ...
119
+
120
+ def __init__(self,
121
+ name : OptionalType[str] = None,
122
+ ref_attr_name : OptionalType[str] = None,
123
+ doc_string : OptionalType[str] = None,
124
+ type : OptionalType[AttributeProto.AttributeType] = None,
125
+ f : OptionalType[float] = None,
126
+ i : OptionalType[int] = None,
127
+ s : OptionalType[bytes] = None,
128
+ t : OptionalType[TensorProto] = None,
129
+ g : OptionalType[GraphProto] = None,
130
+ sparse_tensor : OptionalType[SparseTensorProto] = None,
131
+ tp : OptionalType[TypeProto] = None,
132
+ floats : OptionalType[Iterable[float]] = None,
133
+ ints : OptionalType[Iterable[int]] = None,
134
+ strings : OptionalType[Iterable[bytes]] = None,
135
+ tensors : OptionalType[Iterable[TensorProto]] = None,
136
+ graphs : OptionalType[Iterable[GraphProto]] = None,
137
+ sparse_tensors : OptionalType[Iterable[SparseTensorProto]] = None,
138
+ type_protos : OptionalType[Iterable[TypeProto]] = None,
139
+ ) -> None: ...
140
+ @classmethod
141
+ def FromString(cls, s: bytes) -> AttributeProto: ...
142
+ def MergeFrom(self, other_msg: Message) -> None: ...
143
+ def CopyFrom(self, other_msg: Message) -> None: ...
144
+
145
+ class ValueInfoProto(Message):
146
+ name = ... # type: str
147
+ doc_string = ... # type: str
148
+
149
+ @property
150
+ def type(self) -> TypeProto: ...
151
+
152
+ @property
153
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
154
+
155
+ def __init__(self,
156
+ name : OptionalType[str] = None,
157
+ type : OptionalType[TypeProto] = None,
158
+ doc_string : OptionalType[str] = None,
159
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
160
+ ) -> None: ...
161
+ @classmethod
162
+ def FromString(cls, s: bytes) -> ValueInfoProto: ...
163
+ def MergeFrom(self, other_msg: Message) -> None: ...
164
+ def CopyFrom(self, other_msg: Message) -> None: ...
165
+
166
+ class NodeProto(Message):
167
+ input = ... # type: RepeatedScalarFieldContainer[str]
168
+ output = ... # type: RepeatedScalarFieldContainer[str]
169
+ name = ... # type: str
170
+ op_type = ... # type: str
171
+ domain = ... # type: str
172
+ overload = ... # type: str
173
+ doc_string = ... # type: str
174
+
175
+ @property
176
+ def attribute(self) -> RepeatedCompositeFieldContainer[AttributeProto]: ...
177
+
178
+ @property
179
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
180
+
181
+ def __init__(self,
182
+ input : OptionalType[Iterable[str]] = None,
183
+ output : OptionalType[Iterable[str]] = None,
184
+ name : OptionalType[str] = None,
185
+ op_type : OptionalType[str] = None,
186
+ domain : OptionalType[str] = None,
187
+ overload : OptionalType[str] = None,
188
+ attribute : OptionalType[Iterable[AttributeProto]] = None,
189
+ doc_string : OptionalType[str] = None,
190
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
191
+ ) -> None: ...
192
+ @classmethod
193
+ def FromString(cls, s: bytes) -> NodeProto: ...
194
+ def MergeFrom(self, other_msg: Message) -> None: ...
195
+ def CopyFrom(self, other_msg: Message) -> None: ...
196
+
197
+ class TrainingInfoProto(Message):
198
+
199
+ @property
200
+ def initialization(self) -> GraphProto: ...
201
+
202
+ @property
203
+ def algorithm(self) -> GraphProto: ...
204
+
205
+ @property
206
+ def initialization_binding(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
207
+
208
+ @property
209
+ def update_binding(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
210
+
211
+ def __init__(self,
212
+ initialization : OptionalType[GraphProto] = None,
213
+ algorithm : OptionalType[GraphProto] = None,
214
+ initialization_binding : OptionalType[Iterable[StringStringEntryProto]] = None,
215
+ update_binding : OptionalType[Iterable[StringStringEntryProto]] = None,
216
+ ) -> None: ...
217
+ @classmethod
218
+ def FromString(cls, s: bytes) -> TrainingInfoProto: ...
219
+ def MergeFrom(self, other_msg: Message) -> None: ...
220
+ def CopyFrom(self, other_msg: Message) -> None: ...
221
+
222
+ class ModelProto(Message):
223
+ ir_version = ... # type: int
224
+ producer_name = ... # type: str
225
+ producer_version = ... # type: str
226
+ domain = ... # type: str
227
+ model_version = ... # type: int
228
+ doc_string = ... # type: str
229
+
230
+ @property
231
+ def opset_import(self) -> RepeatedCompositeFieldContainer[OperatorSetIdProto]: ...
232
+
233
+ @property
234
+ def graph(self) -> GraphProto: ...
235
+
236
+ @property
237
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
238
+
239
+ @property
240
+ def training_info(self) -> RepeatedCompositeFieldContainer[TrainingInfoProto]: ...
241
+
242
+ @property
243
+ def functions(self) -> RepeatedCompositeFieldContainer[FunctionProto]: ...
244
+
245
+ def __init__(self,
246
+ ir_version : OptionalType[int] = None,
247
+ opset_import : OptionalType[Iterable[OperatorSetIdProto]] = None,
248
+ producer_name : OptionalType[str] = None,
249
+ producer_version : OptionalType[str] = None,
250
+ domain : OptionalType[str] = None,
251
+ model_version : OptionalType[int] = None,
252
+ doc_string : OptionalType[str] = None,
253
+ graph : OptionalType[GraphProto] = None,
254
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
255
+ training_info : OptionalType[Iterable[TrainingInfoProto]] = None,
256
+ functions : OptionalType[Iterable[FunctionProto]] = None,
257
+ ) -> None: ...
258
+ @classmethod
259
+ def FromString(cls, s: bytes) -> ModelProto: ...
260
+ def MergeFrom(self, other_msg: Message) -> None: ...
261
+ def CopyFrom(self, other_msg: Message) -> None: ...
262
+
263
+ class StringStringEntryProto(Message):
264
+ key = ... # type: str
265
+ value = ... # type: str
266
+
267
+ def __init__(self,
268
+ key : OptionalType[str] = None,
269
+ value : OptionalType[str] = None,
270
+ ) -> None: ...
271
+ @classmethod
272
+ def FromString(cls, s: bytes) -> StringStringEntryProto: ...
273
+ def MergeFrom(self, other_msg: Message) -> None: ...
274
+ def CopyFrom(self, other_msg: Message) -> None: ...
275
+
276
+ class TensorAnnotation(Message):
277
+ tensor_name = ... # type: str
278
+
279
+ @property
280
+ def quant_parameter_tensor_names(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
281
+
282
+ def __init__(self,
283
+ tensor_name : OptionalType[str] = None,
284
+ quant_parameter_tensor_names : OptionalType[Iterable[StringStringEntryProto]] = None,
285
+ ) -> None: ...
286
+ @classmethod
287
+ def FromString(cls, s: bytes) -> TensorAnnotation: ...
288
+ def MergeFrom(self, other_msg: Message) -> None: ...
289
+ def CopyFrom(self, other_msg: Message) -> None: ...
290
+
291
+ class GraphProto(Message):
292
+ name = ... # type: str
293
+ doc_string = ... # type: str
294
+
295
+ @property
296
+ def node(self) -> RepeatedCompositeFieldContainer[NodeProto]: ...
297
+
298
+ @property
299
+ def initializer(self) -> RepeatedCompositeFieldContainer[TensorProto]: ...
300
+
301
+ @property
302
+ def sparse_initializer(self) -> RepeatedCompositeFieldContainer[SparseTensorProto]: ...
303
+
304
+ @property
305
+ def input(self) -> RepeatedCompositeFieldContainer[ValueInfoProto]: ...
306
+
307
+ @property
308
+ def output(self) -> RepeatedCompositeFieldContainer[ValueInfoProto]: ...
309
+
310
+ @property
311
+ def value_info(self) -> RepeatedCompositeFieldContainer[ValueInfoProto]: ...
312
+
313
+ @property
314
+ def quantization_annotation(self) -> RepeatedCompositeFieldContainer[TensorAnnotation]: ...
315
+
316
+ @property
317
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
318
+
319
+ def __init__(self,
320
+ node : OptionalType[Iterable[NodeProto]] = None,
321
+ name : OptionalType[str] = None,
322
+ initializer : OptionalType[Iterable[TensorProto]] = None,
323
+ sparse_initializer : OptionalType[Iterable[SparseTensorProto]] = None,
324
+ doc_string : OptionalType[str] = None,
325
+ input : OptionalType[Iterable[ValueInfoProto]] = None,
326
+ output : OptionalType[Iterable[ValueInfoProto]] = None,
327
+ value_info : OptionalType[Iterable[ValueInfoProto]] = None,
328
+ quantization_annotation : OptionalType[Iterable[TensorAnnotation]] = None,
329
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
330
+ ) -> None: ...
331
+ @classmethod
332
+ def FromString(cls, s: bytes) -> GraphProto: ...
333
+ def MergeFrom(self, other_msg: Message) -> None: ...
334
+ def CopyFrom(self, other_msg: Message) -> None: ...
335
+
336
+ class TensorProto(Message):
337
+ class DataType(int):
338
+ @classmethod
339
+ def Name(cls, number: int) -> str: ...
340
+ @classmethod
341
+ def Value(cls, name: str) -> int: ...
342
+ @classmethod
343
+ def keys(cls) -> List[str]: ...
344
+ @classmethod
345
+ def values(cls) -> List[int]: ...
346
+ @classmethod
347
+ def items(cls) -> List[Tuple[str, int]]: ...
348
+ UNDEFINED = cast(DataType, 0)
349
+ FLOAT = cast(DataType, 1)
350
+ UINT8 = cast(DataType, 2)
351
+ INT8 = cast(DataType, 3)
352
+ UINT16 = cast(DataType, 4)
353
+ INT16 = cast(DataType, 5)
354
+ INT32 = cast(DataType, 6)
355
+ INT64 = cast(DataType, 7)
356
+ STRING = cast(DataType, 8)
357
+ BOOL = cast(DataType, 9)
358
+ FLOAT16 = cast(DataType, 10)
359
+ DOUBLE = cast(DataType, 11)
360
+ UINT32 = cast(DataType, 12)
361
+ UINT64 = cast(DataType, 13)
362
+ COMPLEX64 = cast(DataType, 14)
363
+ COMPLEX128 = cast(DataType, 15)
364
+ BFLOAT16 = cast(DataType, 16)
365
+ FLOAT8E4M3FN = cast(DataType, 17)
366
+ FLOAT8E4M3FNUZ = cast(DataType, 18)
367
+ FLOAT8E5M2 = cast(DataType, 19)
368
+ FLOAT8E5M2FNUZ = cast(DataType, 20)
369
+ UINT4 = cast(DataType, 21)
370
+ INT4 = cast(DataType, 22)
371
+
372
+ class DataLocation(int):
373
+ @classmethod
374
+ def Name(cls, number: int) -> str: ...
375
+ @classmethod
376
+ def Value(cls, name: str) -> int: ...
377
+ @classmethod
378
+ def keys(cls) -> List[str]: ...
379
+ @classmethod
380
+ def values(cls) -> List[int]: ...
381
+ @classmethod
382
+ def items(cls) -> List[Tuple[str, int]]: ...
383
+ DEFAULT = cast(DataLocation, 0)
384
+ EXTERNAL = cast(DataLocation, 1)
385
+
386
+ class Segment(Message):
387
+ begin = ... # type: int
388
+ end = ... # type: int
389
+
390
+ def __init__(self,
391
+ begin : OptionalType[int] = None,
392
+ end : OptionalType[int] = None,
393
+ ) -> None: ...
394
+ @classmethod
395
+ def FromString(cls, s: bytes) -> TensorProto.Segment: ...
396
+ def MergeFrom(self, other_msg: Message) -> None: ...
397
+ def CopyFrom(self, other_msg: Message) -> None: ...
398
+
399
+ dims = ... # type: RepeatedScalarFieldContainer[int]
400
+ data_type = ... # type: int
401
+ float_data = ... # type: RepeatedScalarFieldContainer[float]
402
+ int32_data = ... # type: RepeatedScalarFieldContainer[int]
403
+ string_data = ... # type: RepeatedScalarFieldContainer[bytes]
404
+ int64_data = ... # type: RepeatedScalarFieldContainer[int]
405
+ name = ... # type: str
406
+ doc_string = ... # type: str
407
+ raw_data = ... # type: bytes
408
+ data_location = ... # type: TensorProto.DataLocation
409
+ double_data = ... # type: RepeatedScalarFieldContainer[float]
410
+ uint64_data = ... # type: RepeatedScalarFieldContainer[int]
411
+
412
+ @property
413
+ def segment(self) -> TensorProto.Segment: ...
414
+
415
+ @property
416
+ def external_data(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
417
+
418
+ @property
419
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
420
+
421
+ def __init__(self,
422
+ dims : OptionalType[Iterable[int]] = None,
423
+ data_type : OptionalType[int] = None,
424
+ segment : OptionalType[TensorProto.Segment] = None,
425
+ float_data : OptionalType[Iterable[float]] = None,
426
+ int32_data : OptionalType[Iterable[int]] = None,
427
+ string_data : OptionalType[Iterable[bytes]] = None,
428
+ int64_data : OptionalType[Iterable[int]] = None,
429
+ name : OptionalType[str] = None,
430
+ doc_string : OptionalType[str] = None,
431
+ raw_data : OptionalType[bytes] = None,
432
+ external_data : OptionalType[Iterable[StringStringEntryProto]] = None,
433
+ data_location : OptionalType[TensorProto.DataLocation] = None,
434
+ double_data : OptionalType[Iterable[float]] = None,
435
+ uint64_data : OptionalType[Iterable[int]] = None,
436
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
437
+ ) -> None: ...
438
+ @classmethod
439
+ def FromString(cls, s: bytes) -> TensorProto: ...
440
+ def MergeFrom(self, other_msg: Message) -> None: ...
441
+ def CopyFrom(self, other_msg: Message) -> None: ...
442
+
443
+ class SparseTensorProto(Message):
444
+ dims = ... # type: RepeatedScalarFieldContainer[int]
445
+
446
+ @property
447
+ def values(self) -> TensorProto: ...
448
+
449
+ @property
450
+ def indices(self) -> TensorProto: ...
451
+
452
+ def __init__(self,
453
+ values : OptionalType[TensorProto] = None,
454
+ indices : OptionalType[TensorProto] = None,
455
+ dims : OptionalType[Iterable[int]] = None,
456
+ ) -> None: ...
457
+ @classmethod
458
+ def FromString(cls, s: bytes) -> SparseTensorProto: ...
459
+ def MergeFrom(self, other_msg: Message) -> None: ...
460
+ def CopyFrom(self, other_msg: Message) -> None: ...
461
+
462
+ class TensorShapeProto(Message):
463
+ class Dimension(Message):
464
+ dim_value = ... # type: int
465
+ dim_param = ... # type: str
466
+ denotation = ... # type: str
467
+
468
+ def __init__(self,
469
+ dim_value : OptionalType[int] = None,
470
+ dim_param : OptionalType[str] = None,
471
+ denotation : OptionalType[str] = None,
472
+ ) -> None: ...
473
+ @classmethod
474
+ def FromString(cls, s: bytes) -> TensorShapeProto.Dimension: ...
475
+ def MergeFrom(self, other_msg: Message) -> None: ...
476
+ def CopyFrom(self, other_msg: Message) -> None: ...
477
+
478
+
479
+ @property
480
+ def dim(self) -> RepeatedCompositeFieldContainer[TensorShapeProto.Dimension]: ...
481
+
482
+ def __init__(self,
483
+ dim : OptionalType[Iterable[TensorShapeProto.Dimension]] = None,
484
+ ) -> None: ...
485
+ @classmethod
486
+ def FromString(cls, s: bytes) -> TensorShapeProto: ...
487
+ def MergeFrom(self, other_msg: Message) -> None: ...
488
+ def CopyFrom(self, other_msg: Message) -> None: ...
489
+
490
+ class TypeProto(Message):
491
+ class Tensor(Message):
492
+ elem_type = ... # type: int
493
+
494
+ @property
495
+ def shape(self) -> TensorShapeProto: ...
496
+
497
+ def __init__(self,
498
+ elem_type : OptionalType[int] = None,
499
+ shape : OptionalType[TensorShapeProto] = None,
500
+ ) -> None: ...
501
+ @classmethod
502
+ def FromString(cls, s: bytes) -> TypeProto.Tensor: ...
503
+ def MergeFrom(self, other_msg: Message) -> None: ...
504
+ def CopyFrom(self, other_msg: Message) -> None: ...
505
+
506
+ class Sequence(Message):
507
+
508
+ @property
509
+ def elem_type(self) -> TypeProto: ...
510
+
511
+ def __init__(self,
512
+ elem_type : OptionalType[TypeProto] = None,
513
+ ) -> None: ...
514
+ @classmethod
515
+ def FromString(cls, s: bytes) -> TypeProto.Sequence: ...
516
+ def MergeFrom(self, other_msg: Message) -> None: ...
517
+ def CopyFrom(self, other_msg: Message) -> None: ...
518
+
519
+ class Map(Message):
520
+ key_type = ... # type: int
521
+
522
+ @property
523
+ def value_type(self) -> TypeProto: ...
524
+
525
+ def __init__(self,
526
+ key_type : OptionalType[int] = None,
527
+ value_type : OptionalType[TypeProto] = None,
528
+ ) -> None: ...
529
+ @classmethod
530
+ def FromString(cls, s: bytes) -> TypeProto.Map: ...
531
+ def MergeFrom(self, other_msg: Message) -> None: ...
532
+ def CopyFrom(self, other_msg: Message) -> None: ...
533
+
534
+ class Optional(Message):
535
+
536
+ @property
537
+ def elem_type(self) -> TypeProto: ...
538
+
539
+ def __init__(self,
540
+ elem_type : OptionalType[TypeProto] = None,
541
+ ) -> None: ...
542
+ @classmethod
543
+ def FromString(cls, s: bytes) -> TypeProto.Optional: ...
544
+ def MergeFrom(self, other_msg: Message) -> None: ...
545
+ def CopyFrom(self, other_msg: Message) -> None: ...
546
+
547
+ class SparseTensor(Message):
548
+ elem_type = ... # type: int
549
+
550
+ @property
551
+ def shape(self) -> TensorShapeProto: ...
552
+
553
+ def __init__(self,
554
+ elem_type : OptionalType[int] = None,
555
+ shape : OptionalType[TensorShapeProto] = None,
556
+ ) -> None: ...
557
+ @classmethod
558
+ def FromString(cls, s: bytes) -> TypeProto.SparseTensor: ...
559
+ def MergeFrom(self, other_msg: Message) -> None: ...
560
+ def CopyFrom(self, other_msg: Message) -> None: ...
561
+
562
+ class Opaque(Message):
563
+ domain = ... # type: str
564
+ name = ... # type: str
565
+
566
+ def __init__(self,
567
+ domain : OptionalType[str] = None,
568
+ name : OptionalType[str] = None,
569
+ ) -> None: ...
570
+ @classmethod
571
+ def FromString(cls, s: bytes) -> TypeProto.Opaque: ...
572
+ def MergeFrom(self, other_msg: Message) -> None: ...
573
+ def CopyFrom(self, other_msg: Message) -> None: ...
574
+
575
+ denotation = ... # type: str
576
+
577
+ @property
578
+ def tensor_type(self) -> TypeProto.Tensor: ...
579
+
580
+ @property
581
+ def sequence_type(self) -> TypeProto.Sequence: ...
582
+
583
+ @property
584
+ def map_type(self) -> TypeProto.Map: ...
585
+
586
+ @property
587
+ def optional_type(self) -> TypeProto.Optional: ...
588
+
589
+ @property
590
+ def sparse_tensor_type(self) -> TypeProto.SparseTensor: ...
591
+
592
+ @property
593
+ def opaque_type(self) -> TypeProto.Opaque: ...
594
+
595
+ def __init__(self,
596
+ tensor_type : OptionalType[TypeProto.Tensor] = None,
597
+ sequence_type : OptionalType[TypeProto.Sequence] = None,
598
+ map_type : OptionalType[TypeProto.Map] = None,
599
+ optional_type : OptionalType[TypeProto.Optional] = None,
600
+ sparse_tensor_type : OptionalType[TypeProto.SparseTensor] = None,
601
+ opaque_type : OptionalType[TypeProto.Opaque] = None,
602
+ denotation : OptionalType[str] = None,
603
+ ) -> None: ...
604
+ @classmethod
605
+ def FromString(cls, s: bytes) -> TypeProto: ...
606
+ def MergeFrom(self, other_msg: Message) -> None: ...
607
+ def CopyFrom(self, other_msg: Message) -> None: ...
608
+
609
+ class OperatorSetIdProto(Message):
610
+ domain = ... # type: str
611
+ version = ... # type: int
612
+
613
+ def __init__(self,
614
+ domain : OptionalType[str] = None,
615
+ version : OptionalType[int] = None,
616
+ ) -> None: ...
617
+ @classmethod
618
+ def FromString(cls, s: bytes) -> OperatorSetIdProto: ...
619
+ def MergeFrom(self, other_msg: Message) -> None: ...
620
+ def CopyFrom(self, other_msg: Message) -> None: ...
621
+
622
+ class FunctionProto(Message):
623
+ name = ... # type: str
624
+ input = ... # type: RepeatedScalarFieldContainer[str]
625
+ output = ... # type: RepeatedScalarFieldContainer[str]
626
+ attribute = ... # type: RepeatedScalarFieldContainer[str]
627
+ doc_string = ... # type: str
628
+ domain = ... # type: str
629
+ overload = ... # type: str
630
+
631
+ @property
632
+ def attribute_proto(self) -> RepeatedCompositeFieldContainer[AttributeProto]: ...
633
+
634
+ @property
635
+ def node(self) -> RepeatedCompositeFieldContainer[NodeProto]: ...
636
+
637
+ @property
638
+ def opset_import(self) -> RepeatedCompositeFieldContainer[OperatorSetIdProto]: ...
639
+
640
+ @property
641
+ def value_info(self) -> RepeatedCompositeFieldContainer[ValueInfoProto]: ...
642
+
643
+ @property
644
+ def metadata_props(self) -> RepeatedCompositeFieldContainer[StringStringEntryProto]: ...
645
+
646
+ def __init__(self,
647
+ name : OptionalType[str] = None,
648
+ input : OptionalType[Iterable[str]] = None,
649
+ output : OptionalType[Iterable[str]] = None,
650
+ attribute : OptionalType[Iterable[str]] = None,
651
+ attribute_proto : OptionalType[Iterable[AttributeProto]] = None,
652
+ node : OptionalType[Iterable[NodeProto]] = None,
653
+ doc_string : OptionalType[str] = None,
654
+ opset_import : OptionalType[Iterable[OperatorSetIdProto]] = None,
655
+ domain : OptionalType[str] = None,
656
+ overload : OptionalType[str] = None,
657
+ value_info : OptionalType[Iterable[ValueInfoProto]] = None,
658
+ metadata_props : OptionalType[Iterable[StringStringEntryProto]] = None,
659
+ ) -> None: ...
660
+ @classmethod
661
+ def FromString(cls, s: bytes) -> FunctionProto: ...
662
+ def MergeFrom(self, other_msg: Message) -> None: ...
663
+ def CopyFrom(self, other_msg: Message) -> None: ...