Ti!SSKJr SSKrSSKJrJr SSKrSSKJ r J r SSjr SSSjjr SSjr SSjrSSS jjrSSS .SS jjrSS jrSS jrSSjrSSjrSSjrg)) annotationsN)CallableSequence)ir optimizercU(aQUSRnU(a!UHnX ;dM UR5 M gUHnUR5 M gg)z>Display a list of nodes in the order they appear in the graph.rN)graphdisplay)nodesr nodes \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/rewriter/_ir_utils.py display_nodesr sI a =LLN  c^^^^/mSUUUU4Sjjm[U[R5(a T"US5 OG[U[R5(a(UR 5bT"UR 5S5 [ T5 g)z[Display the (backward or forward) subgraph from a given value or node upto a certain depth.cL>UT;agTRU5 UT:aT(aFURH5nUcMUR5cMT "UR5US-5 M7 gURH'nUR 5HupET "XAS-5 M M) gg)N)appendinputsproduceroutputsuses) r depthinpoutconsumer_backward depth_limitslicevisits r r display_slice..visit s 5=  T ; ;;C3<<>+Ecllneai8' <> D ,,dV4   rcjUcg[U5nUbUR5$g![a gf=f)aConvenience wrapper to get (optional) numpy value from an optional IR Value. This is intended for use in optimizations/rewriting. Note that this does not yet handle the distinction between inputs with default values (values that are both graph inputs and graph initializers), which should not be treated as a constant, and true constant values. The caller should make the distinction, as a value does not contain enough information to determine this. (TODO) N)r,numpyFileNotFoundError)valr*s r get_numpy_valuer1<sI {!#&K $$& & !  s % 22c[U5nUbqURS:XaaUR5nUb%[U[5(aUR U:XaU$[U[ 5(aUR U;aU$g)atReturns element of a single element tensor constant value, and None otherwise. If an int rank is specified, it checks that the value has the given rank. If the rank is a sequence of ints, it checks that the value has one of the given ranks. Thus, `rank=0` checks for a scalar, `rank=1` checks for a 1D tensor, and `rank=(0,1)` checks for either a scalar or a 1D tensor. Nr)r1sizeitemr"intndimr)r0ranknp_valr+s r get_singleton_valuer9QskS !F fkkQ.  Ljs339LL dH % %6;;$+>L r)rtolr7c[XS9nUcg[U5(aU"U5$[U[5(aX:H$Uce[R "XAUS9$)z[Returns True if the value is a single element tensor with given value, and False otherwise.)r7F)rel_tol)r9callabler"r5mathisclose)r0expectedr:r7scalars r is_singleton_valuerBds_! 0F ~(C  !!    <<$ 77rc<Ucg[UR[R5(dg[ U5nUcgUR [ U5:wd2URR[RR:wagUR5nX1:H$)zUReturns True if the value is a 1d int64 tensor with given value, and False otherwise.F) r"typer TypeProtocolr1r3lendtypeDataTypeINT64tolist)r0r@r8valuess r is_1d_valuerLxst { chh 0 0 S !F ~ s8}$#((..BKKtypingrrr.np onnxscriptrrrr&r,r1r9rBrLrPrSrXr]rr rks# %$ 2*.'+ 8 8$8  8 % 8  8( :   $r