o Xi$8@sdZddlmZddlZddlmZddlZddlZGdddej Z Gdddej Z ie j d e j d e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jd e jde j de j!de j"d e j#de j$diZ%ie&de je&de je&de je&de je&de j e&de je&de je&de je&de je&de je&de j'e&de je&de je&de je&de j e&ej(e je&ej)e je&ej*e je&ej+e je&ej,e je&ej-e j"e&ej.e j e&ej/e je&ej0e j!e&ej1e j#e&ej2e j$i Z3dd e34DZ5ie j6d!e jd"e jd#e j d$e jd%e jd&e jd'e jd(e jd)e j"d*e j!d+e jd,e jd-e j#d.e j d/e jd0e jd1e jd2e jd3e jd4e j$d5e jd6e j d7e jd8e jd9e jd:e j'd;i Z7dZ-dUd?d@Z.dVdBdCZ/dVdDdEZ0dVdFdGZ1dVdHdIZ2dUdJdKZ3dUdLdMZ4dNS)WDataTypezIEnum for the data types of ONNX tensors, defined in ``onnx.TensorProto``.rrrrrr r r r r rrrrrdtypenp.dtypercCs|tvr |t|St|tjst|tjrtjSt|drp|jdkr(tj S|jdkr0tj S|jdkr8tj S|jdkr@tj S|jdkrHtj S|jdkrPtjS|jdkrXtjS|jd kr`tjS|jd krhtjS|jd krptjStd |) zReturns the ONNX data type for the numpy dtype. Raises: TypeError: If the data type is not supported by ONNX. names)bfloat16)e4m3fn)e4m3fnuz)e5m2)e5m2fnuz)uint4)int4) float4e2m1)int2)uint2zUnsupported numpy data type: )_NP_TYPE_TO_DATA_TYPEnp issubdtypestr_bytes_r7r+hasattrrFBFLOAT16 FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZUINT4INT4 FLOAT4E2M1INT2UINT2 TypeError)clsrDrrr from_numpyIs4            zDataType.from_numpy short_namercCs"|tvr td||t|S)zReturns the ONNX data type for the short name. Raises: TypeError: If the short name is not available for the data type. zUnknown short name: )_SHORT_NAME_TO_DATA_TYPEra)rbrdrrrfrom_short_nameps zDataType.from_short_namefloatcCs |jdS)z+Returns the size of the data type in bytes.r )bitwidthrrrritemsize{s zDataType.itemsizeintcC|tvr td|t|S)zReturns the bit width of the data type. .. versionadded:: 0.1.2 Raises: TypeError: If the data type is not supported. z+Bitwidth not available for ONNX data type: ) _BITWIDTH_MAPrarrrrrhs zDataType.bitwidthcC&|r t|jStd|)zReturns the bit width of the exponent for floating-point types. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not supported. z+Exponent not available for ONNX data type: )is_floating_point ml_dtypesfinfonumpynexprarrrrexponent_bitwidth zDataType.exponent_bitwidthcCrm)zReturns the bit width of the mantissa for floating-point types. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not supported. z+Mantissa not available for ONNX data type: )rnrorprqnmantrarrrrmantissa_bitwidthrtzDataType.mantissa_bitwidthint | np.floating[Any]cC2|rdS|rt|jStd|)aReturns the difference between 1.0 and the next smallest representable float larger than 1.0 for the ONNX data type. Returns 1 for integers. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. rz&Eps not available for ONNX data type: ) is_integerrnrorprqepsrarrrrrz  z DataType.epscCrx)zReturns the smallest positive non-zero value for the ONNX data type. Returns 1 for integers. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. rz'Tiny not available for ONNX data type: )ryrnrorprqtinyrarrrrr|r{z DataType.tinycC>|r t|jS|rt|jStd|)zReturns the minimum representable value for the ONNX data type. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. z*Minimum not available for ONNX data type: )ryroiinforqminrnrprarrrrr  z DataType.mincCr})zReturns the maximum representable value for the ONNX data type. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. z*Maximum not available for ONNX data type: )ryror~rqmaxrnrprarrrrrrz DataType.maxcCrx)aBReturns the precision for the ONNX dtype if supported. For floats returns the approximate number of decimal digits to which this kind of float is precise. Returns 0 for integers. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. rz,Precision not available for ONNX data type: )ryrnrorprq precisionrarrrrr  zDataType.precisioncCrx)a/Returns the resolution for the ONNX dtype if supported. Returns the approximate decimal resolution of this type, i.e., 10**-precision. Returns 1 for integers. .. versionadded:: 0.1.8 Raises: TypeError: If the data type is not a numeric data type. rz-Resolution not available for ONNX data type: )ryrnrorprq resolutionrarrrrrrzDataType.resolutioncCrk)zReturns the numpy dtype for the ONNX data type. Raises: TypeError: If the data type is not supported by numpy. z'Numpy does not support ONNX data type: )_DATA_TYPE_TO_NP_TYPErarrrrrqszDataType.numpycCrk)aReturns the short name of the data type. The short name is a string that is used to represent the data type in a more compact form. For example, the short name for `DataType.FLOAT` is "f32". To get the corresponding data type back, call ``from_short_name`` on a string. Naming reference: https://github.com/pytorch/pytorch/blob/4bead7b85ea4160243c74109e0ce9bb80686d016/torch/utils/_dtype_abbrs.py Raises: TypeError: If the short name is not available for the data type. z-Short name not available for ONNX data type: )_DATA_TYPE_TO_SHORT_NAMErarrrrrd#s zDataType.short_nameboolc Cs0|tjtjtjtjtjtjtjtjtj tj h vS)z7Returns True if the data type is a floating point type.) r7r)FLOAT16DOUBLErWrXrYrZr[r^ FLOAT8E8M0rrrrrn3szDataType.is_floating_pointc Cs8|tjtjtjtjtjtjtjtjtj tj tj tj h vS)zVReturns True if the data type is an integer. .. versionadded:: 0.1.4 ) r7UINT8INT8UINT16INT16INT32INT64UINT32UINT64r\r]r_r`rrrrryBszDataType.is_integercCsP|tjtjtjtjtjtjtjtjtj tj tj tj tj tjtjtjtjtjhvS)zYReturns True if the data type is a signed type. .. versionadded:: 0.1.4 )r7r)rrrrrr COMPLEX64 COMPLEX128rWrXrYrZr[r]r^rr_rrrr is_signedVs(zDataType.is_signedcCs |tjkS)zYReturns True if the data type is a string type. .. versionadded:: 0.1.8 )r7r+rrrr is_stringps zDataType.is_stringcCrrrrrrrrwrzDataType.__repr__cCrrr rrrrr!zr"zDataType.__str__N)rDrErr7)rdrrr7)rrg)rrj)rrw)rrEr#)rr)5r$r%r&r'r(r)rrrrrrr+BOOLrrrrrrrWrXrYrZr[r\r]r^rr`r_ classmethodrcrfpropertyrirhrsrvrzr|rrrrrqrdrnryrrrr!rrrrr7(sz &                 r7 r r9@rrr complex128 complex64float16float32float64int16int32int64int8objectuint16uint32uint64uint8cCi|]\}}||qSrr.0kvrrr r undefinedbf16f64f32f16f8e4m3fnf8e5m2 f8e4m3fnuz f8e5m2fnuzf8e8m0f4e2m1c64c128i2i4i8i16i32i64b8u2u4u8u16u32u64scCrrrrrrrrr)9r' __future__renumtypingrrorqrRIntEnumrr7r)rrrrrrrrrrrrrrWrXrYrZr[r\r]r^rr_r`rlrDr+rG float8_e4m3fnfloat8_e4m3fnuz float8_e5m2float8_e5m2fnuzfloat8_e8m0fnurMrL float4_e2m1fnrOrPrQitemsrr(rrerrrrs&  X