
    /j                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                  &    e Zd Zedd            ZdS )QLinearMatMulreturnNonec                    dD ]9} t          t          |           }dD ]}t          t          |          }t          j                            dg ddg          }t          j        g dg dg          }|t          j        k    r|d	z  }|                    |          }t          j        d
g|          }t          j        |t          j        k    rdgndg|          }t          j        g dg dg dg dg          }|t          j        k    r|d	z  }|                    |          }t          j        dg|          }	t          j        |t          j        k    rdgndg|          }
t          j        dg|          }t          j        |t          j        k    rdgndg|          }|t          j        k    rt          j        g dg dg          }nt          j        g dg dg          }|                    |          }t          ||||||	|
||g|gd|  d|            t          j        g dg dgg dg dgg          }|t          j        k    r|d	z  }|                    |          }t          j        d
g|          }t          j        |t          j        k    rdgndg|          }t          j        g dg dg dg dgg dg dg dg dgg          }|t          j        k    r|d	z  }|                    |          }t          j        dg|          }	t          j        dg|          }
t          j        dg|          }t          j        |t          j        k    rdgndg|          }|t          j        k    rV|t          j	        k    r#t          j        g dg d gg dg d gg          }nEt          j        g d!g d gg d!g d gg          }n"t          j        g dg dgg dg dgg          }|                    |          }t          ||||||	|
||g|gd"|  d|            ;d S )#N)uint8int8)float32float16r   )aa_scalea_zero_pointbb_scaleb_zero_pointy_scaley_zero_pointy)inputsoutputs)      r      )               gF%u{?)dtypeiq   )   3      )<      r   )r   r!      )r!         g!u|?ir   g'?v   ))   ir-   )   i   )   s   r   )r0   B      test_qlinearmatmul_2D__)r   r   name)u   x   )r3   '   i)r9   t   w   test_qlinearmatmul_3D_)
getattrnponnxhelper	make_nodearrayr   astyper   r   )quant_type_name
quant_type
dtype_namer"   noder   r   r   r   r   r   r   r   outputs                 n/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/qlinearmatmul.py
export_intzQLinearMatMul.export_int   sr   0 M	 M	O _55J4 K K
J//{,,#	 	 	 !E -    H0002C2C2CDEE((HAHHZ(((F85999!x#-#8#8YKKse:      H#^^]]]MMM???S  ((HAHHZ(((G9E:::!x#-#8#8YKKse:      (F85999!x#-#8#8YKKse:      ((X}}}lll&CDDFFX&EFFFz22$$$	 $HP/PPJPP   " H+++->->->?+++->->->?  ((HAHHZ(((F85999!x#-#8#8YKKse:      H'W'W  ((HAHHZ(((G9E:::!xZ@@@(F85999!x#-#8#8YKKse:      ((
**!#!0/// B!0/// B" " "$!0/// B!0/// B" "  X,__lll;,__lll; F  z22$$$	 $HP/PPJPP    {KM	 M	    N)r   r	   )__name__
__module____qualname__staticmethodrM    rN   rL   r   r      s8        N N N \N N NrN   r   )

__future__r   numpyrA   rB   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rS   rN   rL   <module>rX      s    # " " " " "      , , , , , , . . . . . .P P P P PD P P P P PrN   