o r::j<@sddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z m Z mZmZmZmZmZGdddejZedkrKedSdS)) annotationsN) GraphProto ModelProtoOperatorSetIdProto TensorProtocheckerhelpershape_inferencec@seZdZd$d d Zd%d d Zd%ddZd%ddZd%ddZd%ddZd%ddZ d%ddZ d%ddZ d%ddZ d%ddZ d%d d!Zd%d"d#Zd%d$d%Zd%d&d'Zd%d(d)Zd%d*d+Zd%d,d-Zd%d.d/Zd%d0d1Zd%d2d3Zd%d4d5Zd%d6d7Zd%d8d9Zd%d:d;Zd%dd?Zd%d@dAZd%dBdCZd%dDdEZ d%dFdGZ!d%dHdIZ"d%dJdKZ#d%dLdMZ$d%dNdOZ%d%dPdQZ&d%dRdSZ'd%dTdUZ(d%dVdWZ)d%dXdYZ*d%dZd[Z+d%d\d]Z,d%d^d_Z-d%d`daZ.d%dbdcZ/d%dddeZ0d%dfdgZ1d%dhdiZ2d%djdkZ3d%dldmZ4d%dndoZ5d%dpdqZ6d%drdsZ7d&d'dwdxZ8d&d'dydzZ9d%d{d|Z:d%d}d~Z;d%ddZ<d%ddZ=d%ddZ>d%ddZ?d%ddZ@d%ddZAd%ddZBd%ddZCd%ddZDd%ddZEd%ddZFd%ddZGd%ddZHd%ddZId%ddZJd%ddZKd%ddZLd%ddZMd%ddZNd%ddZOd%ddZPd%ddZQd%ddZRd%ddZSddZTd%ddZUd%ddZVeWjWXdddeYjZiddeYjZdfddddiddeYjZdfddddiddeYjZdfgd(dd҄Z[d%ddԄZ\eWjWXddddddeYj]dfddddddeYj]dfddddddeYj]dfddddddeYj]dtfddddddeYj]dtfddddddeYj]dtfddddddeYj]dtfdddddeYj^ddtfddddddeYj_dtfg d)ddZ`eWjWXgdd*ddZad%ddZbd%ddZcd%ddZdeYjZdfd+ddZeeWjWXddgd,ddZfeWjWXddgd,ddZgeWjWXgdd-dd ZheWjWXgd d-d d ZieWjWXgd d.ddZjd%ddZkd%ddZld%ddZmd%ddZnd%ddZod%ddZpd%ddZqd%ddZreWjWXd eYjsfd!eYjtfgd/d"d#ZudS(0TestVersionConvertergraphrinitial_versionrtarget_versionintreturnrcCs.tj|d|gd}tj||}t||S)Nz onnx-test) producer_name opset_imports)r make_modelonnxversion_converterconvert_versionr check_model)selfr r r orig_modelconverted_modelrg/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/test/version_converter_test.py _converteds zTestVersionConverter._convertedNonecdfdd }t|dS)Nrrc stdddgdgtdddgdgtdddgdgg}t|d tdtjd tdtjd tdtjd gtdtjd g}|td d ddS)NAddWZshapeReshapeXAYtestr make_node make_graphmake_tensor_value_inforFLOATrmake_operatorsetidnodesr rrrr'.s z>TestVersionConverter.test_backwards_incompatible..testrr assertRaises RuntimeErrorrr'rr7rtest_backwards_incompatible-sz0TestVersionConverter.test_backwards_incompatiblec Cstdddgdgg}t|dtdtjdtdtjdgtdtjdg}||tddd }|jj d j dks>J|j d j d ksHJdS) NrX1X2r&r'r(r,r.r rr0r1r2rr3rr4r nodeop_type opset_importversionrr6r rrrrtest_backwards_compatibleC z.TestVersionConverter.test_backwards_compatiblecr)Nrrc sZtddgdgg}t|dtdtjdgtdtjdg}|tddddS) NCosr$r&r'r(r,r-r/r5r7rrr'Usz7TestVersionConverter.test_non_existent_op..testr8r9r<rr7rtest_non_existent_opTs z)TestVersionConverter.test_non_existent_opc Ctdddgdgg}t|dtdtjdtdtjdgtdtjdg}||tdd d }|jj d j dks>J|j d j d ksHJdS) Nrr>r?r&r'r(r*r,r-r)rrArFrrr test_add_8_5brHz!TestVersionConverter.test_add_8_5c CrL) Nrr>r?r&r'r(r*r,r)r-rrArFrrr test_add_5_8srHz!TestVersionConverter.test_add_5_8c Cstjdddgdgdddg}t|dtdtjd tdtjd gtdtjd g}||td d d }|jj dj dksAJ|jj dj dksLJ|j dj d ksVJdS)Nrr>r?r&rr+)axis broadcastr')r)r.r(r,r)r- UnsqueezerArFrrrtest_add_5_8_with_unsqueezes z0TestVersionConverter.test_add_5_8_with_unsqueezec CrL) NMulr>r?r&r'r(r*r,r-r)rrArFrrr test_mul_8_5rHz!TestVersionConverter.test_mul_8_5c CrL) NrSr>r?r&r'r(r*r,r)r-rrArFrrr test_mul_5_8rHz!TestVersionConverter.test_mul_5_8c Ctdgddgg}t|dtdtjdtdtjdtdtjdgtdtjdg}||td d d }|jj d j dksEJ|j d j d ksOJdS) NGemmr%BCr&r'r%r)r)rYrZr,r+r-rrArFrrr test_gemm_up<(z!TestVersionConverter.test_gemm_upc CrV) NrWrXr&r'r%r[rYrZr,r-r+rrArFrrrtest_gemm_downr]z#TestVersionConverter.test_gemm_downcr)Nrrc xtdgddgg}t|dtdtjdtdtjdtd tjd gtdtjdg}|td d d dS)NrWrXr&test_gemm_7_6_rejects_1d_inputr%r@rYrZrbr,rJr/r5r7rrr' zATestVersionConverter.test_gemm_7_6_rejects_1d_input..testr8r9r<rr7rr`z3TestVersionConverter.test_gemm_7_6_rejects_1d_inputcr)Nrrc r_)NrWrXr&test_gemm_6_7_rejects_1d_inputr%rarYrcrZrer,rJrfr/r5r7rrr'+rgzATestVersionConverter.test_gemm_6_7_rejects_1d_input..testr8r9r<rr7rri)rhz3TestVersionConverter.test_gemm_6_7_rejects_1d_inputcr)Nrrc r_)NrWrXr&test_gemm_7_6_rejects_1d_Ar% rYr@rbrZrer,rfrJr/r5r7rrr'=rgz=TestVersionConverter.test_gemm_7_6_rejects_1d_A..testr8r9r<rr7rrj;rhz/TestVersionConverter.test_gemm_7_6_rejects_1d_Acr)Nrrc r_)NrWrXr&test_gemm_6_7_rejects_1d_Ar%rkrYrmrZrer,rJrfr/r5r7rrr'Orgz=TestVersionConverter.test_gemm_6_7_rejects_1d_A..testr8r9r<rr7rrnMrhz/TestVersionConverter.test_gemm_6_7_rejects_1d_Ac Ctddgdgg}t|dtdtjdgtdtjdg}||tddd}|jj d j dks6J|j d j dks@JdS) NRelur$r&r'r(r,r)rfrrArFrrr test_relu_5_7`z"TestVersionConverter.test_relu_5_7c Cro) Nrpr$r&r'r(r,rfr)rrArFrrr test_relu_7_5nrrz"TestVersionConverter.test_relu_7_5c Ctdgddgg}t|dtdtjdtdtjdtd tjdtd tjdtd tjdgtdtjdg}||td d d}|jj dj dksSJ|j dj dks]JdS)NBatchNormalizationr$scalerYmeanvarr&r'r$r(rwr*rYrxryr,r-r)rrArFrrrtest_batch_normalization_8_5|$  z1TestVersionConverter.test_batch_normalization_8_5c Crt)Nrurvr&r'r$r(rwr*rYrxryr,r)r-rrArFrrrtest_batch_normalization_5_8r{z1TestVersionConverter.test_batch_normalization_5_8c Cstdgddgg}t|dtdtjdtdtjdtdtjdtd tjdtd tjdgtdtjd g}||td d d}|jj dj dksSJ|j dj dks]JdS)NConcatr>r?X3X4X5r&r'r>r*r?rrrr(r,r@r)rrArFrrrtest_concat_3_5s z$TestVersionConverter.test_concat_3_5c Cstjdgddgddg}t|dtdtjdtd tjdtd tjdtd tjdtd tjdgtdtjd g}||tddd}|jj dj dksUJ|j dj dks_JdS)Nr}r~r&rrOr'r>r*r?rrrr(r,r)r@rArFrrrtest_concat_5_3s  z$TestVersionConverter.test_concat_5_3c Cstjdgdgtdtjdgdgdtdddgd gg}t|d tdtjd gtd tjd g}||t dd d }|j j dj dksIJ|j djd ksSJdS)NConstantr"r,r+r)valuer#r$r&r'r(rJrbrrr0 make_tensorrINT64r1r2r3rr4r rBrCrDrErFrrrtest_reshape_6_4" z%TestVersionConverter.test_reshape_6_4c Cstjddgdgdgdg}t|dtdtjdgtdtjdg}||tdd d }|jj d j d ks9J|jj d j dksDJ|j d j d ksNJdS)Nr#r$r&r)r"r'r(r,rbrJrrr+rArFrrrtest_reshape_4_6sz%TestVersionConverter.test_reshape_4_6c Ctdgddgg}t|dtdtjdtdtjdtdtjdtd tjdtd tjdgtdtjdg}||td d d }|jj dj dksSJ|j dj d ks]JdS)NSumdata_0data_1data_2data_3data_4sumr'rr(rrrrr,rfr-rrArFrrr test_sum_7_8r{z!TestVersionConverter.test_sum_7_8c Cr)Nrrrr'rr(rrrrr,r)rfrrArFrrr test_sum_5_8r{z!TestVersionConverter.test_sum_5_8c Cr)Nrrrr'rr(rrrrr,r-r)rrArFrrr test_sum_8_5+r{z!TestVersionConverter.test_sum_8_5c Ctjddgdgddgdg}t|dtdtjdgtdtjdg}||tddd }|jj d j dks:J|j d j d ksDJdS N AveragePoolr$r&r+ kernel_shaper'r)r)r)r)r,r-rrArFrrrtest_averagepool_upCz(TestVersionConverter.test_averagepool_upc Ctjddgdgddgdg}t|dtdtjdgtdtjdg}||tdd d}|jj d j dks:J|j d j dksDJdSrrArFrrrtest_averagepool_downQrz*TestVersionConverter.test_averagepool_downc Cstjddgdgddg}t|dtdtjdgtdtjdg}||tddd }|jj d j dks8J|j d j d ksBJdS) NDropoutdataoutputr+)is_testr'r[r,r-rrArFrrrtest_dropout_up_s( z$TestVersionConverter.test_dropout_upc Cro) Nrrrr'r[r,r-r+rrArFrrrtest_dropout_downs( z&TestVersionConverter.test_dropout_downc Cd}d}tj}d}tjjddgdgdg}t|dtjd||gtjd||g}||td ||}|j j d j dksAJ|j j d j jj|ksNJ|jd j|ksXJdS) Nrfr-r.r@rbMaxr$r&inputsoutputstest_maxr,rrr3rrr0r1r2rr4r rBrCrtype tensor_type elem_typerDrEr from_opsetto_opset data_type data_shaper6r rrrr test_max_7_8"z!TestVersionConverter.test_max_7_8c Cr) Nrfr-rMinr$r&rtest_minr,rrrrrr test_min_7_8rz!TestVersionConverter.test_min_7_8c Cr) Nrfr-r@Meanr$r&r test_meanr,rrrrrr test_mean_7_8rz"TestVersionConverter.test_mean_7_8c Cr NMaxPoolr$r&r+rr'rr,r-rrArFrrrtest_maxpool_uprz$TestVersionConverter.test_maxpool_upc Csd}d}tj}tjjddgdgdddd g}t|d tjd|gd gtjd|gd g}||td ||}t |j j dksEJ|j j dj dksPJdd|j j dj D}d|vsbJd|vshJd|vsnJ|jdj|ksxJdS)NrJrfUpsampler$r&nearest@@)rrmode width_scale height_scaletest_upsample_6_7r+r+r.r.r+r+rbrJr,r+rcSsg|]}|jqSrname).0attrrrr sz:TestVersionConverter.test_upsample_6_7..scalesrr)rr3rrr0r1r2rr4lenr rBrC attributerDrE)rrrrr6r rattribute_namesrrrrs<     z&TestVersionConverter.test_upsample_6_7c CrrrArFrrrtest_maxpool_downrz&TestVersionConverter.test_maxpool_downc Csd}d}tj}tjdgddgdg}d}td||}td ||d g}td ||d g}td ||d g} td ||d g} td||} t|d|||| | g| g} || td||} | jj dj dksoJ| j dj |ksyJdS)Nr- ru)xsbiasrxryyr)r+r.r+r@rrr+rrxrytest_batchnormalization_8_9r,r) rr3rr0r2r1rr4r rBrCrDrErrrrr6 input_shaperrwrYrxryrr rrrrtest_batch_normalization_8_9)s0z1TestVersionConverter.test_batch_normalization_8_9c Csd}d}tj}tjjdgddgdg}d}tjd||}tjd ||d g}tjd ||d g}tjd ||d g} tjd ||d g} tjd||} tj|d|||| | g| g} || td||} | j j dj dkswJ| j dj |ksJdS)Nrr-rurvr&r)r.r@rbr)r$rwr+rYrxrytest_batchnormalizationr,r)rr3rrr0r2r1rr4r rBrCrDrErrrrtest_batchnormalization_9_8Js0z0TestVersionConverter.test_batchnormalization_9_8c Cd}d}tj}gd}td}tjdgdgtd|||dg}t|d gtj d||g}| |t d||}|j j d jdksFJ|j jd jjj|ksSJ|jd j|ks]JdS) Nr-rrrr&r,rrr test_constantr)rr3nparangerr0rr1rr2rr4r rBrCrrrrrDrE rrrr output_shape output_valuer6r rrrrtest_constant_8_9k0  z&TestVersionConverter.test_constant_8_9c Cr) Nrr-rrrr&r,rrr)rUINT64rrrr0rr1rr2rr4r rBrCrrrrrDrErrrrtest_constant_9_8rz&TestVersionConverter.test_constant_9_8c Csd}d}tj}tjjddgdgddg}t|dtjd|gd gtjd|d d gg}||td ||}|j j d j dksDJ|j j d j jj|ksQJ|jd j|ks[JdS)Nr-rFlattenr$r&r+rrrO test_flattenrr.rlr,rrrrrrr6r rrrrtest_flatten_8_9 z%TestVersionConverter.test_flatten_8_9c Csd}d}tj}tjjddgdgddg}t|dtjd|gd gtjd|d d gg}||td ||}|j j dj dksDJ|j j d j jj|ksQJ|jd j|ks[JdS)Nrr-rr$r&r+rrrr.rlr,rrrrrr0r1r2rr4r rBrCrrrrrDrErrrrtest_flatten_9_8rz%TestVersionConverter.test_flatten_9_8c Csd}d}tj}tjjdddgdgdg}gd}t|d tjd||tjd||gtjd||g}||td ||}|j j d j dksKJ|j j d j jj|ksXJ|jd j|ksbJdS) Nr-rPRelur$Sloper&rr test_prelur,rrrrrrr6rr rrrrtest_prelu_8_9& z#TestVersionConverter.test_prelu_8_9c Csd}d}tj}tjjdddgdgdg}gd}t|d tjd||tjd||gtjd||g}||td ||}|j j d j dksKJ|j j d j jj|ksXJ|jd j|ksbJdS) Nrr-rr$rr&rrrr,r.rrrrrrtest_prelu_9_8rz#TestVersionConverter.test_prelu_9_8c Cd}d}tj}tjjdddgdgdg}gd}t|d tjd||tjd||gtjdtj|g}||t d ||}|j j d j dksLJ|j j d jjjtjksZJ|jd j|ksdJdS) Nr-rGreaterr>r?r&rr test_greaterr,rrr3rrr0r1r2BOOLrr4r rBrCrrrrrDrErrrrtest_greater_8_9* z%TestVersionConverter.test_greater_8_9c Cd}d}tj}tjjdddgdgdg}gd}t|d tjd||tjd||gtjdtj|g}||t d ||}|j j d j dksLJ|j j d jjjtjksZJ|jd j|ksdJdS) Nrr-rr>r?r&rrrr,r.rrrrrr0r1r2rrr4r rBrCrrrrrDrErrrrtest_greater_9_8/rz%TestVersionConverter.test_greater_9_8c Cr) Nr-rLessr>r?r&rr test_lessr,rrrrrr test_less_8_9Mrz"TestVersionConverter.test_less_8_9c Cr ) Nrr-r r>r?r&rrr r,r.rr rrrr test_less_9_8krz"TestVersionConverter.test_less_9_8c Csd}d}tj}tjjdddgdgdg}t|dtjd|d d gtjd|d d ggtjd|d d gg}||td ||}|j j d j dksMJ|j j d j jj|ksZJ|jd j|ksdJdS) Nr-rMatMulr>r?r&r test_matmulr@rbr,rrrrrrtest_matmul_8_9$ z$TestVersionConverter.test_matmul_8_9c Csd}d}tj}tjjdddgdgdg}t|dtjd|d d gtjd|d d ggtjd|d d gg}||td ||}|j j d j dksMJ|j j d j jj|ksZJ|jd j|ksdJdS)Nrr-rr>r?r&rrr@rbr,r.rrrrrrtest_matmul_9_8rz$TestVersionConverter.test_matmul_9_8c Csd}d}tj}tjjdgddgdg}t|dtjd|d d gtjd |d d gtjd |d d ggtjd|d d gg}||td ||}|j j dj dksVJ|j j dj jj|kscJ|jdj|ksmJdS)Nr-rrWr>r?rr&r test_gemmr>r@rbr?rr,rrrrrr test_gemm_8_9( z"TestVersionConverter.test_gemm_8_9c Csd}d}tj}tjjdgddgdg}t|dtjd|d d gtjd |d d gtjd |d d ggtjd|d d gg}||td ||}|j j d j dksVJ|j j dj jj|kscJ|jdj|ksmJdS)Nrr-rWrr&rrr>r@rbr?rr,rrrrrr test_gemm_9_8rz"TestVersionConverter.test_gemm_9_8c Csd}d}tj}tjjddgdgdgddg}t|d tjd|gd gtjd|gd g}||td ||}t |j j d ksFJ|j j dj dksQJ|j j dj dks\Jt |j j dj dksiJ|j j dj djdkswJ|jdj|ksJdS)Nr-rrr$r&r?rrr)rrrrtest_upsample_8_9rrr,r.rrr+r)rr3rrr0r1r2rr4rr rBrCrrrDrErrrrrs4 z&TestVersionConverter.test_upsample_8_9F raw_scaleboolc Cs<d}d}tj}tjjdddgdgddg}gd }tjdtjjd g|r/ttjd g|Rn||}t |d tj d|gd tj d|d ggtj d|gdg|g}| |t d||} | j jdjdksmJt| j jdkswJt| j jdjdksJ| j jdjdjdksJ| jdj|ksJdS)Nrr-rr$Scalesr&rrrrrrb4f test_upsamplerrr,rr.r+r)rr3rrr0rbytesstructpackr1r2rr4r rBrCr initializerrrrDrE) rrrrrr6 scale_value scale_tensorr rrrr helper_upsample_with_initializers@ z5TestVersionConverter.helper_upsample_with_initializerc CsPd}d}tj}gd}tjdtjjdg|r!ttjdg|Rn||}tjjdgdg|d tjjd d dgd gd dg}tj |dtj d |gdgtj d |gdgtj d|dggd}| |t d||} t | jjdksvJ| jjdjd ksJt | jjdjdksJ| jjdjdjdksJ| jdj|ksJdS)Nrr-r const_valuerbr!rConstant_Outputrrr$r&rr r"rr) value_infor,r+rr.r)rr3rrrr#r$r%r0r1r2rr4rr rBrCrrrDrE) rrrrrr'r(r6r rrrrhelper_upsample_with_constantCsJ   z2TestVersionConverter.helper_upsample_with_constantcC|jdddSNFrr-r7rrr$test_upsample_with_constant_node_9_8qz9TestVersionConverter.test_upsample_with_constant_node_9_8cCr.r/)r)r7rrr"test_upsample_with_initializer_9_8ur3z7TestVersionConverter.test_upsample_with_initializer_9_8cCr.NTr0r1r7rrr&test_upsample_with_raw_initializer_9_8yr3z;TestVersionConverter.test_upsample_with_raw_initializer_9_8cCr.r5r1r7rrr(test_upsample_with_raw_constant_node_9_8}r3z=TestVersionConverter.test_upsample_with_raw_constant_node_9_8c Csfd}d}tj}tjjdddgdgd}tjjddgd gd}tj||gd tjd|d gtjd|d ggtjd|d gtjd |d gg}d }tjjd |ddgddg|ddg}tjd|dd g} tjd|gd} tjd|dd g} tjd|gd} tj|d| | g| | g} || td ||}|j j dj d ksJ|j dj |ksJdS)Nr-rrsum_innextsum_outrIdentityscan_out scan_bodyr.r,Scaninitialrrzr+)rrbodynum_scan_inputs)r+r@r. test_scan_8_9r)rr3rrr0r1r2rr4r rBrCrDrE)rrrrnode1node2gno_sequence_lensr6r?rrr@r rrrrrCsT  z"TestVersionConverter.test_scan_8_9c Csd}d}tj}tj}tjjddgdgtjdg}t|dtjd|dd ggtjd|dd gg}||t d ||}|j j d j dksHJ|j j d jjj|ksUJ|jd j|ks_JdS) Nr-rCastr$r&)rrto test_castr.r@r,r)rr3UINT32rrr0r1r2rr4r rBrCrrrrrDrE)rrrdata_type_from data_type_tor6r rrrr test_cast_8_9s(z"TestVersionConverter.test_cast_8_9c Cstjdgdgtdtjdgddgdtdddgd d gg}t|d tdtjd gtd tjd td tjdg}||t ddd}|j j dj dksRJ|j djdks\JdS)Nrsplitr,r.r@rSplitr$Y1Y2r'r(r.r rlrrrFrrrtest_split_13_12s&  z%TestVersionConverter.test_split_13_12c Cstjddgddgddg}t|dtdtjdgtdtjd tdtjd g}||td d d }|jj d j dks@J|j d j d ksJJt |jj d jdksWJdS)NrPr$rQrRr+rtest_split_optional_input)rJrr,rlrr.)rr0r1r2rr3rr4r rBrCrDrErrrFrrrtest_split_with_optional_inputs z3TestVersionConverter.test_split_with_optional_inputc Cstjddgddgddgdg}t|dtdtjd gtdtjd tdtjd g}||td d d}|jj dj dksBJ|jj dj dksMJ|j dj dksWJdS)NrPr$rQrRr.r@)rOr'r(rSrr,rlrTrrr+rArFrrrtest_split_12_13s z%TestVersionConverter.test_split_12_13c Cstjddgddgddg}t|dtdtjdgtdtjd tdtjd g}||td d d }|jj dj dks@J|j dj d ksJJd S)z7Test Split 13->12 conversion with optional split input.rPr$rQrRrrtest_split_13_12_optional r(r,rTrlNrArFrrrtest_split_13_12_optional_inputs z4TestVersionConverter.test_split_13_12_optional_inputc Cstjdgdgtdtjdgdgdtdddgd gg}t|d tdtjd gtd tjd g}||t dd d}|j j dj dksIJ|j djdksSJdS)Nraxesr,r+rr ReduceSumr$r&r'r[r+r)rTrlrrFrrrtest_axes_input_to_attr_13_12$rz2TestVersionConverter.test_axes_input_to_attr_13_12c Cstjddgdgdgdg}t|dtdtjdgtdtjdg}||td d d }|jj dj d ks9J|j dj d ksCJdS) Nr_r$r&r)r^r'r[r`r,rlrTrrArFrrrtest_axes_attr_to_input_12_13:sz2TestVersionConverter.test_axes_attr_to_input_12_13c Cstddgdgg}t|dtdtjdgtdtjdg}||tddd }|jj d j dks6J|j d j d ks@Jd S) z8Test Squeeze 13->11 conversion with optional axes input.Squeezer$r&test_squeeze_13_11_optional)r+r\r+r[r,rT rNrArFrrr test_squeeze_13_11_optional_axesHsz5TestVersionConverter.test_squeeze_13_11_optional_axesc Cstjddgdgddgdg}t|dtdtjdgtdtjd g}||td d d}|jj d j d ks:J|jj dj dksEJ|j d j dksOJdS)zFTest Reshape 4->5 conversion - shape in attribute should become input.r#r$r&r.r)rtest_reshape_4_5r[r.r)r,rbrrr+NrArFrrrtest_reshape_4_5_optional_shapeWsz4TestVersionConverter.test_reshape_4_5_optional_shapec Cstjdgdgtdtjdggddtjdddgd gd d g}t|d tdtjd gtd tjdg}||tddd}|j dj dksKJdS)z:Test Resize 10->11 conversion with proper bounds checking.rrr,rb)rrrrrResizer$r&r)rtest_resize_10_11r)r+r+rbrbr\rerN) rr0rrr3r1r2rr4rDrErFrrrtest_resize_10_11_bounds_checkgs$ z3TestVersionConverter.test_resize_10_11_bounds_checkc Cstjdgddgddg}t|dtdtjdtd tjd td tjd gtdtjdg}||td d d}|j j dj dksGJ|j dj dksQJdS)z;Test Scatter 10->11 conversion with proper bounds checking.Scatterrindicesupdatesr&rrtest_scatter_10_11rrrorSrpr,r\reScatterElementsN)rr0r1r2rr3rrr4r rBrCrDrErFrrrtest_scatter_10_11_bounds_check~s z4TestVersionConverter.test_scatter_10_11_bounds_checkc Cstjddgdgddgddgddgdg}t|d tdtjd gtdtjd g}||td d d}|jj dj dks@J|jj dj dksKJ|jj dj dksVJ|jj dj dksaJ|j dj dkskJt |jj djdksxJt |jj djdksJdS)NSlicer$r&rr+r@r\)r^startsendsr')r\r))r@r\r)r,rrr.rb)rr0r1r2rr3rr4r rBrCrDrErinputrrFrrrtest_slice_9_10s$z$TestVersionConverter.test_slice_9_10c Cs0d}d}tj}d}d}d}d}d}tjjdgdd d g|d g} t| d tjd ||||gtjd||||gtjd||||gtjd||d|ggtjd ||||gg} || td ||} | j j dj dksqJ| j dj |ks{Jt| j j djdksJ| j j djdjdksJdS)NrTr+r.r@r)RNNr$r Rr,Y_hrr hidden_sizetest_rnnr$r r}rYrlayoutrr3rrr0r1r2rr4r rBrCrDrErrr rrrr seq_length batch_size input_sizenum_directionsrr6r rrrrtest_rnn_13_14sV      z#TestVersionConverter.test_rnn_13_14c Cs8d}d}tj}d}d}d}d}d}tjjdgdd d g|d g} t| d tjd ||||gtjd||d||gtjd||d||gtjd||d|ggtjd ||||gg} || td ||} | j j dj dksuJ| j dj |ksJt| j j djdksJ| j j djdjdksJdS)NrTrzr+r.r@r)GRUr|r,r~rtest_grur$r r}rYrJrrrrrrrtest_gru_13_14V    z#TestVersionConverter.test_gru_13_14c Cs8d}d}tj}d}d}d}d}d}tjjdgdd d g|d g} t| d tjd ||||gtjd||d||gtjd||d||gtjd||d|ggtjd ||||gg} || td ||} | j j dj dksuJ| j dj |ksJt| j j djdksJ| j j djdjdksJdS)NrTrzr+r.r@r)LSTMr|r,r~r test_lstmr$r rbr}rYr-rrrrrrrtest_lstm_13_14rz$TestVersionConverter.test_lstm_13_14c Csd}d}tj}d}d}d}d}d}tjjdgdd d g|d d g} t| d tjd||||gtjd||||gtjd||||gtjd||d|ggtjd ||||gg} || td ||} | j j d j dksrJ| j d j |ks|Jt| j j d jdksJdS)NrzrTr+r.r@r)r{r|r,r~rrrrrrr$r r}rYrr3rrr0r1r2rr4r rBrCrDrErrrrrrtest_rnn_14_13KsV     z#TestVersionConverter.test_rnn_14_13c Csd}d}tj}d}d}d}d}d}tjjdgdd d g|d d g} t| d tjd||||gtjd||d||gtjd||d||gtjd||d|ggtjd ||||gg} || td ||} | j j d j dksvJ| j d j |ksJt| j j d jdksJdS)NrzrTr+r.r@r)rr|r,r~rrrr$r r}rYrJrrrrrtest_gru_14_13V   z#TestVersionConverter.test_gru_14_13c Csd}d}tj}d}d}d}d}d}tjjdgdd d g|d d g} t| d tjd||||gtjd||d||gtjd||d||gtjd||d|ggtjd ||||gg} || td ||} | j j d j dksvJ| j d j |ksJt| j j d jdksJdS)NrzrTr+r.r@r)rr|r,r~rrrr$r rbr}rYr-rrrrrtest_lstm_14_13rz$TestVersionConverter.test_lstm_14_13c Csd}tjddgdg|dg}t|dtdtjdgtdtjdg}||td d d }|jj d j dks:J|j d j d ksDJdS)Nrr+r.rr.r+Padr$r&)padsr'r+r.r.r+r)r)r,r\rer+rrArrr6r rrrrtest_pad_10_11sz#TestVersionConverter.test_pad_10_11c Csd}tjddgdg|ddg}t|dtdtjdgtdtjd g}||td d d }|jj d j dks;J|j dj d ksEJdS)Nrrr$r&r)rrr'rrr,r\rer+rrArrrrtest_pad_with_value_10_11sz.TestVersionConverter.test_pad_with_value_10_11cCsd}d}tj}dg}tjjddgdgddd g}tjd||g}tjd||g}t|d ||}tjjddgdgddd g} tjd||g} tjd||g} t| d | | } tjjd d gd g|| dg} tjd tjgg}tjd ||g}td||ddgg}t| d|||}| |t d||}|j j dj d ksJ|jdj|ksJ|j j djdjj dj dksJt|j j djdjj djdksJ|j j djdjj dj dksJt|j j djdjj djdksJdS)Nr\rer.Clipsub_insub_outrr)rrminmaxthen_gIfcondout)rr then_branch else_branchg@g@test_subgraphsr,rr+)rr3rrr0r2r1rrrr4r rBrCrDrErrFr)rrrrr subg1_node subg1_input subg1_outputsubg1 subg2_node subg2_input subg2_outputsubg2rBrxrinitr convertedrrrtest_if_subgraph_10_11 sV $($,z+TestVersionConverter.test_if_subgraph_10_11c Cstdgddgg}ddg}tjdtjjdg|}dd g}tjd tjjdg|}d d g}tjd tjjdg|}ddg}tjdtjjdg|} t|dtdtjdgtdtjdg|||| g} || t ddd} | j j dj dks|J| j djdksJdS)Nrurvr&g?g ףp= ?rwr.g333333?gHzG?rYgzG?g?rxg)\(?g{Gz?ryr'r$)r+r.r.r@r,rerlr)rr0rrrr3r1r2rr4r rBrCrDrE) rr6r'r(b_valueb_tensor mean_value mean_tensor var_value var_tensorr rrrr'test_initializer_not_in_input_above_ir4Gs8  zd}tjddgdg|dg}t|dtdtjdgtdtjdg}||tdd d }|jj dj d ks:J|jj d j d ksEJ|jj d j dj dksSJ|jj d j dj |ksaJ|jj dj dkslJ|jj dj dj dkszJ|jj dj dj dksJ|jj dj dksJ|jdjd ksJdS)NrSoftmaxr$r&rr'r+r.r@r,rerTShaper+rrOr.r@r#rr0r1r2rr3rr4r rBrCrrirDrErrOr6r rrrrtest_softmax_12_13js$z'TestVersionConverter.test_softmax_12_13c Csd}tjddgdg|dg}t|dtdtjdgtdtjdg}||tdd d }|jj d j dks:J|jj d j d j d ksHJ|jj d j d j d ksVJ|jd jd ks`JdS)Nrrr$r&rr'rr,rTrlrrOr.rrrrrtest_softmax_13_12sz'TestVersionConverter.test_softmax_13_12cast_9_8rHrIrr-r* softmax_12_13rrOr+rlrT upsample_9_10rrrr\r_strrCattrsdict[str, int | str]rr output_typertuple[int, ...]c s0dfdd }ttjf|dS)Nrrc shtjgdgfig}t|ddddgtdg}|tddS)Nr&test_r_rejects_missing_inputr,)rr0r1lowerr2rr4r5rrrCrrrrrrr'szGTestVersionConverter.test_rejects_missing_required_inputs..testr8r:r;r InferenceError) rrrCrrrrrr'rrr$test_rejects_missing_required_inputss) z9TestVersionConverter.test_rejects_missing_required_inputscs$dfdd }ttjf|dS)Nrrc sbtjdgdgddtjddgdgddg}t|dgtdtjd g}|td d d dS) NrFr+rrr&r2test_softmax_13_12_rejects_malformed_flatten_inputr*r,rTrlr/r5r7rrr'szUTestVersionConverter.test_softmax_13_12_rejects_malformed_flatten_input..testr8rr<rr7rrs zGTestVersionConverter.test_softmax_13_12_rejects_malformed_flatten_input per_tensorr@NTper_axis_none_block_shaperper_axis_zero_block_shaperper_tensor_positive_block_shaper.per_axis_positive_block_shape blocked_2drarb blocked_3drbr@ rbr@r-per_axis_output_dtypeper_axis_unsupported_typex_shape scale_shape block_size output_dtype int | Nonezero_point_dtype compatiblec s`dfdd } |r tnt} | | ||||||WddS1s)wYdS)Nrrc stjdddgdg|||dg}tdtj|tdtj|g}|r3|td|||djd|p6|}t|d|td||g} | t d d d } dS) NQuantizeLinearr$Sr&)rOrrZPrr'r,rw) rr0r2rr3appendrxr1rr4) rrrOrrrr6r output_type_r rr7rrr'" s2  z6TestVersionConverter.test_quantize_21_20..testr8 contextlib nullcontextr:r;) rrrrrOrrrrr'context_managerrr7rtest_quantize_21_20s N "z(TestVersionConverter.test_quantize_21_20))rrr*NNT)rrrr+NT)rrrr+rT)rrr*r+r.F)rrrr+r.F)rrrarrbF)rrrr.rbFy_shapec s\dfdd }|r tnt}||||||WddS1s'wYdS)Nrrc s~tjdgddg||dg}t|dtdtj|tdtj|tdtj|gtdtj|g}|td d d }dS) NDequantizeLinear)r$rrr&)rOrr'r$rrr,rrw) rr0r1r2rINT8r3rr4)rrrOrr6r rr7rrr'[ s$  z8TestVersionConverter.test_dequantize_21_20..testr8r) rrrrrOrrr'rrr7rtest_dequantize_21_20G s "z*TestVersionConverter.test_dequantize_21_20c Cst}d}tjj|tj}tjj dtj j t ||dd}tjj dtj j gdgd}tjj dddgd gd }tjjd |ggtjd tj j t |g||gd }tj|d }d} tjddg} tjj|| d} d| _tj| |dd| dddtjtj|ddd} |t| jjddd| jjD} |d| | d}||jt j |t|j!ddd|j!D}||d| ||dd||d d!WddS1swYdS)"Nr.r@initializer_tensorT)rrdimsvalsrawinitializer_scalarr)rrrrr sum_outputrSimpleAddition)rr6rrr&ztest_simple_add.onnxztest_simple_add.onnx.datar,rwrr\rF)save_as_external_dataall_tensors_to_one_filelocationsize_thresholdconvert_attribute)load_external_datarr.cSsi|]}|j|qSrr)rrrrr szNTestVersionConverter.test_external_data_version_conversion..r@cSsi|]}|j|jqSr)keyr)redrrrr  sroffset0length24)"tempfileTemporaryDirectoryrrandomrandastypefloat32rrrrr3listr0r1r2ospathjoin make_opsetidr ir_version save_modelrrload assertEqualrr r&assertIn data_locationEXTERNAL external_data)rtemp_dirr" random_datarradd_node graph_defmodel_filename data_filenamer model_defrtensors large_tensorexternal_data_dictrrr%test_external_data_version_conversionw sv    "z:TestVersionConverter.test_external_data_version_conversionc Cstdgddgg}t|dtdtjdtdtjdtdtjdgtdtjdg}||td d d }|j d j d ksDJdS) NWhererrrrwhererrrrr,rr) rr0r1r2rrr3rr4rDrErr6r rrrrtest_where_16_15_success  z-TestVersionConverter.test_where_16_15_successcr)Nrrc sxtdgddgg}t|dtdtjdtdtjdtdtjdgtdtjdg}|td d d dS) Nr1r2r where_bf16rrrrr,rr4) rr0r1r2rrBFLOAT16rr4r5r7rrr' rgzBTestVersionConverter.test_where_bfloat16_16_15_fails..testr8r9r<rr7rtest_where_bfloat16_16_15_fails sz4TestVersionConverter.test_where_bfloat16_16_15_failsop_namedtype reduction str | Nonec Csdddddifdddifd}||\}}}}t|}|dur#||d <tj|gd d gfi|g} t| |td ||td tj|td||gtd ||gS)zIBuild a graph for ScatterElements or ScatterND with standard test shapes.rrOr)rbr))r.r+rh)rr ScatterNDNr=rnrrrorp)dictrr0r1rr2rr) rr;r<r=scatter_graph_configdata_s indices_s updates_srr6rrr_make_scatter_graph s$   z(TestVersionConverter._make_scatter_graph)rr)r?cCs>||tjd}||tddd}|jdjdksJdS)Nnoner,rr4r)rErr3rrr4rDrE)rr;r rrrrtest_scatter_16_15_success sz/TestVersionConverter.test_scatter_16_15_successcs dfdd }t|dS)Nrrcs(tj}|tddddSNr,rr4)rErr9rrr4r r;rrrr' szDTestVersionConverter.test_scatter_16_15_bfloat16_fails..testr8r9)rr;r'rrJr!test_scatter_16_15_bfloat16_fails sz6TestVersionConverter.test_scatter_16_15_bfloat16_fails)rraddrrmulr?rMr?rOc"dfdd }t|dS)Nrrc*tj}|tddddSrHrErr3rrr4rIr;r=rrrr' zMTestVersionConverter.test_scatter_16_15_reduction_add_mul_fails..testr8r9rr;r=r'rrUr*test_scatter_16_15_reduction_add_mul_fails  z?TestVersionConverter.test_scatter_16_15_reduction_add_mul_fails))rrr)rrr)r?r)r?rcrR)NrrcrSNr,rWrTrIrUrrr'+ rVzMTestVersionConverter.test_scatter_18_17_reduction_max_min_fails..testr8r9rWrrUr*test_scatter_18_17_reduction_max_min_fails rYz?TestVersionConverter.test_scatter_18_17_reduction_max_min_fails))rrN)rrrFrLrN)r?N)r?rFrPrQcCs4||tj|}||tddd}t|dSrZ)rErr3rrr4rr)rr;r=r rrrr-test_scatter_18_17_allowed_reductions_success1 szBTestVersionConverter.test_scatter_18_17_allowed_reductions_successc Cstjdtjdgtjddgtjddd}tjtjdd dgd d gd d gdt d tj dgt d tj dt d tj dg|gd}| |t ddd}t dd|jjd jD}t|jddgksdJdS)NrOr.r+r@r<TrrPr$Y0rQrrrFrbr.r+r.r@r.r&r,rTrlcs|] }|jdkr|VqdSrONrrarrr U zMTestVersionConverter.test_split_13_12_raw_data_initializer..rrrrrarrayint64r1r0r2r3rr4r9r rBrrints)r split_initr rrrrr%test_split_13_12_raw_data_initializerF s z:TestVersionConverter.test_split_13_12_raw_data_initializerc Cstjdtjdgtjddgtjddd}tjtdddgd ggd t dtj d gt d tj d g|gd }| |t ddd}t dd|jjD}t dd|jD}t|jddgksaJdS)Nr"r.r)r^Tr_r#r$r&rFr[rhrdr,rbcsre)r#NrCrnrrrrid rjzMTestVersionConverter.test_reshape_5_4_raw_data_initializer..csre)r"Nrrgrrrrie rjrk)r shape_initr rreshaperrrr%test_reshape_5_4_raw_data_initializerX sz:TestVersionConverter.test_reshape_5_4_raw_data_initializerc Cstjdtjdgtjdgtjddd}tjtdddgd ggd t dtj d gt d tj d g|gd }| |t ddd}t dd|jjdjD}t|jdgksXJdS)Nr^r+rr^Tr_r_r$r&rFr[r`rdr,rTrlcsre)r^Nrrgrrrrit rjzNTestVersionConverter.test_axes_input_to_attr_13_12_raw_data..rk)r axes_initr rrrrr&test_axes_input_to_attr_13_12_raw_datah sz;TestVersionConverter.test_axes_input_to_attr_13_12_raw_datac Cstjdtjgtjdgtjddd}tjtdgddggd t d tj d gt dtj d g|gd }| |t d dd}t dd|jjdjD}|jdksTJdS)NrOr+r^Tr_DFT)r$r,rOr&rFr$)r+r-r.rdr,rwcsre)rONrrgrrrri rjzDTestVersionConverter.test_dft_20_19_raw_data_axis..r)rrrrrrlrmr1r0r2r3rr4r9r rBrr)r axis_initr rrrrrtest_dft_20_19_raw_data_axisw sz1TestVersionConverter.test_dft_20_19_raw_data_axisc Csttjdgtddd}ttjdgdg|dtjdd dgd d gd d gdtd tjdgtd tjdtd tjdg}tj |t ddgd}| t t j|dWddS1scwYdS)Nr.z.csrerfrrgrrrri rjrk)rr~r rrOrrrr'test_split_13_12_raw_data_constant_node s" z.css|]}|jdkVqdS)rNrrgrrrri s)rr0r1r2rr3rr4rDrEr9r rBanyr)rr6r r range_noderrr#test_range_27_26_stash_type_removed s" z8TestVersionConverter.test_range_27_26_stash_type_removedfloat16bfloat16cs dfdd }t|dS)Nrrc sptdgddgg}t|dtdgtdgtdggtddg}|tdd d dS) Nrrrrange_27_26_fp16rrrr,rr)rr0r1r2rr4r5r<rrrr' s    zGTestVersionConverter.test_range_27_26_low_precision_fails..testr8r9)rrr<r'rrr$test_range_27_26_low_precision_fails sz9TestVersionConverter.test_range_27_26_low_precision_fails)r rr rr rrrr8)F)rrrr)rrrCrrrrrrrrrrrrr)rrrrrrrOrrrrrrrrrrr)rrrrrrrOrrrrrrr)r;rr<rr=r>rr)r;rrr)r;rr=rrr)r;rr=r>rr)rrr<rrr)v__name__ __module__ __qualname__rr=rGrKrMrNrRrTrUr\r^r`rirjrnrqrsrzr|rrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrrr)r-r2r4r6r7rCrNrUrXrYr]rarbrfrirlrsryrrrrrrrrrrrr parameterizedexpandrr3rrr FLOAT8E4M3FNUINT16rrr0r6r:rErGrKrXr\r]rprvrxr|rrrrFLOAT16r9rrrrrr s           0 /                      &  ! ! ! !            "* .     6             6 6 6 6 6 6  ; #         C0 % J       r __main__) __future__rrrr$runittestnumpyrronnx.version_converterrrrrrrrr TestCaser rmainrrrrsF $ n