bIghddlmZddlZdZdZdZdZ d dZ d dZ d dZ y) ) annotationsNicR|rtjntj}|rtnt}|rt nt }t|tjstj|}tj|||}tj|j|S)aCast to 4bit via rounding and clipping (without packing). Args: x: element to be converted signed: boolean, whether to convert to signed int4. Returns: An ndarray with a single int4 element (sign-extended to int8/uint8) ) npint8uint8INT4_MIN UINT4_MININT4_MAX UINT4_MAX isinstancendarrayasarraycliprintastype)xsigneddtypeclip_low clip_highclippeds I/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/subbyte.pyfloat32_to_4bit_unpackedrsnBGG288E!xyH" I a $ JJqMgga9-G 777  " "5 ))cHt||}t||}|dz|dzzS)afCast two elements to 4bit (via rounding and clipping) and pack to a single byte Args: val_low: element to be packed in the 4 LSB val_high: element to be packed in the 4 MSB signed: boolean, whether to convert to signed int4. Returns: An ndarray with a single int8/uint8 element, containing both int4 elements r)r)val_lowval_highri8_highi8_lows rfloat32x2_to_4bitx2r#$s/'x8G %gv 6F a<&4- ''rc4d} t|tjstj|}|dz}|dz }|r||n|}|r||n|}|rtjntj }|j ||j |fS)NcBtj|dz dk(||dzS)Nr)rwhere)rs rz&unpack_single_4bitx2..9sbhhQ1}aTBrrr)rrrrrr r)rr unpack_signedx_lowx_highrs runpack_single_4bitx2r-6sCM a $ JJqM HE !VF$*M% E&,]6 "&FBGG288E LL u!5 66r)rnp.ndarray | np.dtype | floatrboolreturn np.ndarray)rnp.dtyper r2rr/r0r1)rr.rr/r0ztuple[np.ndarray, np.ndarray]) __future__rnumpyrr r r r rr#r-rrr6s#    *$*.2**,( (!)(37(($7$7.27"7r