
    ϑiN                       S SK Jr  S SKrS SKJr  S SKJrJrJr  S SK	r	S SK
Jr  S SKJr  S SKJrJr  S SKJrJr  S S	KJr  S S
KJr  S SKJrJrJr  S SKJr  S SKJ r   S SK!J"r"J#r#  SSK$J%r%  SSK&J'r'J(r(J)r)J*r*J+r+  SSK,J-r-J.r.  \(       a  S SK/Jr0  \" S5      r1Sr2 " S S5      r3 " S S5      r4 " S S5      r5 " S S\(S9r6S,S jr7S r8S  r9S! r:S" r;S# r< " S$ S%\(S9r= " S& S'\'\(S9r> " S( S)\'\(S9r? " S* S+\ 5      r@g)-    )annotationsN)nullcontext)TYPE_CHECKINGAnyTypeVar)convert_dtype)convert_np_dtype_to_dtype_)UniqueNameGeneratorguard)get_shard_specto_placements)DistributedInputSpec)convert_to_dims_mapping)ALREADY_D2Sextract_tensor_dynamic_dimsgraph_tracing_guard)is_fake_value)	InputSpec)flattenis_sequence   )SymbolicInt)Cache	Singletonget_min_non_specialized_numbermap_if_extendmeta_str)BreakGraphErrorNullMetaBreakDynamicSymbolT___SOT_INFER_META_INNER_VARc                  V    \ rS rSrS	S jr\S
S j5       r\SS j5       rS rSS jr	Sr
g)DistInfo?   Nc                (    Xl         X l        X0l        g Nmeshdims_mappinglocal_shape)selfr(   r)   r*   s       Y/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/jit/sot/infer_meta.py__init__DistInfo.__init__@   s    	(&    c                ^   [        U [        R                  5      (       a  U R                  5       (       d   S[	        U 5       S35       eU R
                  n[        XR                  [        U R                  5      5      n[        X!5      nU R                  5       R                  n[        XU5      $ )NzExpect a Tensor, but got a .)
isinstancepaddleTensoris_disttypeprocess_meshr   
placementslenshaper   _local_valuer#   )tensorr(   sharding_specsr)   r*   s        r,   from_tensorDistInfo.from_tensorE   s    &&--00V^^5E5E 	
)$v,q9	
E ""'##S%6
 /~D))+11K88r/   c                4   [        U [        R                  R                  5      (       a  U R	                  5       (       d   S[        U 5       S35       e[        U R                  5       R                  U R                  5       R                  U R                  5      $ )NzExpect a Value, but got a r1   )r2   r3   pirValuer5   r6   r#   	dist_attrr7   r)   _local_shape)values    r,   
from_valueDistInfo.from_valueS   sx    %!1!122u}} 	
(eQ7	
F OO**OO**
 	
r/   c                    [        [        R                  " U R                  5      [        R                  " U R                  5      [        R                  " U R
                  5      S9$ )Nr'   )r#   copydeepcopyr(   r)   r*   r+   memos     r,   __deepcopy__DistInfo.__deepcopy__^   sC    tyy)t'8'89d&6&67
 	
r/   c                V    SU R                    SU R                   SU R                   S3$ )NzDistInfo(mesh=z, dims_mapping=z, local_shape=)r'   r+   s    r,   __repr__DistInfo.__repr__e   s1    		{/$:K:K9LN[_[k[kZllmnnr/   )r)   r*   r(   )NNN)r<   paddle.Tensorreturnr#   )rE   zpaddle.pir.ValuerU   r#   )rU   str)__name__
__module____qualname____firstlineno__r-   staticmethodr>   rF   rM   rR   __static_attributes__ r/   r,   r#   r#   ?   s:    '
 9 9 
 

or/   r#   c                      \ rS rSrSS jr\S 5       rS rS rS r	      SS jr
S rS	 rS
 r\SS j5       r\SS.     SS jj5       r\SS.     SS jj5       r\SS j5       rS rS rS rSrg)MetaInfoOrNulli   c                    Xl         g r&   metar+   rc   s     r,   r-   MetaInfoOrNull.__init__j   s    	r/   c                     [        S 5      $ r&   r_   r]   r/   r,   nullMetaInfoOrNull.nullm   s    d##r/   c                    U R                   S L $ r&   rb   rQ   s    r,   is_nullMetaInfoOrNull.is_nullq   s    yyD  r/   c                Z    U R                   c  [        [        5       5      eU R                   $ r&   )rc   r   r   rQ   s    r,   unwrap_or_breakgraph#MetaInfoOrNull.unwrap_or_breakgrapht   s#    99!-/22yyr/   c                B    U R                   c   S5       eU R                   $ )NzMetaInfo is Nonerb   rQ   s    r,   unwrap_unsafeMetaInfoOrNull.unwrap_unsafey   s"    yy$8&88$yyr/   c                    U R                   c  [        R                  5       $ U R                   R                  X5      R	                  5       $ r&   )rc   r_   rh   with_dynamic_axeswrap)r+   namedynamic_axess      r,   rt    MetaInfoOrNull.with_dynamic_axes}   s;     99!&&((yy**4>CCEEr/   c                R    U R                   c  g U R                   R                  5       $ r&   )rc   to_input_specrQ   s    r,   rz   MetaInfoOrNull.to_input_spec   s"    99yy&&((r/   c                R    U R                   c  gU R                   R                  5       $ )Nz(Null))rc   	guard_strrQ   s    r,   r}   MetaInfoOrNull.guard_str   s"    99yy""$$r/   c                    U R                   c  [        S 5      $ [        [        R                  " U R                   5      5      $ r&   )rc   r_   rI   rJ   rK   s     r,   rM   MetaInfoOrNull.__deepcopy__   s/    99!$''dmmDII677r/   c                .    [        [        X-   5      5      $ r&   )sortedset)axes1axes2s     r,   mix_axesMetaInfoOrNull.mix_axes   s    c%-())r/   N)rw   c                  U R                  5       (       d  [        R                  5       $ [        U [        R
                  5      (       d   S5       eSU R                  ;  d   S5       e[        U 5      nU=(       d    / n[        R                  U[        U5      5      n[        U R                  5       VVs/ s H  u  p4X1;   a  [        U5      OUPM     nnnU R                  5       (       a  [        R                  U 5      nOS n[        UU R                   U R"                  U R$                  U R&                  U R(                  U R*                  S US9	R-                  5       $ s  snnf )Nz!Expect a Tensor, but got a Value.zITensor shape should not contain -1, maybe you pass a Value to from_tensor	dist_info) _is_dense_tensor_hold_allocationr_   rh   r2   r3   r4   r:   r   r   list	enumerater   r5   r#   r>   MetaInfodtypestop_gradientrv   persistabler6   placeru   )r<   rw   user_specified_dynamic_axesidimr:   r   s          r,   r>   MetaInfoOrNull.from_tensor   sF    6688!&&((&&--00 	
/	
0 % 	
W	
% 'B&&I##)r%..$:;

 $FLL1
1 !" 1Ks:1 	 
 >> ,,V4IILL  KKKKLL

 $&
	
s   4Ec                   [        U R                  5      nU=(       d    / n[        U R                  5       VVs/ s H  u  p4X1;   a
  [	        5       OUPM     nnn[        UUSS S S S S S S9	R                  5       $ s  snnf )NTr   )r	   r   r   r:   r   r   ru   )nparrayrw   r   r   r   r:   s         r,   
from_numpyMetaInfoOrNull.from_numpy   s     +7==9#)r $GMM2
2 .KMC72 	 
 

 $&
		
s   A8c                D   [        U 5      (       a  [        R                  5       $ [        nU R                   Vs/ s H  o"S:X  a
  [        5       OUPM     nnU H.  n[        U[        5      (       d  M  US:  a  M"   SU SU S35       e   [        U [        R                  R                  5      (       a+  U R                  5       (       a  [        R                  U 5      nOS n[        UU R                  U R                   UU R"                  S S S US9	R%                  5       $ s  snf )Nr   r   zKDimensions must be non-negative integers or SymbolicInt. Encountered value z
 in shape r1   r   )r   r_   rh   SOT_INFER_META_INNER_VARr:   r   r2   intr3   rA   rB   r5   r#   rF   r   r   r   r   ru   )rE   rv   r   r:   r   s        r,   rF   MetaInfoOrNull.from_value   s   !&&(('@EL)4LC#s##ax )),ZwaAx  eVZZ--..5==?? ++E2IIKK

 $&
	 Ms   Dc                >    U R                   c  gSU R                    S3$ )NzMetaInfoOrNull(None)zMetaInfoOrNull(rP   rb   rQ   s    r,   rR   MetaInfoOrNull.__repr__   s"    99) 1--r/   c                    U R                   c  UR                   S L $ UR                   c  gU R                   UR                   :H  $ )NFrb   )r+   others     r,   __eq__MetaInfoOrNull.__eq__   s;    99::%%::yyEJJ&&r/   c                \    U R                   c  [        S 5      $ [        U R                   5      $ r&   )rc   hashrQ   s    r,   __hash__MetaInfoOrNull.__hash__   s$    99:DIIr/   rb   )rc   zMetaInfo | None)rv   rV   rw   	list[int]rU   r_   )r   r   r   r   rU   r   )r<   rT   rw   list[int] | NonerU   r_   )r   znpt.NDArray[Any]rw   r   rU   r_   )rU   r_   )rW   rX   rY   rZ   r-   r[   rh   rk   rn   rq   rt   rz   r}   rM   r   r>   r   rF   rR   r   r   r\   r]   r/   r,   r_   r_   i   s     $ $!
FF'0F	F)
%
8
 * * CG##0@#	# #J GK!4D	 *  8.
'r/   r_   c                      \ rS rSr% S\S'     SS jrS r S   SS jjrSS jr\	S	 5       r
S
 rS rSS jrS rS rS rS rS rSrg)r   i  zlist[int | SymbolicInt]r:   Nc
                    SU;  d   S5       eX@l         XPl        X`l        Xpl        Xl        X l        X0l        Xl        Xl        g )Nr   zFNOTE: Shape should not contain -1, consider convert it to SymbolicInt.)	rv   r   r6   r   r:   r   r   r   	spec_name)
r+   r:   r   r   rv   r   r6   r   r   r   s
             r,   r-   MetaInfo.__init__  sK      	
T	
 	&	


*""r/   c                    [        U 5      $ r&   rg   rQ   s    r,   ru   MetaInfo.wrap  s    d##r/   c                t    U R                    Vs/ s H  n[        U[        5      (       a  UOUPM     sn$ s  snf r&   )r:   r2   r   )r+   dynamic_symbolr   s      r,   shape_with_special_symbol"MetaInfo.shape_with_special_symbol!  s>    
 zz
! )k::NC!
 	
 
s   #5c                   [         R                  U R                  U5      n[        U R                  5       VVs/ s H,  u  pEXC;   a   [        U[        5      (       d  [        U5      OUPM.     nnn[        UU R                  U R                  U R                  U R                  U R                  U R                  UU R                  S9	$ s  snnf )N)r   r   )r_   r   rw   r   r:   r2   r   r   r   r   rv   r   r6   r   r   )r+   rv   rw   mixed_dynamic_axesr   r   r:   s          r,   rt   MetaInfo.with_dynamic_axes)  s    +44|
 $DJJ/	
 0	 +JsK4P4P C  	 0 	 	
 JJIIIIJJnn

 
	
	
s   3Cc                    [        U R                  5       VVs/ s H  u  p[        U[        5      (       d  M  UPM      snn$ s  snnf r&   )r   r:   r2   r   )r+   r   r   s      r,   rw   MetaInfo.dynamic_axesE  s>     $DJJ/
/#{+ /
 	
 
s
   AAc                (    U R                   [        :H  $ r&   )rv   r   rQ   s    r,   is_inner_varMetaInfo.is_inner_varM  s    yy444r/   c                2    [        U R                  5      S:  $ )z9
if SymbolicInt in shape, return True
else: return False
r   )r9   rw   rQ   s    r,   is_dynamic_shapeMetaInfo.is_dynamic_shapeP  s    
 4$$%))r/   c           	        U R                  S 5      nU R                  b~  [        U R                  R                  U R                  R                  5      n[        UU R                  U R                  U R                  R                  UU R                  R                  S9$ [        U R                  UU R                  U R                  U R                  S9$ )N)r   r   r(   r8   r*   )r   rv   r   )r   r   r   r)   r(   r   r   r   r*   ConstrainedInputSpecrw   r   )r+   r:   r8   s      r,   rz   MetaInfo.to_input_specW  s    ..t4>>%&++T^^-@-@J (jj"00^^((% NN66  (!!jj^^"00 r/   c                t    U R                  [        5       5      nSU SU R                   SU R                   S3$ )N(z, rP   )r   r   r   r   )r+   r:   s     r,   r}   MetaInfo.guard_strn  s:    ..{}=5'DJJ<r$*<*<)=Q??r/   c                   [        [        U R                  5      U R                  U R                  U R
                  U R                  U R                  U R                  U R                  [        R                  " U R                  5      S9	$ )Nr   )r   r   r:   r   r   rv   r   r6   r   r   rI   rJ   r   rK   s     r,   rM   MetaInfo.__deepcopy__r  s`    JJIIIIJJNNmmDNN3

 
	
r/   c                X    [        U R                  U R                  U R                  5      $ r&   )r   r:   r   r   rQ   s    r,   rR   MetaInfo.__repr__  s    

DJJ0B0BCCr/   c                    U R                   UR                   :H  =(       a9    U R                  UR                  :H  =(       a    U R                  UR                  :H  $ r&   )r:   r   r   rd   s     r,   r   MetaInfo.__eq__  sE    JJ$**$ 9

djj(9""d&8&88	
r/   c                l    [        [        U R                  5      U R                  U R                  45      $ r&   )r   tupler:   r   r   rQ   s    r,   r   MetaInfo.__hash__  s'    U4::&

D4F4FGHHr/   )	r   r   rv   r   r   r:   r   r   r6   )NN)r   )r   r    rU   zlist[int | DynamicSymbolT])rv   rV   rw   r   rU   r   )rU   z+DistributedInputSpec | ConstrainedInputSpec)rW   rX   rY   rZ   __annotations__r-   ru   r   rt   propertyrw   r   r   rz   r}   rM   rR   r   r   r\   r]   r/   r,   r   r     sx    "" #2$ 02
,
	#

8 
 
5*.@
D
Ir/   r   c                  F    \ rS rSrSrS rS
S jrS
S jrSSS jjrS r	Sr
g	)VariableCreatori  z
We use the static graph Variable to infer the meta information of Tensor.
This singleton class is used to create Variable for infer meta.
c                    [        [        5      U l        0 U l        [        R
                  R                  5       U l        [        R
                  R                  5       U l        g r&   )	r
   r   var_name_generator	var_cacher3   staticProgrammain_programstartup_programrQ   s    r,   r-   VariableCreator.__init__  sA    "56N"O"MM113%}}446r/   c                    UR                  5       (       a  gUR                  5       nUR                   SUR                   S3nUSR	                  [        [        UR                  5      5      -  nU$ )Nrh   _)rk   rq   r   r   joinmaprV   r:   )r+   meta_or_nullrc   rv   s       r,   gen_nameVariableCreator.gen_name  sb    !!))+**Qt112!4S$**-..r/   c                   UR                  5       (       a  g UR                  5       nUR                  S5      n[        R                  R                  U R                  U R                  5         [        R                  R                  R                  U R                  UR                  5       5      U[        UR                  5      S9nUR                  Ul        UR                  bg  UR                  R                   n[#        UR                  R$                  U5      n[        R&                  R)                  XEU5      nUR                  Ul        S S S 5        [+        W[        R,                  5      (       a   S5       eU$ ! , (       d  f       N6= f)Nr   )rv   r:   r   z$Expect a Variable, but got a Tensor.)rk   rq   r   r3   r   program_guardr   r   inputdatar   ru   r   r   r   r   r(   r   r)   _pir_opsshard_tensorr2   r4   )r+   r   rc   r:   varr(   r8   s          r,   
create_varVariableCreator.create_var  s4   !!))+..r2]]((t33
 --%%**]]499;/#DJJ/ + C
 !% 2 2C~~)~~***4>>+F+FM
oo223jI$($6$6!
 c6==11 	
2	
1 
%
 
s   ,CE;;
F	c                    U R                  U5      nU(       a  U R                  U5      $ X0R                  ;  a  U R                  U5      U R                  U'   U R                  U   $ r&   )r   r   r   )r+   rc   without_cachevar_feature_names       r,   get_variableVariableCreator.get_variable  sU    ==.??4((>>1/3t/DDNN+,~~.//r/   c           	     &   [         R                  R                  R                  S 5         [	        U R
                  5         U[         R                  R                  L a  [        USS9[        USS9p2O[        U5      [        U5      p2[        5       n[         R                  R                  U R                  U R                  5         [        U[        5      (       a  [!        US   U5      nUSS  n[#        U[$        5      (       a  ['        U R                  5      nU   U" U0 UD6nS S S 5        S S S 5        S S S 5        S S S 5        [)        W5      $ ! , (       d  f       N1= f! , (       d  f       N:= f! , (       d  f       NC= f! , (       d  f       NL= f)NTr   r   r   )r3   base	framework_dygraph_guardUniqueNameGuardr   distributedr   convert_meta_to_variabler   r   r   r   r   r2   rV   getattrhasattrr   r   convert_variable_to_meta_info)r+   funcargskwargsgraph_tracing_context_managerouts         r,   
infer_metaVariableCreator.infer_meta  sA   KK!!006D334v))666,TF,V4H  -T2,V4 
 -8M),,!!4#7#7 dC(( #47D1D8D4--4G))51 3//C 3 5 7: -S11 32  54 76sU   F BE1AE 	E	$E ,E14F
EE  
E.*E11
E?	;F
F)r   r   r   r   N)r   r_   F)rc   r_   )rW   rX   rY   rZ   __doc__r-   r   r   r   r  r\   r]   r/   r,   r   r     s     
7402r/   r   )	metaclassc                (   ^ [        U S U4S jS S9$ )Nc                "    [        U [        5      $ r&   r2   r_   xs    r,   <lambda>*convert_meta_to_variable.<locals>.<lambda>      z!^4r/   c                4   > [        5       R                  U TS9$ )Nr   )r   r   )r
  r   s    r,   r  r    s    /+88] 9 
r/   c                    U $ r&   r]   r	  s    r,   r  r        1r/   predtrue_fnfalse_fnr   )r   r   s    `r,   r   r     s     4
  r/   c                     [        U S S S S9$ )Nc                "    [        U [        5      $ r&   r  r	  s    r,   r  ,convert_meta_to_input_spec.<locals>.<lambda>  r  r/   c                "    U R                  5       $ r&   )rz   r	  s    r,   r  r    s    !//+r/   c                    [        U [        R                  5      (       a)  [        R                  R                  R                  U 5      $ U $ r&   )r2   r3   r4   r   r   r>   r	  s    r,   r  r    s:    !V]]++ MM##//2 r/   r  r  r   s    r,   convert_meta_to_input_specr    s    4+

 
r/   c                     [        U S S S S9$ )Nc                J    [        U [        R                  R                  5      $ r&   )r2   r3   rA   rB   r	  s    r,   r  /convert_variable_to_meta_info.<locals>.<lambda>  s    z!VZZ%5%56r/   c                ,    [         R                  U 5      $ r&   )r_   rF   r	  s    r,   r  r    s    .33A6r/   c                    U $ r&   r]   r	  s    r,   r  r    r  r/   r  r  r  s    r,   r   r     s    66	 r/   c                    [        5       R                  U 5      nU(       a  U" U0 UD6$ [        5       R                  " U /UQ70 UD6$ r&   )SpecialInferMetaget_infermeta_fnr   r  )r   r   r   fns       r,   r  r    sF    			,	,T	2B	4"6""''>t>v>>r/   c           	     ,   [        U [        R                  R                  5      (       d   SU  S35       e[        R                  R                  U SS9n [        X45      u  p4U R                  R                  " U0 UD6u  nnUR                  R                  nUR                  [        R                  R                  [        U5      5       Vs/ s H  n[        U[        5      (       d  M  UPM     sn5      n	U R                  R!                  5         U	$ s  snf )NzExpect a Layer, but got r1   T)
full_graph)r2   r3   nnLayerjit	to_staticr  forwardget_concrete_program_outputsvar_list_restore_oututilsr   r   r_   rollback)
layerr   r   args_kwargs_concrete_programpartial_program_layeroutput_valuesr
  r   s
             r,   infer_meta_for_layerr9    s    eVYY__-- 
"5'+- JJ  4 8E/?NE
 	**E=W= *22;;M

,
, \\))-m<	
 !^,	 	
C 
MMJ	
s   D(Dc           	     (   [        X45      u  p4U R                  " U0 UD6u  nnUR                  [        R                  R                  [        UR                  5      5       Vs/ s H  n[        U[        5      (       d  M  UPM     sn5      nU$ s  snf r&   )
r  r-  r0  r3   r1  r   r   outputsr2   r_   )	static_functionr   r   r4  r5  r6  r7  r
  r   s	            r,   ast_infer_metar=  *  s    /?NE
 	,,e?w?  
,
, \\))-.>.F.FG	
 !^,	 	
C J	
s   #B Bc                  :    \ rS rSrSrS rS r      SS jrSrg)	r#  i?  z
There are some functions that cannot be inferred directly through static graph,
and need to be implemented manually. This class is used to implement infer meta
for these functions.
c                    g r&   r]   rQ   s    r,   r-   SpecialInferMeta.__init__F  s    r/   c                H     UR                   n[        U SU 35      $ !    g = f)N
infermeta_)rW   r   )r+   r%  funcnames      r,   r$  !SpecialInferMeta.get_infermeta_fnI  s1    	{{H4:hZ!899	s    !Nc	                ,    [        U5      (       d  U/nU$ r&   )r   )	r+   r;  inputsgrad_outputsretain_graphcreate_graphonly_inputsallow_unusedno_grad_varss	            r,   infermeta_gradSpecialInferMeta.infermeta_gradQ  s     6""XFr/   r]   )NNFTFN)	rW   rX   rY   rZ   r  r-   r$  rM  r\   r]   r/   r,   r#  r#  ?  s*     r/   r#  c                  4   ^  \ rS rSrU 4S jrS rS rSrU =r$ )InferMetaCacheia  c                    > [         TU ]  SS9  g NT)rI   superr-   r+   	__class__s    r,   r-   InferMetaCache.__init__b      d#r/   c                    U[        [        U5      5      [        UR                  5       5      [        [        U5      5      4$ r&   )r   r   keysr+   r   r   r   s       r,   key_fnInferMetaCache.key_fne  s7     '$- &++- '&/"	
 	
r/   c                     [        U/UQ70 UD6$ r&   )r  r[  s       r,   value_fnInferMetaCache.value_fno  s    $0000r/   r]   	rW   rX   rY   rZ   r-   r\  r_  r\   __classcell__rV  s   @r,   rP  rP  a  s    $
1 1r/   rP  c                  4   ^  \ rS rSrU 4S jrS rS rSrU =r$ )LayerInferMetaCacheis  c                    > [         TU ]  SS9  g rR  rS  rU  s    r,   r-   LayerInferMetaCache.__init__t  rX  r/   c           	        UR                  SS9 Vs/ s H  n[        R                  U5      PM     nnU[        U5      [        [	        U5      5      [        UR                  5       5      [        [	        U5      5      4$ s  snf )NT)include_sublayers)
parametersr_   r>   r   r   rZ  )r+   r3  r   r   r
  paramss         r,   r\  LayerInferMetaCache.key_fnw  s~     %%%=
= &&q)= 	 

 &M'$- &++- '&/"
 	
	
s   A>c                     [        U/UQ70 UD6$ r&   )r9  )r+   r3  r   r   s       r,   r_  LayerInferMetaCache.value_fn  s    #E;D;F;;r/   r]   ra  rc  s   @r,   re  re  s  s    $
< <r/   re  c                  ,   ^  \ rS rSrSU 4S jjrSrU =r$ )r   i  c                   > / U l         [        TU ]  " U0 UD6  [        5       nU H   nU R                   R	                  XTS 45        M"     g r&   )rangesrT  r-   r   append)r+   rw   r   r   min_non_specialized_numberr   rV  s         r,   r-   ConstrainedInputSpec.__init__  sM      	 	$)&)%C%E"AKKtDE r/   )rq  )rw   r   )rW   rX   rY   rZ   r-   r\   rb  rc  s   @r,   r   r     s    F Fr/   r   r  )A
__future__r   rI   
contextlibr   typingr   r   r   r3   paddle.base.data_feederr   paddle.base.frameworkr	   paddle.base.unique_namer
   r   r   /paddle.distributed.auto_parallel.placement_typer   r   7paddle.distributed.auto_parallel.static.dist_input_specr   -paddle.distributed.auto_parallel.static.utilsr   paddle.jit.dy2static.utilsr   r   r   
paddle.pirr   paddle.staticr   paddle.utilsr   r   symbolic_shape.symbolic_valuer   r1  r   r   r   r   r   utils.exceptionsr   r   numpy.typingnptr    r   r#   r_   r   r   r   r  r   r  r9  r=  r#  rP  re  r   r]   r/   r,   <module>r     s    #  " . .  1 < 
 % # - 6  =)*8 'o 'oTV VrHI HIVU2	 U2p?8* D1Ui 1$<%9 <*F9 Fr/   