§
    1�Å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edd„¦   «         Zedd„¦   «         Zedd„¦   «         Zedd„¦   «         Zedd„¦   «         Z	edd	„¦   «         Z
edd
„¦   «         Zedd„¦   «         Zedd„¦   «         ZdS )ÚSplitÚreturnÚNonec            
     ó:  — t          j        g d¢¦  «                             t           j        ¦  «        } t          j                             ddgg d¢d¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        d	d
g¦  «                             t           j        ¦  «        t          j        ddg¦  «                             t           j        ¦  «        g}t          || g|dt          j                             dd¦  «        g¬¦  «         t          j        ddg¦  «                             t           j	        ¦  «        }t          j                             dddgddgd¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        g d¢¦  «                             t           j        ¦  «        g}t          || |g|dt          j                             dd¦  «        g¬¦  «         d S )N©ç      ð?ç       @ç      @ç      @ç      @ç      @r   Úinput©Úoutput_1Úoutput_2Úoutput_3r   ©ÚinputsÚoutputsÚaxisr   r   r   r   r   r   Ú!test_split_equal_parts_1d_opset13Ú é   ©r   r   ÚnameÚopset_importsé   é   Úsplitr   r   ©r   r   r   r   Ú$test_split_variable_parts_1d_opset13©
ÚnpÚarrayÚastypeÚfloat32ÚonnxÚhelperÚ	make_noder   Úmake_opsetidÚint64©Ú
node_inputÚnodeÚexpected_outputsr#   s       úRC:\Projects\Historical\venv\Lib\site-packages\onnx/backend/test/case/node/split.pyÚexport_1d_opset13zSplit.export_1d_opset13   sö  € å”XÐ<Ð<Ð<Ñ=Ô=×DÒDÅRÄZÑPÔPˆ
åŒ{×$Ò$ØØ�9Ø8Ð8Ð8Øð	 %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ð
Ðõ
 	ØØ�<Ø$Ø4Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$ØØ˜WÐ%Ø Ð,Øð	 %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒHÐ)Ð)Ð)Ñ*Ô*×1Ò1µ"´*Ñ=Ô=ð
Ðõ 	ØØ Ð&Ø$Ø7Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
ð 	
ð 	
ó    c            
     óþ  — t          j        g d¢g d¢g¦  «                             t           j        ¦  «        } t          j                             ddgddgd¬¦  «        }t          j        g d	¢g d
¢g¦  «                             t           j        ¦  «        t          j        g d¢g d¢g¦  «                             t           j        ¦  «        g}t          || g|dt          j                             dd¦  «        g¬¦  «         t          j        ddg¦  «                             t           j	        ¦  «        }t          j                             dddgddgd¬¦  «        }t          j        ddgddgg¦  «                             t           j        ¦  «        t          j        g d¢g d¢g¦  «                             t           j        ¦  «        g}t          || |g|dt          j                             dd¦  «        g¬¦  «         d S )Nr   ©ç      @ç       @ç      "@ç      $@ç      &@ç      (@r   r   r   r   é   r   ©r   r   r   ©r9   r:   r;   ©r   r   r   ©r<   r=   r>   Ú!test_split_equal_parts_2d_opset13r   r   r   r!   r"   r#   r   r   r9   r:   r$   ©r;   r<   r=   r>   Ú$test_split_variable_parts_2d_opset13r&   r0   s       r4   Úexport_2d_opset13zSplit.export_2d_opset13:   s  € å”XØ+Ð+Ð+Ð-NÐ-NÐ-NÐOñ
ô 
ç
Š&•”Ñ
Ô
ð 	õ Œ{×$Ò$Ø˜W˜I°
¸JÐ/GÈað %ñ 
ô 
ˆõ
 ŒH�o�o�o   Ð7Ñ8Ô8×?Ò?ÅÄ
ÑKÔKÝŒH�o�o�oÐ'9Ð'9Ð'9Ð:Ñ;Ô;×BÒBÅ2Ä:ÑNÔNð
Ðõ
 	ØØ�<Ø$Ø4Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$ØØ˜WÐ%Ø Ð,Øð	 %ñ 
ô 
ˆõ ŒH�s˜C�j 3¨ *Ð-Ñ.Ô.×5Ò5µb´jÑAÔAÝŒHÐ*Ð*Ð*Ð,CÐ,CÐ,CÐDÑEÔE×LÒLÝ”
ñô ð
Ðõ 	ØØ Ð&Ø$Ø7Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c            
     ó6  — t          j        g d¢¦  «                             t           j        ¦  «        } t          j                             ddgg d¢¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        dd	g¦  «                             t           j        ¦  «        t          j        d
dg¦  «                             t           j        ¦  «        g}t          || g|dt          j                             dd¦  «        g¬¦  «         t          j        ddg¦  «                             t           j	        ¦  «        }t          j                             dddgddg¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        g d¢¦  «                             t           j        ¦  «        g}t          || |g|dt          j                             dd¦  «        g¬¦  «         d S )Nr   r   r   r   ©r   r   r   r   r   r   r   r   Ú+test_split_equal_parts_default_axis_opset13r   r   r   r!   r"   r#   r   r   r$   Ú.test_split_variable_parts_default_axis_opset13r&   r0   s       r4   Úexport_default_values_opset13z#Split.export_default_values_opset13h   sï  € å”XÐ<Ð<Ð<Ñ=Ô=×DÒDÅRÄZÑPÔPˆ
õ Œ{×$Ò$Ø˜W˜IÐ/SÐ/SÐ/Sð %ñ 
ô 
ˆõ
 ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ð
Ðõ
 	ØØ�<Ø$Ø>Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$Ø˜W gÐ.¸ÀZÐ8Pð %ñ 
ô 
ˆõ
 ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒHÐ)Ð)Ð)Ñ*Ô*×1Ò1µ"´*Ñ=Ô=ð
Ðõ 	ØØ Ð&Ø$ØAÝœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c            
     ó¨  — t          j        g ¦  «                             t           j        ¦  «        } t          j        g d¢¦  «                             t           j        ¦  «        }t
          j                             dddgg d¢¬¦  «        }t          j        g ¦  «                             t           j        ¦  «        t          j        g ¦  «                             t           j        ¦  «        t          j        g ¦  «                             t           j        ¦  «        g}t          || |g|dt
          j         	                    dd	¦  «        g¬
¦  «         d S )N©r   r   r   r   r   r#   r   rI   Ú#test_split_zero_size_splits_opset13r   r   r   )
r'   r(   r)   r*   r/   r+   r,   r-   r   r.   ©r1   r#   r2   r3   s       r4   Úexport_zero_size_splits_opset13z%Split.export_zero_size_splits_opset13�   s  € õ ”X˜b‘\”\×(Ò(­¬Ñ4Ô4ˆ
õ ”˜˜˜Ñ#Ô#×*Ò*­2¬8Ñ4Ô4ˆÝŒ{×$Ò$ØØ˜WÐ%Ø8Ð8Ð8ð %ñ 
ô 
ˆõ ŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ÝŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ÝŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ð
Ðõ
 	ØØ Ð&Ø$Ø6Ýœ;×3Ò3°B¸Ñ;Ô;Ð<ð	
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 ó¼  — t          j        g d¢¦  «                             t           j        ¦  «        } t          j                             ddgg d¢dd¬¦  «        }t          j        dd	g¦  «                             t           j        ¦  «        t          j        d
dg¦  «                             t           j        ¦  «        t          j        ddg¦  «                             t           j        ¦  «        g}t          || g|d¬¦  «         t          j        ddg¦  «                             t           j        ¦  «        }t          j                             dddgddgd¬¦  «        }t          j        dd	g¦  «                             t           j        ¦  «        t          j        g d¢¦  «                             t           j        ¦  «        g}t          || |g|d¬¦  «         d S )Nr   r   r   r   r   é   ©r   r   r   Únum_outputsr   r   r   r   r   r   Ú!test_split_equal_parts_1d_opset18©r   r   r   r!   r"   r#   r   r   r   r$   Ú$test_split_variable_parts_1d_opset18©	r'   r(   r)   r*   r+   r,   r-   r   r/   r0   s       r4   Úexport_1d_opset18zSplit.export_1d_opset18©   sÉ  € å”XÐ<Ð<Ð<Ñ=Ô=×DÒDÅRÄZÑPÔPˆ
åŒ{×$Ò$ØØ�9Ø8Ð8Ð8ØØð %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ð
Ðõ
 	ØØ�<Ø$Ø4ð		
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$ØØ˜WÐ%Ø Ð,Øð	 %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒHÐ)Ð)Ð)Ñ*Ô*×1Ò1µ"´*Ñ=Ô=ð
Ðõ 	ØØ Ð&Ø$Ø7ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 ó€  — t          j        g d¢g d¢g¦  «                             t           j        ¦  «        } t          j                             ddgddgdd¬	¦  «        }t          j        g d
¢g d¢g¦  «                             t           j        ¦  «        t          j        g d¢g d¢g¦  «                             t           j        ¦  «        g}t          || g|d¬¦  «         t          j        ddg¦  «                             t           j        ¦  «        }t          j                             dddgddgd¬¦  «        }t          j        ddgddgg¦  «                             t           j        ¦  «        t          j        g d¢g d¢g¦  «                             t           j        ¦  «        g}t          || |g|d¬¦  «         d S )Nr   r8   r   r   r   r   r?   r!   rT   r@   rA   rB   rC   Útest_split_equal_parts_2drW   r"   r#   r   r   r   r9   r:   r$   rE   Ú$test_split_variable_parts_2d_opset18rY   r0   s       r4   Úexport_2d_opset18zSplit.export_2d_opset18Ô   sê  € å”XØ+Ð+Ð+Ð-NÐ-NÐ-NÐOñ
ô 
ç
Š&•”Ñ
Ô
ð 	õ Œ{×$Ò$ØØ�9Ø Ð,ØØð %ñ 
ô 
ˆõ ŒH�o�o�o   Ð7Ñ8Ô8×?Ò?ÅÄ
ÑKÔKÝŒH�o�o�oÐ'9Ð'9Ð'9Ð:Ñ;Ô;×BÒBÅ2Ä:ÑNÔNð
Ðõ
 	ØØ�<Ø$Ø,ð		
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$ØØ˜WÐ%Ø Ð,Øð	 %ñ 
ô 
ˆõ ŒH�s˜C�j 3¨ *Ð-Ñ.Ô.×5Ò5µb´jÑAÔAÝŒHÐ*Ð*Ð*Ð,CÐ,CÐ,CÐDÑEÔE×LÒLÝ”
ñô ð
Ðõ 	ØØ Ð&Ø$Ø7ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 ó¸  — t          j        g d¢¦  «                             t           j        ¦  «        } t          j                             ddgg d¢d¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        d	d
g¦  «                             t           j        ¦  «        t          j        ddg¦  «                             t           j        ¦  «        g}t          || g|d¬¦  «         t          j        ddg¦  «                             t           j        ¦  «        }t          j                             dddgddg¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        g d¢¦  «                             t           j        ¦  «        g}t          || |g|d¬¦  «         d S )Nr   r   r   r   rS   ©r   r   rU   r   r   r   r   r   r   Ú+test_split_equal_parts_default_axis_opset18rW   r!   r"   r#   r   r   rI   r$   Ú.test_split_variable_parts_default_axis_opset18rY   r0   s       r4   Úexport_default_values_opset18z#Split.export_default_values_opset18  sÃ  € å”XÐ<Ð<Ð<Ñ=Ô=×DÒDÅRÄZÑPÔPˆ
õ Œ{×$Ò$ØØ�9Ø8Ð8Ð8Øð	 %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ð
Ðõ
 	ØØ�<Ø$Ø>ð		
ñ 	
ô 	
ð 	
õ ”˜!˜Q˜Ñ Ô ×'Ò'­¬Ñ1Ô1ˆÝŒ{×$Ò$Ø˜W gÐ.¸ÀZÐ8Pð %ñ 
ô 
ˆõ
 ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒHÐ)Ð)Ð)Ñ*Ô*×1Ò1µ"´*Ñ=Ô=ð
Ðõ 	ØØ Ð&Ø$ØAð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 óh  — t          j        g ¦  «                             t           j        ¦  «        } t          j        g d¢¦  «                             t           j        ¦  «        }t
          j                             dddgg d¢¬¦  «        }t          j        g ¦  «                             t           j        ¦  «        t          j        g ¦  «                             t           j        ¦  «        t          j        g ¦  «                             t           j        ¦  «        g}t          || |g|d¬¦  «         d S )	NrN   r   r   r#   r   rI   Ú#test_split_zero_size_splits_opset18rW   )	r'   r(   r)   r*   r/   r+   r,   r-   r   rP   s       r4   Úexport_zero_size_splits_opset18z%Split.export_zero_size_splits_opset18,  s  € õ ”X˜b‘\”\×(Ò(­¬Ñ4Ô4ˆ
õ ”˜˜˜Ñ#Ô#×*Ò*­2¬8Ñ4Ô4ˆÝŒ{×$Ò$ØØ˜WÐ%Ø8Ð8Ð8ð %ñ 
ô 
ˆõ ŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ÝŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ÝŒH�R‰LŒL×Ò¥¤
Ñ+Ô+ð
Ðõ
 	ØØ Ð&Ø$Ø6ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 ór  — t          j        g d¢¦  «                             t           j        ¦  «        } t          j                             ddgg d¢d¬¦  «        }t          j        ddg¦  «                             t           j        ¦  «        t          j        d	d
g¦  «                             t           j        ¦  «        t          j        ddg¦  «                             t           j        ¦  «        t          j        dg¦  «                             t           j        ¦  «        g}t          || g|d¬¦  «         d S )N)r   r   r   r   r   r   r9   r   r   )r   r   r   Úoutput_4r"   r`   r   r   r   r   r   r   r9   Ú"test_split_1d_uneven_split_opset18rW   ©r'   r(   r)   r*   r+   r,   r-   r   ©r1   r2   r3   s      r4   Úexport_1d_uneven_split_opset18z$Split.export_1d_uneven_split_opset18E  s  € å”XÐAÐAÐAÑBÔB×IÒIÍ"Ì*ÑUÔUˆ
õ Œ{×$Ò$ØØ�9ØDÐDÐDØð	 %ñ 
ô 
ˆõ ŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c˜3�ZÑ Ô ×'Ò'­¬
Ñ3Ô3ÝŒH�c�U‰OŒO×"Ò"¥2¤:Ñ.Ô.ð	
Ðõ 	ØØ�<Ø$Ø5ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   c                 ó2  — t          j        g d¢g d¢g¦  «                             t           j        ¦  «        } t          j                             ddgg d¢dd¬¦  «        }t          j        g d	¢g d
¢g¦  «                             t           j        ¦  «        t          j        g d¢g d¢g¦  «                             t           j        ¦  «        t          j        ddgddgg¦  «                             t           j        ¦  «        g}t          || g|d¬¦  «         d S )N)r   r   r   r   r   r   r9   r:   )r;   r<   r=   r>   ç      *@ç      ,@ç      .@ç      0@r   r   r   r?   rS   rT   r@   )r;   r<   r=   rB   )r>   rn   ro   r9   r:   rp   rq   Ú"test_split_2d_uneven_split_opset18rW   rj   rk   s      r4   Úexport_2d_uneven_split_opset18z$Split.export_2d_uneven_split_opset18^  s2  € å”Xà8Ð8Ð8Ø?Ð?Ð?ðñ
ô 
÷
 Š&•”Ñ
Ô
ð 	õ Œ{×$Ò$ØØ�9Ø8Ð8Ð8ØØð %ñ 
ô 
ˆõ ŒH�o�o�oÐ'8Ð'8Ð'8Ð9Ñ:Ô:×AÒAÅ"Ä*ÑMÔMÝŒH�o�o�oÐ'9Ð'9Ð'9Ð:Ñ;Ô;×BÒBÅ2Ä:ÑNÔNÝŒH�s˜C�j 4¨ ,Ð/Ñ0Ô0×7Ò7½¼
ÑCÔCð
Ðõ 	ØØ�<Ø$Ø5ð		
ñ 	
ô 	
ð 	
ð 	
ð 	
r6   N)r   r	   )Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr5   rG   rL   rQ   rZ   r^   rc   rf   rl   rs   © r6   r4   r   r      sX  € € € € € Øð)
ð )
ð )
ñ „\ð)
ðV ð+
ð +
ð +
ñ „\ð+
ðZ ð$
ð $
ð $
ñ „\ð$
ðL ð
ð 
ð 
ñ „\ð
ð2 ð(
ð (
ð (
ñ „\ð(
ðT ð-
ð -
ð -
ñ „\ð-
ð^ ð%
ð %
ð %
ñ „\ð%
ðN ð
ð 
ð 
ñ „\ð
ð0 ð
ð 
ð 
ñ „\ð
ð0 ð
ð 
ð 
ñ „\ð
ð 
ð 
r6   r   )
Ú
__future__r   Únumpyr'   r+   Úonnx.backend.test.case.baser   Úonnx.backend.test.case.noder   r   rx   r6   r4   ú<module>r}      s�   ðð #Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð à €€€Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,Ø .Ð .Ð .Ð .Ð .Ð .ðm
ð m
ð m
ð m
ð m
ˆDñ m
ô m
ð m
ð m
ð m
r6   