o q::jR@sHddlmZddlZddlZddlmZddlmZGdddeZ dS)) annotationsN)Base)expectc@seZdZedddZdS) QLinearMatMulreturnNonec CsldD]}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 } |tjkrtgdgd ggdgd 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)ir.r.)r0'itest_qlinearmatmul_3D_) getattrnponnxhelper make_nodearrayr astyper)quant_type_name quant_type dtype_namernoder r rrrrrroutputrDp/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/qlinearmatmul.py export_ints                   zQLinearMatMul.export_intN)rr)__name__ __module__ __qualname__ staticmethodrFrDrDrDrEr sr) __future__rnumpyr9r:onnx.backend.test.case.baseronnx.backend.test.case.noderrrDrDrDrEs