o wi@sHddlmZddlZddlZddlmZddlmZGdddeZ dS)) annotationsN)Base)expectc@seZdZedddZdS) QLinearMatMulreturnNonec CsdD]}tt|}dD]}tt|}tjjdgddgd}tgdgdg}|tjkr4|d 8}||}tjd g|d }tj|tjkrKd gnd g|d }tgdgdgdgdg}|tjkrk|d 8}||}tjdg|d } tj|tjkrdgndg|d } tjdg|d } tj|tjkrdgndg|d } |tjkrtgdgdg} n tgdgdg} | |} t|||||| | | | g| gd|d|dtgdgdggdgdgg}|tjkr|d 8}||}tjd g|d }tj|tjkr d gnd g|d }tgdgdgdgdggdgdgdgdgg}|tjkr;|d 8}||}tjdg|d } tjdg|d } tjdg|d } tj|tjkrcdgndg|d } |tjkr|tj krtgdgd ggdgd gg} n'tgd!gd ggd!gd gg} ntgdgdggdgdgg} | |} t|||||| | | | g| gd"|d|dq qdS)#N)uint8int8)float32float16r)aa_scale a_zero_pointbb_scale b_zero_pointy_scale y_zero_pointy)inputsoutputs)r)gF%u{?)dtypeiq)3)<r)rr)rg!u|?irg'?v))ir*)i)sr)r-Btest_qlinearmatmul_2D__)rrname)ux)r0'i)r6twtest_qlinearmatmul_3D_) getattrnponnxhelper make_nodearrayr astyperr )quant_type_name quant_type dtype_namernoder r rrrrrroutputrIf/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/qlinearmatmul.py export_ints                    zQLinearMatMul.export_intN)rr)__name__ __module__ __qualname__ staticmethodrKrIrIrIrJr sr) __future__rnumpyr>r?onnx.backend.test.case.baseronnx.backend.test.case.noderrrIrIrIrJs