Ti[OVSSKJr /SQrSSKrSSKrSSKJrJrJr SSK r SSK r SSK r SSK Jr SSKrSSKJr SrSrS rS r\4S jr\4S jrS rS"SjrSrSrSrSrSrSrSrSr "SS5r!Sr"Sr#Sr$Sr%S#Sjr&S$Sjr'S%Sjr(S&S jr)S'S(S!jjr*g))) annotations)assert_isomorphicassert_isomorphic_graphassert_isomorphic_functionassert_onnx_proto_equalN)Any CollectionSequence)parser)ircN[[U5[U55(deg)z.Assert two graphs or functions are isomorphic.N) _isomorphic_to_function_or_graph)graph_or_function_1graph_or_function_2s Z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/testing/__init__.pyrrs+ 1212   cN[[U5[U55(deg)z!Assert two graphs are isomorphic.N)r_to_graph_proto)graph1graph2s rrr!s v.0G H HH HrcN[[U5[U55(deg)z$Assert two functions are isomorphic.N)r_to_function_proto)fn1fn2s rrr&s! )#.0B30G H HH Hrc X:H$N)xys r_default_equality_opr!+s 6MrcURU5(a3URU5=(a U"[X5[X 55$URU5(+$)zCheck two proto object have same value for optional field. This is restricted to simple field types where == comparison is sufficient. )HasFieldgetattr)fieldobj1obj2equalss r_same_optionalr)/sJ }}U}}U#Zwt/CWTEY(ZZ}}U# ##rcr^[U5[U5:wag[U4Sj[X555$)NFc38># UHupT"X5v M g7frr).0val1val2r(s r !_same_repeated..;sJ4Ijdvd!!4Islenallzip)values1values2r(s `r_same_repeatedr78s, 7|s7|# JC4IJ JJrc*SnU"U5U"U5:H$)z0Compare repeated StringStringEntryProto as maps.cZUVs0sHoRUR_M sn$s snfr)keyvalue)protors rto_map'_same_string_string_map..to_mapAs$(-.1qww...s (r)proto1proto2r=s r_same_string_string_maprA>s/ &>VF^ ++rcURUR:wag[SX5(dgURS5(dURS5(agURURs=:XaURR :XagO Od[ R"U5n[ R"U5n[R"UR5UR5SS9(dg[SX5(dg[SX5(dg[URUR5(dgg)NF data_typesegmentT) equal_nan doc_string data_location) dimsr)r#rG DataLocationDEFAULTr from_protonp array_equalnumpyrA external_data)tp1tp2tensor1tensor2s r _same_tensorrTGs xx388 +s 0 0 ||I#,,y"9"9 C--I1A1A1I1II--$--$~~gmmow}}$O , 1 1 /3 4 4 "3#4#4c6G6G H H rc@[SX5=(a [SX5$)N dim_value dim_param)r))dim1dim2s r _same_dimrZ_s +t 2 ^~kSW7^^rcL[URUR[5$r)r7dimrZ)shape1shape2s r _same_shaper_cs &**fjj) <>% ENN5$9$9) rc[U5[U5:wagUVs0sHo3RU_M nnUH6nURU;a gXERn[XeU5(aM6 g gs snf)NFT)r2rjr)attrs1attrs2ra attrs1mapr~r}s r _same_attrsrsn 6{c&k!$*+FqFI+ ::Y &**%%77  ,sA3c\[U[R5(a UR$U$)z9Return the name of an input/output of a function or graph) isinstanceonnxValueInfoProtorj)rs r_ionamers#4#6#677166>Q>rcF\rSrSrSrS SjrSrSrSrSr Sr S r S r g ) _Matcherz7An isomorphism matcher for two functions or two graphs.cpSnU"U5UlU"U5UlXlX l0UlX0lg)Nc0n[UR5Hup#SU4U[U5'M [UR5H(upE[UR5H upcXF4X'M M* U$)zCompute a map from variables v to their definition-sites. A definition-site (n, i) indicates the i-th output of n-th node The special value (-1, i) is used to indicate the i-th input of a function/graph. ) enumerateinputrnodeoutput)rqresultrrrninxis rdefmap!_Matcher.__init__..defmapsj F!!''*&(!Wwqz"+"166*&qxx0EB!#FI1+Mrdefmap1defmap2fg1fg2 node_mapping outer_scope)selfrrrrs r__init___Matcher.__init__s7 c{ c{ BD&rclUS:XaUS:H$US:XagXR;dX R;aGXR;dX R;d URcgURRX5$URUup4URUupVXF:H=(a UR X55$)zMatch two variables (strings).F)rrr same_value same_node)rvar1var2node1index1node2index2s rr_Matcher.same_values 2:2:  2: || #t<<'? $$,,*>DDTDTD\##..t: :,,t,,,t, BdnnU&BBrc US:XaUS:XagUS:XdUS:XagXR;aURUU:H$URRUnURRUnURUR:wagUR UR :wag[ URURUR5(dgURURUR5(dgX RU'g)zDMatch two node-indices. The special node-index -1 represents inputs.rTF) rrrrop_typedomainr attributesame_sub_graphsame_value_listr)rn1n2rrs rr_Matcher.same_nodes "H28 "H"( "" "$$R(B. . b! b! ==EMM ) <<5<< '5??EOOT=P=PQQ##EKK==!#"rcr^[U5[U5:wag[U4Sj[X555$)z@Match two lists of variables (either a string or ValueInfoProto)Fc3p># UH+upTR[U5[U55v M- g7fr)rr)r,rr rs rr/+_Matcher.same_value_list..s+YGXtq4??71:wqz::GXs36r1)rlist1list2s` rr_Matcher.same_value_lists, u:U #Ys5GXYYYrc:[XU5nUR5$)Match two sub-graphs.)r same_graph)rg1g2sub_graph_matchers rr_Matcher.same_sub_graphs$RT2 ++--rcnURnURn[URUR[5(dgUR (dUR (agUR (dUR (agURURUR5(dgg)rFT) rrr7rrn initializersparse_initializerrr)rrrs rr_Matcher.same_graphst XX XXbhh2BCC >>R^^ B$9$9##BIIryy99rch[URR5[URR5:wag[ URR 5[ URR 5:wagSnU"UR5U"UR5:wag[URR 5[URR 5:wagURURRURR5(dg[UR5[URR 5:wag[[ URR555[URR 5:wagg)z Match (top-level) two functions.FcnURVs0sHoRUR_M sn$s snfr) opset_importrversion)rqentrys rimports'_Matcher.same_function..importss)=>^^L^ELL%--/^L LLs 2T) r2rrrsetrrrrrvalues)rrs r same_function_Matcher.same_functions txx~~ #dhhnn"5 5 txx!! "c$((*<*<&= = M 488  1 1 txx}} TXX]]!3 3##DHHOOTXX__EE t  !S%7 7 s4$$++-. /3txx}}3E ErrN)returnNone) __name__ __module__ __qualname____firstlineno____doc__rrrrrrr__static_attributes__rrrrrs*A'*C 0Z .  %rrc[XS5n[U[R5(a:[U[R5(d [ S5eUR 5$[U[R 5(a:[U[R 5(d [ S5eUR5$[ S5e)zChecks that two function/graph bodies are isomorphic. Assumes that the inputs are valid FunctionProto/GraphProto. Use a separate check to verify that the inputs satisfy FunctionProto/GraphProto requirements (like no duplicate attributes). Nz1Both inputs must be same type (function or graph)z3Inputs must be either a FunctionProto or GraphProto)rrr FunctionProto TypeErrorr GraphProtor)rrmatchers rrr*s s&G#t))**#t1122OP P$$&&#t''#t//OP P!!## I JJrc([U[R5(aU$[U[R5(aUR 5$[U[ 5(a[R"U5$[S[U5S35e)NCannot convert z to FunctionProto) rrr onnxscript OnnxFunctionto_function_protostrr parse_functionrrk)rqs rrr<ss!T''((!Z,,--""$$!S$$Q'' od1gY.?@ AArc<[U[R5(aU$[U[R5(aUR 5R $[U[5(a[R"U5$[S[U5S35e)Nrz to ModelProto) rrrrrto_model_protographrr parse_graphrrk)rxs rrrFsv!T__%%!Z,,--!'''!S!!!$$ od1gYn= >>rcj[U[R5(aU$[U[R5(aU$[U[R5(a UR $[U[ R5(aUR5$[S[U5S35e)Nrz to FunctionProto or GraphProto) rrrr ModelProtorrrrrrk)objs rrrPs#t))** #t'' #t''yy#z..//$$&& od3i[0OP QQrc2URUR4$r)rr)rs r_opset_import_keyr\s   !5!5 66rcUR$r)rj) value_infos r_value_info_keyr`s ??rcJURUR[USS54$)Noverloadr)rrjr$)functions r _function_keyrds OOX]]GHj",M NNrc~[5n/nUH*nX1;aURU5 URU5 M, U$)z5Return a list of duplicated elements in a collection.)rappendadd)with_duplicatesseen duplicatesrs r_find_duplicatesrhs> 5DJ  9   a    rc|[U5[U5LdS[U5S[U535eUR5VVs0sHup4URU_M nnnUR5VVs0sHup4URU_M nnn[[ UR 55[ UR 55-5n[ U[R5(aj[ U[R5(aKURVs1sHoRiM n nURVs1sHoRiM n nO[ 5n [ 5n /n UGHjn[X5n [X5n [ U [5(Ga@[ U [5(Ga*[ U [[45(Gd[ U [[45(Gd/n/nUS:XaT[U [S9n [U [S9n U Vs/sHn[U5PM nnU Vs/sHn[U5PM nnGO>US:XaU(a[ U[R5(ag[ U[R5(aHU Vs/sHnURU ;dMUPM n nU Vs/sHnURU ;dMUPM n n[U [S9n [U [S9n U Vs/sHn[U5PM nnU Vs/sHn[U5PM nnOXUS:XaR[U [ S9n [U [ S9n U Vs/sHn[!U5PM nnU Vs/sHn[!U5PM nnX:waq[ U5[ U5- n[ U5[ U5- nSUSUS US [U 5S [#U5S [#U53 nU R%U5 GM['U 5['U 5:wa=SUS ['U 5S['U 5S[U 53nU R%U5 GM[)['U 55GHonXnXn[ U[*R,R.R05(aA[ U[*R,R.R05(a[3UUUS9 MUU:wdM[ U[65(aM[ U[65(a8[8R:"U5(a[8R:"U5(aMSUSUS[U5S[U53n[<R>"[U5RA5[U5RA55H nUSU-- nM U R%U5 GMr GMp[ U [*R,R.R05(a@[ U [*R,R.R05(a [3XUS9 GMX:wdGM[ U [65(aN[ U [65(a9[8R:"U 5(a[8R:"U 5(aGMNSUSU SU 3nU R%U5 GMm U (a6[5S[U5S[U5S3SRCU 5-5egs snnfs snnfs snfs snfs snfs snfs snfs snfs snfs snfs snfs snf![4aTnSUSUS[U5S[U5SUSU3 n[U5S-U-nU R%U5 SnAGMKSnAff=f)aAssert that two ONNX protos are equal. Equality is defined as having the same fields with the same values. When a field takes the default value, it is considered equal to the field not being set. Sequential fields with name `opset_import`, `value_info`, and `functions` are compared disregarding the order of their elements. Args: actual: The first ONNX proto. expected: The second ONNX proto. ignore_initializer_value_proto: Ignore value protos for initializers if there are extra ones in the actual proto. zType not equal: z != r)r:r functionszField z not equal: keys_only_in_actual=z, keys_only_in_expected=z. Field type: z. Duplicated a_keys: z, duplicated b_keys: z not equal: len(a)=z , len(b)=z Field type: )ignore_initializer_value_protoz index z. in sequence not equal. type(actual_value_i): z, type(expected_value_i): z, actual_value_i: z, expected_value_i: z Caused by the above error N z not equal. field_actual: z, field_expected: zProtos not equal: )"rk ListFieldsrjsortedrkeysrrrrr$r rbytesrrrrrr2rangegoogleprotobufmessageMessagerAssertionErrorfloatmathisnandifflibndiff splitlinesjoin)actualexpectedrr%r;a_fieldsb_fields all_fieldsrractual_initializer_namesexpected_initializer_nameserrorsa_valueb_valuea_keysb_keysrrrkeys_only_in_actualkeys_only_in_expected error_messageactual_value_iexpected_value_ielines rrrssb( <4> ) 4<.T(^,<= )7=6G6G6IJ6Ile E!6IHJ6>6I6I6KL6Kle E!6KHLHMMO,s8==?/CCDJ&$//**z(DOO/T/T4:4F4F#G4FqFF4F #G6>6J6J%K6Jff6J"%K"#&5 %(U"F&((* w ) )7H--we 55we 55!#F "F& .?@ .?@NUVgl+L9gVNUVgl+L9gV,&2"64??;;"8T__== +2*1J%??2JJ#*1+2*1J%??2LL#*1 !o> o>HOP*/*5PHOP*/*5P+% m< m<DKLGy- 2GLDKLGy- 2GL&)&kCK&?#(+F c&k(A%UG#CDWCXXprGqHH##'=/2**:6*B)CCXYijpYqXrt  m,WW-UG#6s7|nIcRYl^\##'=/3 m,s7|,A%,ZN'.z$!&(?(?(G(G$%5v7N7N7V7VWW 93 . 0?](+;;&~u== *+;U C C $ > : : $ +; < <%*0wqcAoptvDqEpFF`aefvaw`x)y $+MM/::< 01<<>%D*TD[8M %  m4C-D!8!8!@!@ A Aj V__,,44G G  $A_  7E**w..JJw''JJw''9'BTU\T]^  MM- (ef fd4>2B" E RXHY Y  KL$H%K.WV QPML> .9.4UG71#EstxzHuItJJdeijze{d||NO]N^^rsCrD-EM #A)L L} \*#MM-88 9sm \9\%\+\0\55\:\?5\?]]] ]]5] ] ^;'A^66^;)rPonnx.TensorProtorQr))rzonnx.OperatorSetIdProtorztuple[str, int])rzonnx.ValueInfoProtorr)rzonnx.FunctionProtorztuple[str, str, str])rzCollection[Any]rz list[Any])F)r%google.protobuf.message.Message | Anyrr*rboolrr)+ __future__r__all__rrtypingrr r google.protobuf.messager rNrLrr rr rrrr!r)r7rArTrZr_rerhrnrrrrrrrrrrrrrrrrr0s#  ,, I I .B$-AK ,0_=_F : ? @@FK$B? R7O,1Z 1Z 3Z %)Z  Z r