
     TiC                     j    S r SSKrSSKJr  SSKJr  SSKJr  SSK	J
r
Jr  S rS r " S S	5      rS
 rg)z
Onnxscript version of "hf-internal-testing_tiny-random-bart".

See: https://huggingface.co/hf-internal-testing/tiny-random-bart
    N)script)opset20)FLOATINT64c                    ^ ^^^^^^^^^	^
^^^^^^ [        5       S[        S   S[        S   4UU UUUUUUUU	U
UUUUUU4S jj5       nUR                  5       $ )N	input_ids)   Nreturn)NN   c           
      "  > [         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n	[         R                  " T5      n
[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " T5      n[         R                  " U 5      n[         R                  " SS9n[         R                  " UU5      n[         R                  " S/S9n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " UUSS9n[         R                  " U USS9n[         R                  " TU5      n[         R                  " S/S9n[         R                  " UU5      n[         R                  " U 5      n[         R                  " SS9n[         R                  " UUSS9n [         R                  " SS9n![         R                  " USS	9n"[         R                  " SS9n#[         R                  " U!U"U#5      n$[         R                  " S/S9n%[         R
                  " U U%5      n&[         R                  " S/S9n'[         R                  " U&U'SS9n([         R                  " S/S9n)[         R                  " U(U)5      n*[         R                  " U*5      n+[         R                  " U+[        R                  " [        R                  " S/[        R                   S
95      S9n,[         R                  " S/S9n-[         R                  " U,U-5      n.[         R"                  " U*U.5      n/[         R$                  " U/U,U*5      n0[         R&                  " U$U05      n1[         R                  " SS9n2[         R(                  " U1U25      n3[         R                  " TU35      n4[         R                  " U4SS	9n5[         R(                  " UU55      n6[         R*                  " U6UUSSS9n7[         R                  " U75      n8[         R                  " SS9n9[         R                  " U8U9SS9n:[         R                  " U75      n;[         R                  " SS9n<[         R                  " U;U<SS9n=[         R,                  " U7T5      n>[         R(                  " UU>5      n?[         R                  " S/S9n@[         R
                  " U:U@5      nA[         R                  " S/S9nB[         R                  " S/S9nC[         R                  " S/S9nD[         R                  " UAUBUCUDSS9nE[         R                  " S/S9nF[         R
                  " U:UF5      nG[         R                  " S/S9nH[         R                  " S/S9nI[         R                  " S/S9nJ[         R                  " UGUHUIUJSS9nK[         R                  " S/S9nL[         R
                  " U:UL5      nM[         R                  " S/S9nN[         R                  " S/S9nO[         R                  " S/S9nP[         R                  " UMUNUOUPSS9nQ[         R                  " U?UESS9nR[         R.                  " UR/ SQS9nS[         R,                  " U7T5      nT[         R(                  " TUT5      nU[         R,                  " U7T5      nV[         R(                  " UUV5      nW[         R                  " UUUKSS9nX[         R                  " UWUQSS9nY[         R.                  " UY/ SQS9nZ[         R                  " US5      n[[         R                  " S/S9n\[         R                  " S/S9n][         R0                  " U[U\U]5      n^[         R                  " U^SS	9n_[         R2                  " U_5      n`[         R                  " S/S9na[         R4                  " UaU`5      nb[         R                  " UbSS	9nc[         R.                  " UX/ SQS9nd[         R2                  " Uc5      ne[         R                  " USUe5      nf[         R2                  " Uc5      ng[         R                  " UdUg5      nh[         R,                  " UfUh5      ni[         R6                  " UiSS9nj[         R,                  " UjUZ5      nk[         R.                  " Uk/ SQS9nl[         R                  " S/S9nm[         R
                  " U:Um5      nn[         R                  " S/S9no[         R
                  " U=Uo5      np[         R                  " S/S9nq[         R                  " UnUpUqSS9nr[         R                  " UlUrSS9ns[         R,                  " UsT5      nt[         R(                  " UUt5      nu[         R(                  " U7Uu5      nv[         R*                  " UvTUSSS9nw[         R,                  " UwT5      nx[         R(                  " TUx5      ny[         R8                  " UySS9nz[         R,                  " UzT5      n{[         R(                  " UU{5      n|[         R(                  " UwU|5      n}[         R*                  " U}UUSSS9n~[         R                  " U~5      n[         R                  " SS9n[         R                  " UUSS9n[         R                  " U~5      n[         R                  " SS9n[         R                  " UUSS9n[         R,                  " U~T5      n[         R(                  " U	U5      n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " S/S9n[         R                  " S/S9n[         R                  " S/S9n[         R                  " UUUUSS9n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " S/S9n[         R                  " S/S9n[         R                  " S/S9n[         R                  " UUUUSS9n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " S/S9n[         R                  " S/S9n[         R                  " S/S9n[         R                  " UUUUSS9n[         R                  " UUSS9n[         R.                  " U/ SQS9n[         R,                  " U~T5      n[         R(                  " UU5      n[         R,                  " U~T5      n[         R(                  " U
U5      n[         R                  " UUSS9n[         R                  " UUSS9n[         R.                  " U/ SQS9n[         R                  " U5      n[         R                  " S/S9n[         R                  " S/S9n[         R0                  " UUU5      n[         R                  " USS	9n[         R2                  " U5      n[         R                  " S/S9n[         R4                  " UU5      n[         R                  " USS	9n[         R.                  " U/ SQS9n[         R2                  " U5      n[         R                  " UU5      n[         R2                  " U5      n[         R                  " UU5      n[         R,                  " UU5      n[         R6                  " USS9n[         R,                  " UU5      n[         R.                  " U/ SQS9n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " S/S9n[         R
                  " UU5      n[         R                  " S/S9n[         R                  " UUUSS9n[         R                  " UUSS9n[         R,                  " UT5      n[         R(                  " UU5      n[         R(                  " U~U5      n[         R*                  " UUUSSS9n[         R,                  " UT5      n[         R(                  " UU5      n[         R8                  " USS9n[         R,                  " UT5      n[         R(                  " UU5      n[         R(                  " UU5      n[         R*                  " UUUSSS9nU$ )Nr	   )valuer   )axis)	allowzerog      ?   )todtype   g   >)r   epsilon   )r   r   r	      )perml    )r   r   r   r	   r   g  #@none)approximate)r   IdentityShapeConstantGather	UnsqueezeConcatReshapeMulCastRangeConstantOfShapeirtensornparrayint64EqualWhereExpandAddLayerNormalizationMatMul	TransposeSliceSqrtDivSoftmaxGelu)r    encoder_layernorm_embedding_bias"encoder_layernorm_embedding_weight&encoder_layers_1_final_layer_norm_bias(encoder_layers_1_final_layer_norm_weightencoder_layers_1_fc2_bias*encoder_layers_1_self_attn_layer_norm_bias,encoder_layers_1_self_attn_layer_norm_weight(encoder_layers_1_self_attn_out_proj_bias&encoder_layers_1_self_attn_q_proj_bias&encoder_layers_1_self_attn_v_proj_bias&encoder_layers_1_self_attn_k_proj_bias&encoder_layers_0_final_layer_norm_bias(encoder_layers_0_final_layer_norm_weightencoder_layers_0_fc2_biasencoder_layers_1_fc1_bias(encoder_layers_0_self_attn_out_proj_bias&encoder_layers_0_self_attn_q_proj_bias&encoder_layers_0_self_attn_v_proj_biasencoder_shape_output_0encoder_constant_output_0encoder_gather_output_0encoder_constant_1_output_0unsqueeze_43encoder_unsqueeze_output_0encoder_concat_output_0encoder_reshape_output_0$encoder_embed_tokens_gather_output_0&encoder_embed_tokens_constant_output_0!encoder_embed_tokens_mul_output_0&encoder_embed_positions_shape_output_0)encoder_embed_positions_constant_output_0'encoder_embed_positions_gather_output_0+encoder_embed_positions_constant_1_output_0%encoder_embed_positions_cast_output_0+encoder_embed_positions_constant_2_output_0&encoder_embed_positions_range_output_0+encoder_embed_positions_constant_3_output_0*encoder_embed_positions_unsqueeze_output_0+encoder_embed_positions_constant_4_output_0'encoder_embed_positions_concat_output_0+encoder_embed_positions_constant_5_output_0(encoder_embed_positions_reshape_output_0(encoder_embed_positions_shape_1_output_00encoder_embed_positions_constantofshape_output_0+encoder_embed_positions_constant_6_output_0$encoder_embed_positions_mul_output_0&encoder_embed_positions_equal_output_0&encoder_embed_positions_where_output_0'encoder_embed_positions_expand_output_0+encoder_embed_positions_constant_7_output_0$encoder_embed_positions_add_output_0)encoder_embed_positions_gather_1_output_0encoder_cast_output_0encoder_add_output_07encoder_layernorm_embedding_layernormalization_output_0)encoder_layers_0_self_attn_shape_output_0,encoder_layers_0_self_attn_constant_output_0*encoder_layers_0_self_attn_gather_output_0+encoder_layers_0_self_attn_shape_1_output_0.encoder_layers_0_self_attn_constant_1_output_0,encoder_layers_0_self_attn_gather_1_output_01encoder_layers_0_self_attn_q_proj_matmul_output_0.encoder_layers_0_self_attn_q_proj_add_output_0unsqueeze_88-encoder_layers_0_self_attn_unsqueeze_output_0.encoder_layers_0_self_attn_constant_2_output_0.encoder_layers_0_self_attn_constant_3_output_0.encoder_layers_0_self_attn_constant_4_output_0*encoder_layers_0_self_attn_concat_output_0unsqueeze_97/encoder_layers_0_self_attn_unsqueeze_1_output_0.encoder_layers_0_self_attn_constant_5_output_0.encoder_layers_0_self_attn_constant_6_output_0.encoder_layers_0_self_attn_constant_7_output_0,encoder_layers_0_self_attn_concat_1_output_0unsqueeze_106/encoder_layers_0_self_attn_unsqueeze_2_output_0.encoder_layers_0_self_attn_constant_8_output_0.encoder_layers_0_self_attn_constant_9_output_0/encoder_layers_0_self_attn_constant_10_output_0,encoder_layers_0_self_attn_concat_2_output_0+encoder_layers_0_self_attn_reshape_output_0-encoder_layers_0_self_attn_transpose_output_01encoder_layers_0_self_attn_k_proj_matmul_output_0.encoder_layers_0_self_attn_k_proj_add_output_01encoder_layers_0_self_attn_v_proj_matmul_output_0.encoder_layers_0_self_attn_v_proj_add_output_0-encoder_layers_0_self_attn_reshape_1_output_0-encoder_layers_0_self_attn_reshape_2_output_0/encoder_layers_0_self_attn_transpose_1_output_0+encoder_layers_0_self_attn_shape_2_output_0/encoder_layers_0_self_attn_constant_11_output_0/encoder_layers_0_self_attn_constant_12_output_0)encoder_layers_0_self_attn_slice_output_0(encoder_layers_0_self_attn_cast_output_0(encoder_layers_0_self_attn_sqrt_output_0/encoder_layers_0_self_attn_constant_13_output_0'encoder_layers_0_self_attn_div_output_0*encoder_layers_0_self_attn_cast_1_output_0/encoder_layers_0_self_attn_transpose_2_output_0*encoder_layers_0_self_attn_sqrt_1_output_0'encoder_layers_0_self_attn_mul_output_0*encoder_layers_0_self_attn_sqrt_2_output_0)encoder_layers_0_self_attn_mul_1_output_0*encoder_layers_0_self_attn_matmul_output_0+encoder_layers_0_self_attn_softmax_output_0,encoder_layers_0_self_attn_matmul_1_output_0/encoder_layers_0_self_attn_transpose_3_output_0unsqueeze_145/encoder_layers_0_self_attn_unsqueeze_3_output_0unsqueeze_147/encoder_layers_0_self_attn_unsqueeze_4_output_0/encoder_layers_0_self_attn_constant_14_output_0,encoder_layers_0_self_attn_concat_3_output_0-encoder_layers_0_self_attn_reshape_3_output_03encoder_layers_0_self_attn_out_proj_matmul_output_00encoder_layers_0_self_attn_out_proj_add_output_0encoder_layers_0_add_output_0Aencoder_layers_0_self_attn_layer_norm_layernormalization_output_0$encoder_layers_0_fc1_matmul_output_0!encoder_layers_0_fc1_add_output_0,encoder_layers_0_activation_fn_gelu_output_0$encoder_layers_0_fc2_matmul_output_0!encoder_layers_0_fc2_add_output_0encoder_layers_0_add_1_output_0=encoder_layers_0_final_layer_norm_layernormalization_output_0)encoder_layers_1_self_attn_shape_output_0,encoder_layers_1_self_attn_constant_output_0*encoder_layers_1_self_attn_gather_output_0+encoder_layers_1_self_attn_shape_1_output_0.encoder_layers_1_self_attn_constant_1_output_0,encoder_layers_1_self_attn_gather_1_output_01encoder_layers_1_self_attn_q_proj_matmul_output_0.encoder_layers_1_self_attn_q_proj_add_output_0unsqueeze_176-encoder_layers_1_self_attn_unsqueeze_output_0.encoder_layers_1_self_attn_constant_2_output_0.encoder_layers_1_self_attn_constant_3_output_0.encoder_layers_1_self_attn_constant_4_output_0*encoder_layers_1_self_attn_concat_output_0unsqueeze_185/encoder_layers_1_self_attn_unsqueeze_1_output_0.encoder_layers_1_self_attn_constant_5_output_0.encoder_layers_1_self_attn_constant_6_output_0.encoder_layers_1_self_attn_constant_7_output_0,encoder_layers_1_self_attn_concat_1_output_0unsqueeze_194/encoder_layers_1_self_attn_unsqueeze_2_output_0.encoder_layers_1_self_attn_constant_8_output_0.encoder_layers_1_self_attn_constant_9_output_0/encoder_layers_1_self_attn_constant_10_output_0,encoder_layers_1_self_attn_concat_2_output_0+encoder_layers_1_self_attn_reshape_output_0-encoder_layers_1_self_attn_transpose_output_01encoder_layers_1_self_attn_k_proj_matmul_output_0.encoder_layers_1_self_attn_k_proj_add_output_01encoder_layers_1_self_attn_v_proj_matmul_output_0.encoder_layers_1_self_attn_v_proj_add_output_0-encoder_layers_1_self_attn_reshape_1_output_0-encoder_layers_1_self_attn_reshape_2_output_0/encoder_layers_1_self_attn_transpose_1_output_0+encoder_layers_1_self_attn_shape_2_output_0/encoder_layers_1_self_attn_constant_11_output_0/encoder_layers_1_self_attn_constant_12_output_0)encoder_layers_1_self_attn_slice_output_0(encoder_layers_1_self_attn_cast_output_0(encoder_layers_1_self_attn_sqrt_output_0/encoder_layers_1_self_attn_constant_13_output_0'encoder_layers_1_self_attn_div_output_0*encoder_layers_1_self_attn_cast_1_output_0/encoder_layers_1_self_attn_transpose_2_output_0*encoder_layers_1_self_attn_sqrt_1_output_0'encoder_layers_1_self_attn_mul_output_0*encoder_layers_1_self_attn_sqrt_2_output_0)encoder_layers_1_self_attn_mul_1_output_0*encoder_layers_1_self_attn_matmul_output_0+encoder_layers_1_self_attn_softmax_output_0,encoder_layers_1_self_attn_matmul_1_output_0/encoder_layers_1_self_attn_transpose_3_output_0unsqueeze_232/encoder_layers_1_self_attn_unsqueeze_3_output_0unsqueeze_234/encoder_layers_1_self_attn_unsqueeze_4_output_0/encoder_layers_1_self_attn_constant_14_output_0,encoder_layers_1_self_attn_concat_3_output_0-encoder_layers_1_self_attn_reshape_3_output_03encoder_layers_1_self_attn_out_proj_matmul_output_00encoder_layers_1_self_attn_out_proj_add_output_0encoder_layers_1_add_output_0Aencoder_layers_1_self_attn_layer_norm_layernormalization_output_0$encoder_layers_1_fc1_matmul_output_0!encoder_layers_1_fc1_add_output_0,encoder_layers_1_activation_fn_gelu_output_0$encoder_layers_1_fc2_matmul_output_0!encoder_layers_1_fc2_add_output_0encoder_layers_1_add_1_output_0encoder_outputencoder_embed_positions_weightencoder_embed_tokens_weightencoder_layers_0_fc1_bias&encoder_layers_0_self_attn_k_proj_bias,encoder_layers_0_self_attn_layer_norm_weight
matmul_257
matmul_267
matmul_268
matmul_270
matmul_271
matmul_272
matmul_273
matmul_283
matmul_284
matmul_286
matmul_287
matmul_288s                                                                                                                                                                                                         g/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/rewriter/models/_bart_encoder.py
main_graphmake_model.<locals>.main_graph$   s   +2+;+;8,
( .5-=-=8.
* 291A1A22
. 4;3C3C84
0 %,$4$45[$\!5<5E5E26
2 8?7G7G88
4 4;3C3C24
0 291A1A22
. 291A1A22
. 291A1A22
. 291A1A22
. 4;3C3C84
0 %,$4$45[$\!$+$4$45N$O!3:3C3C24
0 291A1A22
. 291A1A22
. ")y!9$+$4$41$=!").."$=#
 '.&6&6bT&B#''qc2%,%6%67NP\%]"")..')C!#
 $+??.!$
  07~~')A0
, 291A1A1N.,3KK02X-
) 29y1I.4;4D4D14M129..253
/
 7>6F6FQ6O307=TYZ0[-6=6F6FQ6O3187172
.
 7>6F6FaS6Q35<5F5F376
2 7>6F6FbT6R329..673
/
 7>6F6FbT6R33:??374
0 4;==44
0 <C;R;R4))BHHaS9:<
8 7>6F6FbT6R3/6{{<70
, 2946Z2
. 292<42
.
 3:..24Z3
/ 7>6F6FQ6O3/6{{370
, 5<NN*,P5
1 !(-V[\ ]&{{-/D 
 CJB\B\ .,)C
? 5<MMC5
1 8?7G7Ga7P45<^^586
2
 7>mmC7
3 :A9I9IPQ9R67>~~7:8
4
 =DNNCZ=
9 :A2=:
6 ''qc28?8I8I69
5 :A9I9IQSPT9U69@9I9IQRPS9T69@9I9IQRPS9T65<^^9:::6
2 ''qc2:A:K:K6;
7 :A9I9IQSPT9U69@9I9IQRPS9T69@9I9IQRPS9T67>~~;:::8
4  ((s3:A:K:K6;
7 :A9I9IQSPT9U69@9I9IQRPS9T6:A:J:JRSQT:U77>~~;::;8
4 7>oo:67
3
 9@8I8I7l9
5 =DNNCZ=
9 :A2=:
6 =DNNCZ=
9 :A2=:
6 9@:89
5
 9@:89
5
 ;B:K:K9;
7 7>mm97
3 ;B:J:JRTQU:V7:A:J:J&';
7 5<MM7;;5
1
 4;<<5!4
0 4;<<44
0 ;B:J:JRUQV:W729++;43
/ 6=\\36
2 ;B:K:K9;
7 6=\\66
2 3:++963
/ 6=\\66
2 5<KK;65
1 6=^^35^6
2 7>oo6R7
3 8?~~7;8
4 ;B:K:K8|;
7  ((s3:A:K:K6;
7  ((s3:A:K:K8-;
7 ;B:J:JRTQU:V77>~~;;;	8
4 9@;89
5
 ?Fnn9:?
; <C;;4?<
8 )0C<)
%
 &&-<0, 	J 07~~Mz0
, -4KK%'K-
) 8?||-68
4 07~~8*0
, -4KK%'K-
) +2++M-+
'
 &&/86- 	F 5<MMI5
1 8?7G7Ga7P45<^^586
2
 7>mmI7
3 :A9I9IPQ9R67>~~7:8
4
 =DNNI:=
9 :A2=:
6  ((s38?8I8I69
5 :A9I9IQSPT9U69@9I9IQRPS9T69@9I9IQRPS9T65<^^9:::6
2  ((s3:A:K:K6;
7 :A9I9IQSPT9U69@9I9IQRPS9T69@9I9IQRPS9T67>~~;:::8
4  ((s3:A:K:K6;
7 :A9I9IQSPT9U69@9I9IQRPS9T6:A:J:JRSQT:U77>~~;::;8
4 7>oo:67
3
 9@8I8I7l9
5 =DNNI:=
9 :A2=:
6 =DNNI:=
9 :A2=:
6 9@:89
5
 9@:89
5
 ;B:K:K9;
7 7>mm97
3 ;B:J:JRTQU:V7:A:J:J&';
7 5<MM7;;5
1
 4;<<5!4
0 4;<<44
0 ;B:J:JRUQV:W729++;43
/ 6=\\36
2 ;B:K:K9;
7 6=\\66
2 3:++963
/ 6=\\66
2 5<KK;65
1 6=^^35^6
2 7>oo6R7
3 8?~~7;8
4 ;B:K:K8|;
7  ((s3:A:K:K6;
7  ((s3:A:K:K8-;
7 ;B:J:JRTQU:V77>~~;;;	8
4 9@;89
5
 ?Fnn9:?
; <C;;4?<
8 )0I<)
%
 &&-<:- 	J 07~~Mz0
, -4KK%'K-
) 8?||-68
4 07~~8*0
, -4KK%'K-
) +2++M-+
' !33+42)
     )r   r   r   to_model_proto)r   r   r   r  r   r  r  r  r  r  r  r  r	  r
  r  r  r  r  s   ````````````````` r  
make_modelr     sT    & XW	eGn W	~1F W	 W	 W	 W	r $$&&r  c                  `   [         R                  R                  SS5      R                  [         R                  5      n [         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  S5      R                  [         R                  5      n[         R                  R                  S5      R                  [         R                  5      n[         R
                  " S[         R                  S9n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n	[         R                  R                  SS5      R                  [         R                  5      n
[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[         R                  R                  SS5      R                  [         R                  5      n[        S0 SU_SU _SU_S	U_S
U_SU_SU_SU_SU_SU	_SU
_SU_SU_SU_SU_SU_SU_6nU$ )N   r   f   r   r   r   r   r   r  r   r  r  r  r  r  r  r  r	  r
  r  r  r   )r)   randomrandastypefloat32zerosr  )r   r   r   r  r   r  r  r  r  r  r  r  r	  r
  r  r  r  models                     r  make_model_with_random_weightsr    s    "$))..r":"A"A"**"M%'YY^^C%<%C%CBJJ%O"-/YY^^B-?-F-Frzz-R*3599>>"3E3L3LRZZ3X0 "!BJJ ?B'..rzz:JB'..rzz:JB'..rzz:JB'..rzz:JA&--bjj9J2&--bjj9JB'..rzz:JB'..rzz:JB'..rzz:JB'..rzz:JA&--bjj9J2&--bjj9J 'E$? 0V 6b	
 #<            !" #E& Lr  c                        \ rS rSrS rS rSrg)_BartEncoderTesti  c                     [        U S5      (       d/  [        5       n[        R                  R	                  U5      nX l        U R
                  $ )N_onnx_model)hasattrr  r'   serdedeserialize_modelr"  )selfmodel_protor  s      r  get_onnx_model_BartEncoderTest.get_onnx_model  s=    t]++8:KHH..{;E$r  c                     [        U S5      (       dF  S[        R                  R                  SSS5      R	                  [        R
                  5      0nXl        U R                  $ )N_ort_inputsr   r   r  )r	   r   )r#  r)   r  randintr  r+   r+  )r&  inputss     r  get_ort_inputs_BartEncoderTest.get_ort_inputs  sQ    t]++RYY..q$@GGQF  &r  )r"  r+  N)__name__
__module____qualname____firstlineno__r(  r.  __static_attributes__r  r  r  r   r     s      r  r   c                      [        5       $ )N)r   r  r  r  bart_encoder_testr6    s    r  )__doc__numpyr)   onnxscript.irr'   
onnxscriptr   onnxscript.onnx_opsetr   onnxscript.onnx_typesr   r   r  r  r   r6  r  r  r  <module>r=     s7       ) .m	'`'T   "r  