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.
- onnx/__init__.py +10 -10
- onnx/backend/base.py +13 -14
- onnx/backend/sample/ops/abs.py +1 -1
- onnx/backend/test/case/model/__init__.py +0 -1
- onnx/backend/test/case/node/ai_onnx_ml/tree_ensemble.py +122 -0
- onnx/backend/test/case/node/averagepool.py +15 -30
- onnx/backend/test/case/node/cast.py +88 -11
- onnx/backend/test/case/node/dequantizelinear.py +155 -0
- onnx/backend/test/case/node/groupnormalization.py +13 -9
- onnx/backend/test/case/node/gru.py +2 -2
- onnx/backend/test/case/node/isinf.py +4 -4
- onnx/backend/test/case/node/isnan.py +2 -2
- onnx/backend/test/case/node/lppool.py +8 -16
- onnx/backend/test/case/node/lstm.py +1 -1
- onnx/backend/test/case/node/maxpool.py +40 -34
- onnx/backend/test/case/node/pow.py +1 -1
- onnx/backend/test/case/node/qlinearmatmul.py +143 -109
- onnx/backend/test/case/node/quantizelinear.py +298 -7
- onnx/backend/test/case/node/reducemax.py +26 -0
- onnx/backend/test/case/node/rnn.py +1 -1
- onnx/backend/test/case/node/scan.py +6 -2
- onnx/backend/test/case/node/scatterelements.py +1 -1
- onnx/backend/test/case/node/topk.py +1 -1
- onnx/backend/test/case/utils.py +1 -3
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/model.onnx +0 -0
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_set_membership/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/model.onnx +0 -0
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_ai_onnx_ml_tree_ensemble_single_tree/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_BFLOAT16_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_DOUBLE_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_DOUBLE_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_DOUBLE/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_INT4/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT16_to_UINT4/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FNUZ_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E4M3FN_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2FNUZ_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT8E5M2_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_BFLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_DOUBLE/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_INT4/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_STRING/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_FLOAT_to_UINT4/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_cast_INT4_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_INT8/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_INT4_to_INT8/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_cast_INT4_to_INT8/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_STRING_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_FLOAT16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_UINT4_to_UINT8/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +2 -2
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT16_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FN/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_castlike_BFLOAT16_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_BFLOAT16_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT16_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_DOUBLE_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT16_to_DOUBLE/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT16_to_DOUBLE_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT16_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT16_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FN_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E4M3FN_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E5M2FNUZ_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E5M2FNUZ_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E5M2_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT8E5M2_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_BFLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_BFLOAT16_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_DOUBLE/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_DOUBLE_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT16/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT16_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FN/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FNUZ_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E4M3FN_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_FLOAT8E5M2_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_STRING/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_FLOAT_to_STRING_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_STRING_to_FLOAT/model.onnx +0 -0
- onnx/backend/test/data/node/test_castlike_STRING_to_FLOAT_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_constant/model.onnx +0 -0
- onnx/backend/test/data/node/test_constant_pad/model.onnx +0 -0
- onnx/backend/test/data/node/test_constant_pad_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_constant_pad_negative_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_constantofshape_float_ones/model.onnx +0 -0
- onnx/backend/test/data/node/test_constantofshape_int_shape_zero/model.onnx +0 -0
- onnx/backend/test/data/node/test_constantofshape_int_zeros/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_axis/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_blocked/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_blocked/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_e4m3fn/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_e4m3fn_float16/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_e4m3fn_zero_point/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_e5m2/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int16/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int4/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_int4/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint16/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint4/model.onnx +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_dequantizelinear_uint4/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_edge_pad/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_axis0/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_axis1/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_axis2/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_axis3/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_default_axis/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_negative_axis1/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_negative_axis2/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_negative_axis3/model.onnx +0 -0
- onnx/backend/test/data/node/test_flatten_negative_axis4/model.onnx +0 -0
- onnx/backend/test/data/node/test_group_normalization_epsilon/model.onnx +0 -0
- onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_0.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_1.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/input_2.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_0.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_1.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/input_2.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_epsilon_expanded/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_group_normalization_example/model.onnx +0 -0
- onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/input_1.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/input_2.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_example/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_group_normalization_example_expanded/model.onnx +0 -0
- onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/input_1.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/input_2.pb +1 -1
- onnx/backend/test/data/node/test_group_normalization_example_expanded/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_identity/model.onnx +0 -0
- onnx/backend/test/data/node/test_identity_sequence/model.onnx +0 -0
- onnx/backend/test/data/node/test_lrn_default/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/model.onnx +0 -0
- onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_maxpool_2d_ceil_output_size_reduce_by_one/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_mvn/test_data_set_0/output_0.pb +1 -1
- onnx/backend/test/data/node/test_mvn_expanded/test_data_set_0/output_0.pb +1 -1
- onnx/backend/test/data/node/test_mvn_expanded_ver18/test_data_set_0/output_0.pb +1 -1
- onnx/backend/test/data/node/test_pow/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_1.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_4.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_6.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float16/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_int8_float32/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_1.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_4.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float16/test_data_set_0/input_6.pb +2 -0
- onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float32}/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_2D_uint8_float32/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_1.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_4.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_6.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float16/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_1.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_4.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_6.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_int8_float32/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_1.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_4.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_6.pb +2 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float16/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float32}/model.onnx +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_1.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_3.pb +0 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_4.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_5.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_6.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/input_7.pb +1 -0
- onnx/backend/test/data/node/test_qlinearmatmul_3D_uint8_float32/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_axis/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_asymmetric/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_blocked_symmetric/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e4m3fn/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e4m3fn/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e4m3fn/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e5m2/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e5m2/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_e5m2/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int16/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/input_2.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int4/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear_int4/test_data_set_0/output_0.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear_uint16/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear_uint16/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint4/model.onnx +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_0.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_1.pb +0 -0
- onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/input_2.pb +1 -0
- onnx/backend/test/data/node/test_quantizelinear_uint4/test_data_set_0/output_0.pb +0 -0
- onnx/backend/test/data/node/test_reflect_pad/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_allowzero_reordered/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_extended_dims/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_negative_dim/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_negative_extended_dims/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_one_dim/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_reduced_dims/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_reordered_all_dims/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_reordered_last_dims/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_zero_and_negative_dim/model.onnx +0 -0
- onnx/backend/test/data/node/test_reshape_zero_dim/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_clip_end/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_clip_start/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_end_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_end_negative_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_example/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_start_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_start_1_end_2/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_start_1_end_negative_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_shape_start_negative_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_size/model.onnx +0 -0
- onnx/backend/test/data/node/test_size_example/model.onnx +0 -0
- onnx/backend/test/data/node/test_squeeze/model.onnx +0 -0
- onnx/backend/test/data/node/test_squeeze_negative_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_0/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_2/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_3/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_4/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_all_permutations_5/model.onnx +0 -0
- onnx/backend/test/data/node/test_transpose_default/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_axis_0/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_axis_1/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_axis_2/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_negative_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_three_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_two_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_unsqueeze_unsorted_axes/model.onnx +0 -0
- onnx/backend/test/data/node/test_wrap_pad/model.onnx +0 -0
- onnx/backend/test/loader/__init__.py +0 -1
- onnx/backend/test/runner/__init__.py +43 -15
- onnx/checker.cc +104 -99
- onnx/checker.h +23 -3
- onnx/checker.py +56 -20
- onnx/common/assertions.cc +10 -5
- onnx/common/common.h +19 -0
- onnx/common/file_utils.h +3 -1
- onnx/common/interned_strings.h +7 -1
- onnx/common/ir.h +30 -7
- onnx/common/ir_pb_converter.cc +6 -0
- onnx/common/path.h +18 -2
- onnx/common/proto_util.h +43 -0
- onnx/common/version.h +1 -1
- onnx/cpp2py_export.cc +88 -56
- onnx/defs/__init__.py +29 -8
- onnx/defs/controlflow/defs.cc +16 -16
- onnx/defs/controlflow/old.cc +177 -0
- onnx/defs/data_propagators.h +2 -0
- onnx/defs/data_type_utils.cc +2 -0
- onnx/defs/generator/defs.cc +6 -4
- onnx/defs/generator/old.cc +115 -0
- onnx/defs/math/defs.cc +37 -142
- onnx/defs/math/old.cc +96 -12
- onnx/defs/math/utils.cc +127 -0
- onnx/defs/math/utils.h +8 -0
- onnx/defs/nn/defs.cc +72 -59
- onnx/defs/nn/old.cc +181 -2
- onnx/defs/object_detection/defs.cc +2 -2
- onnx/defs/object_detection/old.cc +2 -2
- onnx/defs/operator_sets.h +51 -0
- onnx/defs/operator_sets_ml.h +14 -0
- onnx/defs/parser.cc +112 -54
- onnx/defs/parser.h +14 -2
- onnx/defs/printer.cc +14 -7
- onnx/defs/quantization/defs.cc +111 -44
- onnx/defs/quantization/old.cc +130 -1
- onnx/defs/schema.cc +62 -18
- onnx/defs/schema.h +194 -48
- onnx/defs/shape_inference.cc +28 -19
- onnx/defs/shape_inference.h +2 -0
- onnx/defs/tensor/defs.cc +54 -96
- onnx/defs/tensor/old.cc +939 -34
- onnx/defs/tensor/utils.cc +6 -3
- onnx/defs/tensor/utils.h +5 -1
- onnx/defs/tensor_proto_util.cc +2 -0
- onnx/defs/tensor_util.cc +2 -0
- onnx/defs/traditionalml/defs.cc +273 -117
- onnx/defs/traditionalml/old.cc +329 -14
- onnx/defs/traditionalml/utils.h +27 -0
- onnx/external_data_helper.py +12 -26
- onnx/helper.py +242 -169
- onnx/hub.py +104 -70
- onnx/inliner/inliner.cc +89 -31
- onnx/inliner/inliner.h +5 -0
- onnx/inliner.py +2 -0
- onnx/mapping.py +9 -0
- onnx/model_container.py +346 -0
- onnx/numpy_helper.py +100 -38
- onnx/onnx-ml.proto +50 -13
- onnx/onnx.in.proto +50 -13
- onnx/onnx.proto +50 -13
- onnx/onnx_cpp2py_export/__init__.pyi +5 -0
- onnx/onnx_cpp2py_export/checker.pyi +21 -0
- onnx/onnx_cpp2py_export/defs.pyi +202 -0
- onnx/onnx_cpp2py_export/inliner.pyi +19 -0
- onnx/onnx_cpp2py_export/parser.pyi +32 -0
- onnx/onnx_cpp2py_export/printer.pyi +3 -0
- onnx/onnx_cpp2py_export/shape_inference.pyi +16 -0
- onnx/onnx_cpp2py_export/version_converter.pyi +4 -0
- onnx/onnx_cpp2py_export.cp311-win_amd64.pyd +0 -0
- onnx/onnx_data_pb2.pyi +146 -0
- onnx/onnx_ml_pb2.py +52 -52
- onnx/onnx_ml_pb2.pyi +663 -0
- onnx/onnx_operators_ml_pb2.pyi +67 -0
- onnx/reference/__init__.py +2 -0
- onnx/reference/custom_element_types.py +2 -0
- onnx/reference/op_run.py +166 -121
- onnx/reference/ops/_op.py +27 -50
- onnx/reference/ops/_op_list.py +36 -24
- onnx/reference/ops/aionnx_preview_training/_op_list.py +15 -8
- onnx/reference/ops/aionnxml/_common_classifier.py +3 -5
- onnx/reference/ops/aionnxml/_op_list.py +16 -8
- onnx/reference/ops/aionnxml/op_array_feature_extractor.py +4 -6
- onnx/reference/ops/aionnxml/op_linear_classifier.py +1 -2
- onnx/reference/ops/aionnxml/op_normalizer.py +3 -3
- onnx/reference/ops/aionnxml/op_svm_helper.py +1 -3
- onnx/reference/ops/aionnxml/op_svm_regressor.py +1 -3
- onnx/reference/ops/aionnxml/op_tree_ensemble.py +257 -0
- onnx/reference/ops/aionnxml/op_tree_ensemble_helper.py +2 -6
- onnx/reference/ops/aionnxml/op_tree_ensemble_regressor.py +4 -4
- onnx/reference/ops/experimental/_op_list.py +15 -8
- onnx/reference/ops/op_blackman_window.py +5 -6
- onnx/reference/ops/op_cast.py +22 -0
- onnx/reference/ops/op_cast_like.py +6 -0
- onnx/reference/ops/op_clip.py +5 -8
- onnx/reference/ops/op_col2im.py +1 -3
- onnx/reference/ops/op_constant.py +7 -1
- onnx/reference/ops/op_dequantize_linear.py +43 -40
- onnx/reference/ops/op_det.py +1 -1
- onnx/reference/ops/op_dynamic_quantize_linear.py +2 -2
- onnx/reference/ops/op_grid_sample.py +2 -4
- onnx/reference/ops/op_hamming_window.py +3 -6
- onnx/reference/ops/op_hann_window.py +3 -6
- onnx/reference/ops/op_if.py +4 -3
- onnx/reference/ops/op_loop.py +7 -9
- onnx/reference/ops/op_matmul.py +1 -2
- onnx/reference/ops/op_max_pool.py +5 -0
- onnx/reference/ops/op_optional.py +1 -1
- onnx/reference/ops/op_pool_common.py +3 -6
- onnx/reference/ops/op_qlinear_matmul.py +2 -2
- onnx/reference/ops/op_quantize_linear.py +166 -71
- onnx/reference/ops/op_resize.py +25 -21
- onnx/reference/ops/op_rnn.py +20 -12
- onnx/reference/ops/op_scan.py +23 -15
- onnx/reference/ops/op_scatter_elements.py +7 -6
- onnx/reference/ops/op_stft.py +3 -5
- onnx/reference/ops/op_string_normalizer.py +7 -7
- onnx/reference/ops/op_tfidf_vectorizer.py +7 -8
- onnx/reference/ops/op_topk.py +9 -11
- onnx/reference/ops/op_unique.py +1 -1
- onnx/reference/reference_evaluator.py +119 -63
- onnx/shape_inference/implementation.cc +160 -127
- onnx/shape_inference.py +11 -10
- onnx/subbyte.py +72 -0
- onnx/test/__init__.pyi +6 -0
- onnx/test/checker_test.py +21 -1
- onnx/test/compose_test.py +26 -74
- onnx/test/cpp/inliner_test.cc +76 -1
- onnx/test/cpp/ir_test.cc +60 -0
- onnx/test/cpp/parser_test.cc +106 -0
- onnx/test/function_test.py +1 -3
- onnx/test/helper_test.py +64 -4
- onnx/test/model_container_refeval_test.py +139 -0
- onnx/test/model_container_test.py +136 -0
- onnx/test/model_inference_test.py +44 -0
- onnx/test/reference_evaluator_ml_test.py +448 -47
- onnx/test/reference_evaluator_model_test.py +130 -0
- onnx/test/reference_evaluator_test.py +901 -14
- onnx/test/schema_test.py +166 -1
- onnx/test/shape_inference_test.py +285 -6
- onnx/test/symbolic_shape_test.py +3 -8
- onnx/test/test_backend_onnxruntime.py +238 -224
- onnx/test/test_backend_reference.py +11 -0
- onnx/test/test_external_data.py +51 -2
- onnx/test/version_converter/automatic_conversion_test_base.py +2 -1
- onnx/test/version_converter/automatic_upgrade_test.py +12 -10
- onnx/test/version_converter_test.py +166 -0
- onnx/tools/replace_constants.py +23 -26
- onnx/tools/update_model_dims.py +1 -2
- onnx/version.py +2 -2
- onnx/version_converter/adapters/group_normalization_20_21.h +128 -0
- onnx/version_converter/adapters/q_dq_21_20.h +77 -0
- onnx/version_converter/convert.h +67 -2
- onnx/version_converter.py +6 -142
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/METADATA +18 -15
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/RECORD +572 -406
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/WHEEL +1 -1
- onnx/examples/Protobufs.ipynb +0 -639
- onnx/examples/check_model.ipynb +0 -128
- onnx/examples/load_model.ipynb +0 -116
- onnx/examples/make_model.ipynb +0 -176
- onnx/examples/np_array_tensorproto.ipynb +0 -136
- onnx/examples/resources/single_relu.onnx +0 -12
- onnx/examples/resources/single_relu_new.onnx +0 -12
- onnx/examples/resources/tensor.pb +0 -0
- onnx/examples/resources/two_transposes.onnx +0 -0
- onnx/examples/save_model.ipynb +0 -56
- onnx/examples/shape_inference.ipynb +0 -111
- onnx/test/reference_evaluator_backend_test.py +0 -876
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_1.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_4.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_int8_float32}/test_data_set_0/input_6.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_0.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_2.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_3.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_5.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/input_7.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_2D → test_qlinearmatmul_2D_uint8_float16}/test_data_set_0/output_0.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_1.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_2.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_4.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_5.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_6.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_2D_uint8_float32}/test_data_set_0/input_7.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/input_0.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/input_3.pb +0 -0
- /onnx/backend/test/data/node/{test_qlinearmatmul_3D → test_qlinearmatmul_3D_uint8_float16}/test_data_set_0/output_0.pb +0 -0
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/LICENSE +0 -0
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/entry_points.txt +0 -0
- {onnx-1.15.0.dist-info → onnx-1.16.1.dist-info}/top_level.txt +0 -0
onnx/onnx-ml.proto
CHANGED
|
@@ -108,7 +108,11 @@ enum Version {
|
|
|
108
108
|
// IR VERSION 9 published on May 5, 2023
|
|
109
109
|
// Added AttributeProto to FunctionProto so that default attribute values can be set.
|
|
110
110
|
// Added FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ.
|
|
111
|
-
|
|
111
|
+
IR_VERSION_2023_5_5 = 0x0000000000000009;
|
|
112
|
+
|
|
113
|
+
// IR VERSION 10 published on TBD
|
|
114
|
+
// Added UINT4, INT4.
|
|
115
|
+
IR_VERSION = 0x000000000000000A;
|
|
112
116
|
}
|
|
113
117
|
|
|
114
118
|
// Attributes
|
|
@@ -118,6 +122,8 @@ enum Version {
|
|
|
118
122
|
// An AttributeProto MUST contain the name field, and *only one* of the
|
|
119
123
|
// following content fields, effectively enforcing a C/C++ union equivalent.
|
|
120
124
|
message AttributeProto {
|
|
125
|
+
reserved 12, 16 to 19;
|
|
126
|
+
reserved "v";
|
|
121
127
|
|
|
122
128
|
// Note: this enum is structurally identical to the OpSchema::AttrType
|
|
123
129
|
// enum defined in schema.h. If you rev one, you likely need to rev the other.
|
|
@@ -190,6 +196,8 @@ message ValueInfoProto {
|
|
|
190
196
|
optional TypeProto type = 2;
|
|
191
197
|
// A human-readable documentation for this value. Markdown is allowed.
|
|
192
198
|
optional string doc_string = 3;
|
|
199
|
+
// Named metadata values; keys should be distinct.
|
|
200
|
+
repeated StringStringEntryProto metadata_props = 4;
|
|
193
201
|
}
|
|
194
202
|
|
|
195
203
|
// Nodes
|
|
@@ -211,12 +219,17 @@ message NodeProto {
|
|
|
211
219
|
optional string op_type = 4; // namespace Operator
|
|
212
220
|
// The domain of the OperatorSet that specifies the operator named by op_type.
|
|
213
221
|
optional string domain = 7; // namespace Domain
|
|
222
|
+
// Overload identifier, used only to map this to a model-local function.
|
|
223
|
+
optional string overload = 8;
|
|
214
224
|
|
|
215
225
|
// Additional named attributes.
|
|
216
226
|
repeated AttributeProto attribute = 5;
|
|
217
227
|
|
|
218
228
|
// A human-readable documentation for this node. Markdown is allowed.
|
|
219
229
|
optional string doc_string = 6;
|
|
230
|
+
|
|
231
|
+
// Named metadata values; keys should be distinct.
|
|
232
|
+
repeated StringStringEntryProto metadata_props = 9;
|
|
220
233
|
}
|
|
221
234
|
|
|
222
235
|
// Training information
|
|
@@ -401,7 +414,7 @@ message ModelProto {
|
|
|
401
414
|
|
|
402
415
|
// A list of function protos local to the model.
|
|
403
416
|
//
|
|
404
|
-
//
|
|
417
|
+
// The (domain, name, overload) tuple must be unique across the function protos in this list.
|
|
405
418
|
// In case of any conflicts the behavior (whether the model local functions are given higher priority,
|
|
406
419
|
// or standard operator sets are given higher priotity or this is treated as error) is defined by
|
|
407
420
|
// the runtimes.
|
|
@@ -475,6 +488,9 @@ message GraphProto {
|
|
|
475
488
|
// which means, tensor 'a_scale' and tensor 'a_zero_point' are scale and zero point of tensor 'a' in the model.
|
|
476
489
|
repeated TensorAnnotation quantization_annotation = 14;
|
|
477
490
|
|
|
491
|
+
// Named metadata values; keys should be distinct.
|
|
492
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
493
|
+
|
|
478
494
|
reserved 3, 4, 6 to 9;
|
|
479
495
|
reserved "ir_version", "producer_version", "producer_tag", "domain";
|
|
480
496
|
}
|
|
@@ -517,10 +533,14 @@ message TensorProto {
|
|
|
517
533
|
// Operators supported FP8 are Cast, CastLike, QuantizeLinear, DequantizeLinear.
|
|
518
534
|
// The computation usually happens inside a block quantize / dequantize
|
|
519
535
|
// fused by the runtime.
|
|
520
|
-
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
521
|
-
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
536
|
+
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
537
|
+
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
522
538
|
FLOAT8E5M2 = 19; // follows IEEE 754, supports nan, inf, mostly used for gradients
|
|
523
|
-
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, inf, mostly used for gradients, no negative zero
|
|
539
|
+
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, not inf, mostly used for gradients, no negative zero
|
|
540
|
+
|
|
541
|
+
// 4-bit data-types
|
|
542
|
+
UINT4 = 21; // Unsigned integer in range [0, 15]
|
|
543
|
+
INT4 = 22; // Signed integer in range [-8, 7], using two's-complement representation
|
|
524
544
|
|
|
525
545
|
// Future extensions go here.
|
|
526
546
|
}
|
|
@@ -555,11 +575,13 @@ message TensorProto {
|
|
|
555
575
|
// When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
|
|
556
576
|
repeated float float_data = 4 [packed = true];
|
|
557
577
|
|
|
558
|
-
// For int32, uint8, int8, uint16, int16, bool, float8
|
|
578
|
+
// For int32, uint8, int8, uint16, int16, uint4, int4, bool, float8 and float16 values
|
|
559
579
|
// float16 and float8 values must be bit-wise converted to an uint16_t prior
|
|
560
580
|
// to writing to the buffer.
|
|
581
|
+
// uint4 and int4 values must be packed to 4bitx2 prior to writing to the buffer, the first element is stored in
|
|
582
|
+
// the 4 LSB and the second element is stored in the 4 MSB.
|
|
561
583
|
// When this field is present, the data_type field MUST be
|
|
562
|
-
// INT32, INT16, INT8, UINT16, UINT8, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
584
|
+
// INT32, INT16, INT8, INT4, UINT16, UINT8, UINT4, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
563
585
|
repeated int32 int32_data = 5 [packed = true];
|
|
564
586
|
|
|
565
587
|
// For strings.
|
|
@@ -589,6 +611,7 @@ message TensorProto {
|
|
|
589
611
|
// Complex64 elements must be written as two consecutive FLOAT values, real component first.
|
|
590
612
|
// Complex128 elements must be written as two consecutive DOUBLE values, real component first.
|
|
591
613
|
// Boolean type MUST be written one byte per tensor element (00000001 for true, 00000000 for false).
|
|
614
|
+
// uint4 and int4 values must be packed to 4bitx2, the first element is stored in the 4 LSB and the second element is stored in the 4 MSB.
|
|
592
615
|
//
|
|
593
616
|
// Note: the advantage of specific field rather than the raw_data field is
|
|
594
617
|
// that in some cases (e.g. int data), protobuf does a better packing via
|
|
@@ -631,6 +654,9 @@ message TensorProto {
|
|
|
631
654
|
// When this field is present, the data_type field MUST be
|
|
632
655
|
// UINT32 or UINT64
|
|
633
656
|
repeated uint64 uint64_data = 11 [packed = true];
|
|
657
|
+
|
|
658
|
+
// Named metadata values; keys should be distinct.
|
|
659
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
634
660
|
}
|
|
635
661
|
|
|
636
662
|
// A serialized sparse-tensor value
|
|
@@ -791,9 +817,8 @@ enum OperatorStatus {
|
|
|
791
817
|
}
|
|
792
818
|
|
|
793
819
|
message FunctionProto {
|
|
794
|
-
// The name of the function, similar
|
|
795
|
-
//
|
|
796
|
-
// the FunctionProto.
|
|
820
|
+
// The name of the function, similar to op_type in NodeProto.
|
|
821
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
797
822
|
optional string name = 1;
|
|
798
823
|
|
|
799
824
|
// Deprecated since IR Version 8
|
|
@@ -840,11 +865,23 @@ message FunctionProto {
|
|
|
840
865
|
|
|
841
866
|
repeated OperatorSetIdProto opset_import = 9;
|
|
842
867
|
|
|
843
|
-
// The domain which this function belongs to.
|
|
844
|
-
// the
|
|
868
|
+
// The domain which this function belongs to.
|
|
869
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
845
870
|
optional string domain = 10;
|
|
846
|
-
}
|
|
847
871
|
|
|
872
|
+
// The overload identifier of the function.
|
|
873
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
874
|
+
optional string overload = 13;
|
|
875
|
+
|
|
876
|
+
// Information for the values in the function. The ValueInfoProto.name's
|
|
877
|
+
// must be distinct and refer to names in the function (including inputs,
|
|
878
|
+
// outputs, and intermediate values). It is optional for a value to appear
|
|
879
|
+
// in value_info list.
|
|
880
|
+
repeated ValueInfoProto value_info = 12;
|
|
881
|
+
|
|
882
|
+
// Named metadata values; keys should be distinct.
|
|
883
|
+
repeated StringStringEntryProto metadata_props = 14;
|
|
884
|
+
}
|
|
848
885
|
|
|
849
886
|
// For using protobuf-lite
|
|
850
887
|
option optimize_for = LITE_RUNTIME;
|
onnx/onnx.in.proto
CHANGED
|
@@ -105,7 +105,11 @@ enum Version {
|
|
|
105
105
|
// IR VERSION 9 published on May 5, 2023
|
|
106
106
|
// Added AttributeProto to FunctionProto so that default attribute values can be set.
|
|
107
107
|
// Added FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ.
|
|
108
|
-
|
|
108
|
+
IR_VERSION_2023_5_5 = 0x0000000000000009;
|
|
109
|
+
|
|
110
|
+
// IR VERSION 10 published on TBD
|
|
111
|
+
// Added UINT4, INT4.
|
|
112
|
+
IR_VERSION = 0x000000000000000A;
|
|
109
113
|
}
|
|
110
114
|
|
|
111
115
|
// Attributes
|
|
@@ -115,6 +119,8 @@ enum Version {
|
|
|
115
119
|
// An AttributeProto MUST contain the name field, and *only one* of the
|
|
116
120
|
// following content fields, effectively enforcing a C/C++ union equivalent.
|
|
117
121
|
message AttributeProto {
|
|
122
|
+
reserved 12, 16 to 19;
|
|
123
|
+
reserved "v";
|
|
118
124
|
|
|
119
125
|
// Note: this enum is structurally identical to the OpSchema::AttrType
|
|
120
126
|
// enum defined in schema.h. If you rev one, you likely need to rev the other.
|
|
@@ -187,6 +193,8 @@ message ValueInfoProto {
|
|
|
187
193
|
optional TypeProto type = 2;
|
|
188
194
|
// A human-readable documentation for this value. Markdown is allowed.
|
|
189
195
|
optional string doc_string = 3;
|
|
196
|
+
// Named metadata values; keys should be distinct.
|
|
197
|
+
repeated StringStringEntryProto metadata_props = 4;
|
|
190
198
|
}
|
|
191
199
|
|
|
192
200
|
// Nodes
|
|
@@ -208,12 +216,17 @@ message NodeProto {
|
|
|
208
216
|
optional string op_type = 4; // namespace Operator
|
|
209
217
|
// The domain of the OperatorSet that specifies the operator named by op_type.
|
|
210
218
|
optional string domain = 7; // namespace Domain
|
|
219
|
+
// Overload identifier, used only to map this to a model-local function.
|
|
220
|
+
optional string overload = 8;
|
|
211
221
|
|
|
212
222
|
// Additional named attributes.
|
|
213
223
|
repeated AttributeProto attribute = 5;
|
|
214
224
|
|
|
215
225
|
// A human-readable documentation for this node. Markdown is allowed.
|
|
216
226
|
optional string doc_string = 6;
|
|
227
|
+
|
|
228
|
+
// Named metadata values; keys should be distinct.
|
|
229
|
+
repeated StringStringEntryProto metadata_props = 9;
|
|
217
230
|
}
|
|
218
231
|
|
|
219
232
|
// Training information
|
|
@@ -398,7 +411,7 @@ message ModelProto {
|
|
|
398
411
|
|
|
399
412
|
// A list of function protos local to the model.
|
|
400
413
|
//
|
|
401
|
-
//
|
|
414
|
+
// The (domain, name, overload) tuple must be unique across the function protos in this list.
|
|
402
415
|
// In case of any conflicts the behavior (whether the model local functions are given higher priority,
|
|
403
416
|
// or standard operator sets are given higher priotity or this is treated as error) is defined by
|
|
404
417
|
// the runtimes.
|
|
@@ -472,6 +485,9 @@ message GraphProto {
|
|
|
472
485
|
// which means, tensor 'a_scale' and tensor 'a_zero_point' are scale and zero point of tensor 'a' in the model.
|
|
473
486
|
repeated TensorAnnotation quantization_annotation = 14;
|
|
474
487
|
|
|
488
|
+
// Named metadata values; keys should be distinct.
|
|
489
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
490
|
+
|
|
475
491
|
reserved 3, 4, 6 to 9;
|
|
476
492
|
reserved "ir_version", "producer_version", "producer_tag", "domain";
|
|
477
493
|
}
|
|
@@ -514,10 +530,14 @@ message TensorProto {
|
|
|
514
530
|
// Operators supported FP8 are Cast, CastLike, QuantizeLinear, DequantizeLinear.
|
|
515
531
|
// The computation usually happens inside a block quantize / dequantize
|
|
516
532
|
// fused by the runtime.
|
|
517
|
-
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
518
|
-
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
533
|
+
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
534
|
+
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
519
535
|
FLOAT8E5M2 = 19; // follows IEEE 754, supports nan, inf, mostly used for gradients
|
|
520
|
-
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, inf, mostly used for gradients, no negative zero
|
|
536
|
+
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, not inf, mostly used for gradients, no negative zero
|
|
537
|
+
|
|
538
|
+
// 4-bit data-types
|
|
539
|
+
UINT4 = 21; // Unsigned integer in range [0, 15]
|
|
540
|
+
INT4 = 22; // Signed integer in range [-8, 7], using two's-complement representation
|
|
521
541
|
|
|
522
542
|
// Future extensions go here.
|
|
523
543
|
}
|
|
@@ -552,11 +572,13 @@ message TensorProto {
|
|
|
552
572
|
// When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
|
|
553
573
|
repeated float float_data = 4 [packed = true];
|
|
554
574
|
|
|
555
|
-
// For int32, uint8, int8, uint16, int16, bool, float8
|
|
575
|
+
// For int32, uint8, int8, uint16, int16, uint4, int4, bool, float8 and float16 values
|
|
556
576
|
// float16 and float8 values must be bit-wise converted to an uint16_t prior
|
|
557
577
|
// to writing to the buffer.
|
|
578
|
+
// uint4 and int4 values must be packed to 4bitx2 prior to writing to the buffer, the first element is stored in
|
|
579
|
+
// the 4 LSB and the second element is stored in the 4 MSB.
|
|
558
580
|
// When this field is present, the data_type field MUST be
|
|
559
|
-
// INT32, INT16, INT8, UINT16, UINT8, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
581
|
+
// INT32, INT16, INT8, INT4, UINT16, UINT8, UINT4, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
560
582
|
repeated int32 int32_data = 5 [packed = true];
|
|
561
583
|
|
|
562
584
|
// For strings.
|
|
@@ -586,6 +608,7 @@ message TensorProto {
|
|
|
586
608
|
// Complex64 elements must be written as two consecutive FLOAT values, real component first.
|
|
587
609
|
// Complex128 elements must be written as two consecutive DOUBLE values, real component first.
|
|
588
610
|
// Boolean type MUST be written one byte per tensor element (00000001 for true, 00000000 for false).
|
|
611
|
+
// uint4 and int4 values must be packed to 4bitx2, the first element is stored in the 4 LSB and the second element is stored in the 4 MSB.
|
|
589
612
|
//
|
|
590
613
|
// Note: the advantage of specific field rather than the raw_data field is
|
|
591
614
|
// that in some cases (e.g. int data), protobuf does a better packing via
|
|
@@ -628,6 +651,9 @@ message TensorProto {
|
|
|
628
651
|
// When this field is present, the data_type field MUST be
|
|
629
652
|
// UINT32 or UINT64
|
|
630
653
|
repeated uint64 uint64_data = 11 [packed = true];
|
|
654
|
+
|
|
655
|
+
// Named metadata values; keys should be distinct.
|
|
656
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
631
657
|
}
|
|
632
658
|
|
|
633
659
|
// A serialized sparse-tensor value
|
|
@@ -792,9 +818,8 @@ enum OperatorStatus {
|
|
|
792
818
|
}
|
|
793
819
|
|
|
794
820
|
message FunctionProto {
|
|
795
|
-
// The name of the function, similar
|
|
796
|
-
//
|
|
797
|
-
// the FunctionProto.
|
|
821
|
+
// The name of the function, similar to op_type in NodeProto.
|
|
822
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
798
823
|
optional string name = 1;
|
|
799
824
|
|
|
800
825
|
// Deprecated since IR Version 8
|
|
@@ -841,8 +866,20 @@ message FunctionProto {
|
|
|
841
866
|
|
|
842
867
|
repeated OperatorSetIdProto opset_import = 9;
|
|
843
868
|
|
|
844
|
-
// The domain which this function belongs to.
|
|
845
|
-
// the
|
|
869
|
+
// The domain which this function belongs to.
|
|
870
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
846
871
|
optional string domain = 10;
|
|
847
|
-
}
|
|
848
872
|
|
|
873
|
+
// The overload identifier of the function.
|
|
874
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
875
|
+
optional string overload = 13;
|
|
876
|
+
|
|
877
|
+
// Information for the values in the function. The ValueInfoProto.name's
|
|
878
|
+
// must be distinct and refer to names in the function (including inputs,
|
|
879
|
+
// outputs, and intermediate values). It is optional for a value to appear
|
|
880
|
+
// in value_info list.
|
|
881
|
+
repeated ValueInfoProto value_info = 12;
|
|
882
|
+
|
|
883
|
+
// Named metadata values; keys should be distinct.
|
|
884
|
+
repeated StringStringEntryProto metadata_props = 14;
|
|
885
|
+
}
|
onnx/onnx.proto
CHANGED
|
@@ -106,7 +106,11 @@ enum Version {
|
|
|
106
106
|
// IR VERSION 9 published on May 5, 2023
|
|
107
107
|
// Added AttributeProto to FunctionProto so that default attribute values can be set.
|
|
108
108
|
// Added FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ.
|
|
109
|
-
|
|
109
|
+
IR_VERSION_2023_5_5 = 0x0000000000000009;
|
|
110
|
+
|
|
111
|
+
// IR VERSION 10 published on TBD
|
|
112
|
+
// Added UINT4, INT4.
|
|
113
|
+
IR_VERSION = 0x000000000000000A;
|
|
110
114
|
}
|
|
111
115
|
|
|
112
116
|
// Attributes
|
|
@@ -116,6 +120,8 @@ enum Version {
|
|
|
116
120
|
// An AttributeProto MUST contain the name field, and *only one* of the
|
|
117
121
|
// following content fields, effectively enforcing a C/C++ union equivalent.
|
|
118
122
|
message AttributeProto {
|
|
123
|
+
reserved 12, 16 to 19;
|
|
124
|
+
reserved "v";
|
|
119
125
|
|
|
120
126
|
// Note: this enum is structurally identical to the OpSchema::AttrType
|
|
121
127
|
// enum defined in schema.h. If you rev one, you likely need to rev the other.
|
|
@@ -188,6 +194,8 @@ message ValueInfoProto {
|
|
|
188
194
|
optional TypeProto type = 2;
|
|
189
195
|
// A human-readable documentation for this value. Markdown is allowed.
|
|
190
196
|
optional string doc_string = 3;
|
|
197
|
+
// Named metadata values; keys should be distinct.
|
|
198
|
+
repeated StringStringEntryProto metadata_props = 4;
|
|
191
199
|
}
|
|
192
200
|
|
|
193
201
|
// Nodes
|
|
@@ -209,12 +217,17 @@ message NodeProto {
|
|
|
209
217
|
optional string op_type = 4; // namespace Operator
|
|
210
218
|
// The domain of the OperatorSet that specifies the operator named by op_type.
|
|
211
219
|
optional string domain = 7; // namespace Domain
|
|
220
|
+
// Overload identifier, used only to map this to a model-local function.
|
|
221
|
+
optional string overload = 8;
|
|
212
222
|
|
|
213
223
|
// Additional named attributes.
|
|
214
224
|
repeated AttributeProto attribute = 5;
|
|
215
225
|
|
|
216
226
|
// A human-readable documentation for this node. Markdown is allowed.
|
|
217
227
|
optional string doc_string = 6;
|
|
228
|
+
|
|
229
|
+
// Named metadata values; keys should be distinct.
|
|
230
|
+
repeated StringStringEntryProto metadata_props = 9;
|
|
218
231
|
}
|
|
219
232
|
|
|
220
233
|
// Training information
|
|
@@ -399,7 +412,7 @@ message ModelProto {
|
|
|
399
412
|
|
|
400
413
|
// A list of function protos local to the model.
|
|
401
414
|
//
|
|
402
|
-
//
|
|
415
|
+
// The (domain, name, overload) tuple must be unique across the function protos in this list.
|
|
403
416
|
// In case of any conflicts the behavior (whether the model local functions are given higher priority,
|
|
404
417
|
// or standard operator sets are given higher priotity or this is treated as error) is defined by
|
|
405
418
|
// the runtimes.
|
|
@@ -473,6 +486,9 @@ message GraphProto {
|
|
|
473
486
|
// which means, tensor 'a_scale' and tensor 'a_zero_point' are scale and zero point of tensor 'a' in the model.
|
|
474
487
|
repeated TensorAnnotation quantization_annotation = 14;
|
|
475
488
|
|
|
489
|
+
// Named metadata values; keys should be distinct.
|
|
490
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
491
|
+
|
|
476
492
|
reserved 3, 4, 6 to 9;
|
|
477
493
|
reserved "ir_version", "producer_version", "producer_tag", "domain";
|
|
478
494
|
}
|
|
@@ -515,10 +531,14 @@ message TensorProto {
|
|
|
515
531
|
// Operators supported FP8 are Cast, CastLike, QuantizeLinear, DequantizeLinear.
|
|
516
532
|
// The computation usually happens inside a block quantize / dequantize
|
|
517
533
|
// fused by the runtime.
|
|
518
|
-
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
519
|
-
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
534
|
+
FLOAT8E4M3FN = 17; // float 8, mostly used for coefficients, supports nan, not inf
|
|
535
|
+
FLOAT8E4M3FNUZ = 18; // float 8, mostly used for coefficients, supports nan, not inf, no negative zero
|
|
520
536
|
FLOAT8E5M2 = 19; // follows IEEE 754, supports nan, inf, mostly used for gradients
|
|
521
|
-
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, inf, mostly used for gradients, no negative zero
|
|
537
|
+
FLOAT8E5M2FNUZ = 20; // follows IEEE 754, supports nan, not inf, mostly used for gradients, no negative zero
|
|
538
|
+
|
|
539
|
+
// 4-bit data-types
|
|
540
|
+
UINT4 = 21; // Unsigned integer in range [0, 15]
|
|
541
|
+
INT4 = 22; // Signed integer in range [-8, 7], using two's-complement representation
|
|
522
542
|
|
|
523
543
|
// Future extensions go here.
|
|
524
544
|
}
|
|
@@ -553,11 +573,13 @@ message TensorProto {
|
|
|
553
573
|
// When this field is present, the data_type field MUST be FLOAT or COMPLEX64.
|
|
554
574
|
repeated float float_data = 4 [packed = true];
|
|
555
575
|
|
|
556
|
-
// For int32, uint8, int8, uint16, int16, bool, float8
|
|
576
|
+
// For int32, uint8, int8, uint16, int16, uint4, int4, bool, float8 and float16 values
|
|
557
577
|
// float16 and float8 values must be bit-wise converted to an uint16_t prior
|
|
558
578
|
// to writing to the buffer.
|
|
579
|
+
// uint4 and int4 values must be packed to 4bitx2 prior to writing to the buffer, the first element is stored in
|
|
580
|
+
// the 4 LSB and the second element is stored in the 4 MSB.
|
|
559
581
|
// When this field is present, the data_type field MUST be
|
|
560
|
-
// INT32, INT16, INT8, UINT16, UINT8, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
582
|
+
// INT32, INT16, INT8, INT4, UINT16, UINT8, UINT4, BOOL, FLOAT16, BFLOAT16, FLOAT8E4M3FN, FLOAT8E4M3FNUZ, FLOAT8E5M2, FLOAT8E5M2FNUZ
|
|
561
583
|
repeated int32 int32_data = 5 [packed = true];
|
|
562
584
|
|
|
563
585
|
// For strings.
|
|
@@ -587,6 +609,7 @@ message TensorProto {
|
|
|
587
609
|
// Complex64 elements must be written as two consecutive FLOAT values, real component first.
|
|
588
610
|
// Complex128 elements must be written as two consecutive DOUBLE values, real component first.
|
|
589
611
|
// Boolean type MUST be written one byte per tensor element (00000001 for true, 00000000 for false).
|
|
612
|
+
// uint4 and int4 values must be packed to 4bitx2, the first element is stored in the 4 LSB and the second element is stored in the 4 MSB.
|
|
590
613
|
//
|
|
591
614
|
// Note: the advantage of specific field rather than the raw_data field is
|
|
592
615
|
// that in some cases (e.g. int data), protobuf does a better packing via
|
|
@@ -629,6 +652,9 @@ message TensorProto {
|
|
|
629
652
|
// When this field is present, the data_type field MUST be
|
|
630
653
|
// UINT32 or UINT64
|
|
631
654
|
repeated uint64 uint64_data = 11 [packed = true];
|
|
655
|
+
|
|
656
|
+
// Named metadata values; keys should be distinct.
|
|
657
|
+
repeated StringStringEntryProto metadata_props = 16;
|
|
632
658
|
}
|
|
633
659
|
|
|
634
660
|
// A serialized sparse-tensor value
|
|
@@ -775,9 +801,8 @@ enum OperatorStatus {
|
|
|
775
801
|
}
|
|
776
802
|
|
|
777
803
|
message FunctionProto {
|
|
778
|
-
// The name of the function, similar
|
|
779
|
-
//
|
|
780
|
-
// the FunctionProto.
|
|
804
|
+
// The name of the function, similar to op_type in NodeProto.
|
|
805
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
781
806
|
optional string name = 1;
|
|
782
807
|
|
|
783
808
|
// Deprecated since IR Version 8
|
|
@@ -824,11 +849,23 @@ message FunctionProto {
|
|
|
824
849
|
|
|
825
850
|
repeated OperatorSetIdProto opset_import = 9;
|
|
826
851
|
|
|
827
|
-
// The domain which this function belongs to.
|
|
828
|
-
// the
|
|
852
|
+
// The domain which this function belongs to.
|
|
853
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
829
854
|
optional string domain = 10;
|
|
830
|
-
}
|
|
831
855
|
|
|
856
|
+
// The overload identifier of the function.
|
|
857
|
+
// This is part of the unique-id (domain, name, overload) of FunctionProtos in a model.
|
|
858
|
+
optional string overload = 13;
|
|
859
|
+
|
|
860
|
+
// Information for the values in the function. The ValueInfoProto.name's
|
|
861
|
+
// must be distinct and refer to names in the function (including inputs,
|
|
862
|
+
// outputs, and intermediate values). It is optional for a value to appear
|
|
863
|
+
// in value_info list.
|
|
864
|
+
repeated ValueInfoProto value_info = 12;
|
|
865
|
+
|
|
866
|
+
// Named metadata values; keys should be distinct.
|
|
867
|
+
repeated StringStringEntryProto metadata_props = 14;
|
|
868
|
+
}
|
|
832
869
|
|
|
833
870
|
// For using protobuf-lite
|
|
834
871
|
option optimize_for = LITE_RUNTIME;
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
class CheckerContext:
|
|
2
|
+
ir_version: int = ...
|
|
3
|
+
opset_imports: dict[str, int] = ...
|
|
4
|
+
|
|
5
|
+
|
|
6
|
+
class LexicalScopeContext:
|
|
7
|
+
ir_version: int = ...
|
|
8
|
+
opset_imports: dict[str, int] = ...
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
class ValidationError(Exception): ...
|
|
12
|
+
|
|
13
|
+
def check_value_info(bytes: bytes, checker_context: CheckerContext) -> None: ... # noqa: A002
|
|
14
|
+
def check_tensor(bytes: bytes, checker_context: CheckerContext) -> None: ... # noqa: A002
|
|
15
|
+
def check_sparse_tensor(bytes: bytes, checker_context: CheckerContext) -> None: ... # noqa: A002
|
|
16
|
+
def check_attribute(bytes: bytes, checker_context: CheckerContext, lexical_scope_context: LexicalScopeContext) -> None: ... # noqa: A002
|
|
17
|
+
def check_node(bytes: bytes, checker_context: CheckerContext, lexical_scope_context: LexicalScopeContext) -> None: ... # noqa: A002
|
|
18
|
+
def check_function(bytes: bytes, checker_context: CheckerContext, lexical_scope_context: LexicalScopeContext) -> None: ... # noqa: A002
|
|
19
|
+
def check_graph(bytes: bytes, checker_context: CheckerContext, lexical_scope_context: LexicalScopeContext) -> None: ... # noqa: A002
|
|
20
|
+
def check_model(bytes: bytes, full_check: bool, skip_opset_compatibility_check: bool, check_custom_domain: bool) -> None: ... # noqa: A002
|
|
21
|
+
def check_model_path(path: str, full_check: bool, skip_opset_compatibility_check: bool, check_custom_domain: bool) -> None: ...
|