SinSSSKJr SSKrSSKJrJr SSKrSSKrSSK Jr SSK r SSK J r \(aSSKJr SSSjjrSSjrSSjrSS jrSS jrSSS jjrSS jrSSS jjrSSjrS S!SjjrS"SjrSS#SjjrS$SjrS S%SjjrS&S'SjjrS(Sjrg))) annotationsN) TYPE_CHECKINGAny)helper)Sequencec@[R"U[RS9nUR[R5nUS- S-nUR [R 5nUS:Hn[R"U[RS9nSXv'U)nUS:XaUS-S:R [R5n US-S:R [R5n US-S:R [R5n US:R [R5n U S :HU S :HU S :H-U S :H--n [R"U5nS XU-'U(aUS :HU -U-nSX'X^- nO]US :XaBUS -S:nUU-n U(a US :HU -nS X'X]R [R 5- nOUS:XaO[SU35eUR [R5nXXXx'UR[R5$)aRConvert float32 NumPy array to float8e8m0 representation. If the input is not a float32 array, it will be cast to one first. Args: x: Input array to convert. saturate: Whether to saturate at max/min float8e8m0 value. round_mode: "nearest", "up", or "down". Returns: np.ndarray: Array of ml_dtypes.float8_e8m0fnu values. dtypenearesti@ri iupiOFdownzUnsupported rounding mode: ) npasarrayfloat32viewuint32astypeuint16 zeros_likeuint8 ValueError ml_dtypesfloat8_e8m0fnu)xsaturate round_modex_f32f_bitsexponent special_maskoutput normal_maskgrslsbround_up incrementmax_mask has_fractions P/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/numpy_helper.py to_float8e8m0r0s JJq +E ZZ "F" $H H t#L ]]8288 4FF -KYx1 $ , ,RXX 6x1 $ , ,RXX 6x1 $ , ,RXX 6!|##BHH-FQ162cQh?@MM(+ ,- [()  D(H4{BH"#I  t )Q. +-  D(H4H!&H OOBII.. v  6zlCDDrxx(H"/F ;;y// 00c[R"URS-/URS9nU[R"S5-nU[R"S5-nU[R"S5-nX2SSS2'XBSSS2'UR[R "U5S-:XaUSS nUR US S 9 U$) zConvert a packed uint4 array to unpacked uint4 array represented as uint8. Args: data: A numpy array. dims: The dimensions are used to reshape the unpacked buffer. Returns: A numpy array of int8/uint8 reshaped to dims. r rNrFrefcheck)remptysizer rprodresize)datadimsresult array_low array_highs r/ _unpack_4bitrC_sXXtyy1}oTZZ 8Frxx~%I&J288A;J14a4L14a4L {{bggdma'' MM$M' Mr1c"UR5R[R5R 5nUR nUS-S:HnU(aUR US-/SS9 US-nUSSS2==S-ss'USSS2USSS2-$) z[Convert a numpy array to flatten, packed int4/uint4. Elements must be in the correct range.r3rFr8r4Nr6rravelrrrcopyr;r=)array array_flatr; odd_sizeds r/ _pack_4bitx2rKxs##BHH-224J ::DqA I4!8*u5$Jqt!t add jA. ..r1cd[R"URS-/URS9nUS-USSS2'US- S-USSS2'US- S-USSS2'US- S-USSS2'UR[R"U5:aUS[R"U5nUR US S 9 U$) zConvert a packed uint2 array to unpacked uint2 array represented as uint8. Args: data: A numpy array. dims: The dimensions are used to reshape the unpacked buffer. Returns: A numpy array of int8/uint8 reshaped to dims. r6r rNr3rFr8)rr:r;r r<r=)r>r?r@s r/ _unpack_2bitrOsXXtyy1}oTZZ 8F$;F14a4LAI%F14a4LAI%F14a4LAI%F14a4L {{RWWT]""''$-( MM$M' Mr1cUR5R[R5R 5nUR nUS-nU(aUR USU- -/SS9 US-nUSSS2==S-ss'USSS2==S-ss'USSS2==S-ss'US SS2USSS2-USSS2-USSS2-$) z[Convert a numpy array to flatten, packed int2/uint2. Elements must be in the correct range.r6Fr8rMrNr3rNrrE)rHrIr;pad_lens r/ _pack_2bitx4rRs##BHH-224J ::DQhG41w;/05A$Jqt!tqt!tqt!t add jA. .ADqD1A AJqtRStDT TTr1cURS5(a [S5eUR[RR :Xa [ S5eURn[R"U5n[R"[R"U55n[R"U5nURnUR[RR:Xa_[X5nUVs/sHoRS5PM n n[R "U 5R#U5R%U5$[R&R)U5(a[R&R+X5 URS5(GavUR,n [.R0S:Xa0[R2"XS9R55R75n U[RR8[RR:[RR<1;a=[R2"U [R>S9n [AX5RCU5$U[RRD[RRF1;a=[R2"U [R>S9n [IX5RCU5$[R2"XS9R%U5$U[RRJ[RRL[RRN[RRP1;a[RR"URT[RVS9RC[RX5R#[RZ5R%U5RCU5$U[RR\[RR^[RR`[RRb[RRd[RRf1;a[RR"URT[RVS9RC[RX5R#[R>5RCU5R%U5$U[RR:[RR8[RR<1;a[RR"URT[RVS9RC[RX5R#[R>5n [AX5RCU5$U[RRD[RRF1;a[RR"URT[RVS9RC[RX5R#[R>5n [IX5RCU5$[X5n U[RRh[RRj4;a0[RR"XS9RCUS9R%U5$[R "XS9R#U5R%U5$s snf)aConverts a tensor def object to a numpy array. This function uses ml_dtypes if the dtype is not a native numpy dtype. Args: tensor: a TensorProto object. base_dir: if external tensor exists, base_dir can help to find the path to it Returns: arr: the converted array. segmentz*Currently not supporting loading segments.z2The element type in the input tensor is UNDEFINED.utf-8raw_databigr )6HasFieldr data_typeonnx TensorProto UNDEFINED TypeErrorrtensor_dtype_to_np_dtype$tensor_dtype_to_storage_tensor_dtypetensor_dtype_to_fieldr?STRINGgetattrdecoderrrreshapeexternal_data_helperuses_external_dataload_external_data_for_tensorrVsys byteorder frombufferbyteswaptobytesINT4UINT4 FLOAT4E2M1rrCrUINT2INT2rOBFLOAT16FLOAT16INT16UINT16rH int32_dataint32rr FLOAT8E4M3FNFLOAT8E4M3FNUZ FLOAT8E5M2FLOAT8E5M2FNUZ FLOAT8E8M0BOOL COMPLEX64 COMPLEX128) tensorbase_dir tensor_dtypenp_dtypestorage_np_dtype storage_fieldr? utf8_stringsr)ssrVr>s r/to_arrayrs#y!!EFF 4++555LMM##L..|M ;;D 4++222v5 )5 6Ahhw 6zz"~$$X.66t<<   33F;; !!??Q z""?? ==E !}}X>GGIQQSH     ! !    " "    ' '  ==:D+00: :     " "    ! !  ==:D+00: :}}X6>>tDD !!       HHV&&bhh 7 T"))_ VBII  WT] T(^   %% '' ## '' ##   HHV&&bhh 7 T"))_ VBHH  T(^ WT]     ## HHV&&bhh 7 < rW=<)r rirhr newbyteorderrl)rHs r/tobytes_little_endianrs^ {{# 5;;#8#8C#? U[[55c:; ==?r1c `[R"5nURRUR5 U(aXlUR [:Xd4[R"UR [R5(a[RRUl UR5nUHn[U[5(a,UR R#UR%S55 MD[U[&5(aUR R#U5 Mv[)S[[+U555e U$[,R."UR 5nU[RR0[RR2[RR41;a [7U5nU[RR8[RR:1;a [=U5n[?U5Ul XRl U$)zConverts an array into a TensorProto including Args: array: a numpy array. name: (optional) the name of the tensor. Returns: TensorProto: the converted tensor def. rUzMUnrecognized object in the object array, expect a string, or array of bytes: )!rZr[r?extendshapenamer objectr issubdtypestr_rarYflatten isinstancestr string_dataappendencodebytesNotImplementedErrortypernp_dtype_to_tensor_dtypermrnrorKrprqrRrrV)rHrr flat_arrayer s r/ from_arrayr2s   F KKu{{#   {{f ekk277 C C++22]]_ A!S!!""))!((7*;<Au%%""))!,)cQL   + +EKK 8E    ## U#    U#+E2FO Mr1cxURnU[RR:Xa&URVs/sHn[ U5PM sn$U[RR :Xa&URVs/sHn[ U5PM sn$U[RR:Xa&URVs/sHn[U5PM sn$U[RR:Xa&URVs/sHn[U5PM sn$[S5es snfs snfs snfs snf)zConverts a sequence def to a Python list. Args: sequence: a SequenceProto object. Returns: list: the converted list. z8The element type in the input sequence is not supported.) elem_typerZ SequenceProtoTENSOR tensor_valuesr SPARSE_TENSORsparse_tensor_valuesSEQUENCEsequence_valuesto_listMAP map_valuesto_dictr])sequencervs r/rrms""ID&&---%-%;%;<%; %;<<D&&444%-%B%BC%B %BCCD&&///$,$<$<=$ U$U[RR:Xa0TH(nUR$R['U5/5 M* U$U[RR :Xa0TH(nUR(R[+U5/5 M* U$[S5e)a=Converts a list into a sequence def. Args: lst: a Python list name: (optional) the name of the sequence. dtype: (optional) type of element in the input list, used for specifying sequence values when converting an empty list. Returns: SequenceProto: the converted sequence def. rc3T># UHn[U[TS55v M g7f)rN)rr).0elemlsts r/ from_list..s#!QST*T4A<"@"@Ss%(zqThe element type in the input list is not the same for all elements and therefore is not supported as a sequence.zZThe element type in the input list is not a tensor, sequence, or map and is not supported.)rZrrlenrdictrlistrrrallr]rrrrrr from_listr from_dict) rrr rr first_elemrseqmappings ` r/rrs!!#H    SAV j$ ' '**..I  D ) )**33I**11I&&-- " C1 c!QS!QQQ M  D&&---F  " " ) ):bjj6H+I*J K O d((11 1C  $ $ + +Ys^,< = O d((,, ,G    & & '(:'; < O  5  r1cZ/nUR[RR:Xa[ UR 5nO[ UR 5n[UR5n[U5[U5:wa[SURS5e[[XSS95$)z|Converts a map def to a Python dictionary. Args: map_proto: a MapProto object. Returns: The converted dictionary. z2Length of keys and values for MapProto (map name: z) are not the same.F)strict)key_typerZr[rar string_keyskeysrvaluesr IndexErrorrrzip) map_protokey_list value_lists r/rrsHT--444 --. '))*J 8}J' @ NN !  H7 88r1cZ^^ [R"5nU(aXl[U5n[R "US5m[ R"T5n[RR[RR[RR[RR[RR[RR[RR[RR 1n[#U4SjU55(d [%S5e[UR'55n[R "US5m [#U 4SjU55(d [%S5e[)U5nXBlU[RR,:XaUR.R1U5 O XE;aUR2R1U5 UR&R5U5 U$)zConverts a Python dictionary into a map def. Args: dict_: Python dictionary name: (optional) the name of the map. Returns: MapProto: the converted map def. rc3V># UHn[R"U5T:Hv M g7fNr result_type)rkey raw_key_types r/rfrom_dict..sDts#|3t&)zfThe key type in the input dictionary is not the same for all keys and therefore is not valid as a map.c3V># UHn[R"U5T:Hv M g7frr)rvalraw_value_types r/rrsGr~~c"n4rzjThe value type in the input dictionary is not the same for all values and therefore is not valid as a map.)rZMapProtorrrrrrr[INT8rtINT32INT64UINT8ruUINT32UINT64rr]rrrrarrrCopyFrom) dict_rrrrvalid_key_int_typesr value_seqrrs @@r/rrs I  ;D>>$q'*L..|EEeLL ugDE FFr1c[R"U[R5(dBU[R[R [R [R4;a-[R"U5n[R"U5nO[R"U5n[R"XRUR5RU5$)zSaturate cast for numeric types. This function ensures that values outside the representable range of the target dtype are clamped to the maximum or minimum representable value of that dtype. )rrintegerrint4uint4int2uint2rroundfinfocliprrr)rr infos r/ saturate_castr ts }}UBJJ''5 5, u% HHQKu% 771hh ) 0 0 77r1)Tr)r np.ndarrayrboolr rreturnr )r>npt.NDArray[np.uint8]r?z Sequence[int]rr)rHr rr))ronnx.TensorProtorrrr )rHr rrr)r str | NonerHr rr)ronnx.SequenceProtor list[Any])NN)rrrrr int | Nonerr)r onnx.MapProtordict[Any, Any])rrrrrr)ronnx.OptionalProtor Any | None)rrrrr rrr)r)rz tuple[int]r np.dtyperintrr )rr r rrr ) __future__rrhtypingrrrnumpyr numpy.typingnptonnx.external_data_helperrZrcollections.abcrr0rCrKrOrRrrrrrrrrrrr r1r/r$sY# % ( H1H1H1H1 H1V '42 / '40 UoSd(8vP,BF6 6$64>66r922jP4CG0 0%05?00h;<GG$,G47GG>8r1