TiMSSKJr SSKrSSKrSSKrSSKrSSKrSSKrSSKJrJ r J r J r J r J r JrJr SSKrSSKrSSKrSSKJr SSKJrJrJr SSKJr SSKJr SSKJr \"S 5r \"S 5r!S S!S jjr"\RFRHRJRL\'\RFRHRJRP\)\RFRHRJRT\+\RFRHRJRXS\RFRHRJRZS\RFRHRJR\S\RFRHRJR^S\RFRHRJR`\\'\RFRHRJRb\\)\RFRHRJRd\\+\RFRHRJRfS\RFRHRJRhS\RFRHRJRjS\RFRHRJRlS0r7\8"5r9\Rt"\;5r<"S S 5r=S"Sjr>S#Sjr?\R"SS\ 55rA"SS\A5rB\R"SSS9"SS55rD"SS\B\ \!\ 45rE"SS\B5rF"SS5rG"SS\G5rHg)$) annotationsN)AnyCallableClassVarGenericOptionalProtocolSequenceTypeVar) ParamSpec) ast_utils irbuilder sourceinfo) converter)_schemas)ONNXType_R_PcUS:XaSnX4[RR;a8[S[RRR 555$[RRX4n[US5$)z=Selects a suitable ONNX ir_version for a given opset version.ai.onnxc3@# UHupUSS:XdMUv M g7f)rrN).0kvs Z/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/_internal/values.py $select_ir_version..,s# C$!qtyGXAACs  )onnxhelperOP_SET_ID_VERSION_MAPmaxitems)versiondomainrequired_min_versions rselect_ir_versionr)'sw | A AA ++;;AAC    ;;<XU4nURRU5nU(aU$[TU] U5nXlX%l0UlXPRU'U$N)r0getsuper__new__r'r& function_defs)clsr'r&keyexistinginstance __class__s rr5 Opset.__new__YsXG$99==% O7?3' "!#! #r*cgr2r)selfr'r&s r__init__Opset.__init__es r*clURRSUR<SUR<S3$)N(z, ))r;__name__r'r&r>s r__repr__Opset.__repr__is/..))*!DKK?"T\\opnameschemas r __getitem__Opset.__getitem__lsD YY))&,, LFdF+ +  s?A AAc[RRXRUR5 g![ a gf=f)NTF)r!rIrJr&r'rL)r>rMs r __contains__Opset.__contains__ss<  II t{{ C  s47 AAcUR$r2)r'rEs r__str__ Opset.__str__{s {{r*c[RRXRUR5n[ XU5$![ an[SUS35UeSnAff=f)Nz Attribute z not found.)r!rIrJr&r'rKrLAttributeError)r>attrrNexcs r __getattr__Opset.__getattr__~s[ JYY))$ dkkJFd&) ) J :dV;!?@c I Js?A A! AA!c4XRUR'gr2)r6name)r>funs radd_function_defOpset.add_function_defs'*388$r*ct[U5nU(a%UScUR5 U(aUScMU$)zTrims 'None' values from the end of the inputs list. This is used to support omitting optional inputs when no more required inputs follow to prepare a valid call against the Op. Used by the static opset code generator. )listpop)r>_inputs input_lists r_prepare_inputsOpset._prepare_inputss7 &\ Z^3 NN Z^3r*r)r'r.r&r/)NN)r'z Optional[str]r&z Optional[int]returnr.)rYr.rlrK)rfzonnx.defs.OpSchema)rD __module__ __qualname____firstlineno____doc____annotations__r0r5r?rFrOrRrUr[r`ri__static_attributes__ __classcell__r;s@rr,r,JsS K L:Ar*rNrkrlr,rlzOptional[_schemas.OpSignature]) rDrmrnrorppropertyr^rrrrrr*rrrs/<  !! AAr*rc\rSrSrSrS S SjjrSr\SSj5r\SSj5r \SSj5r \SS j5r \ RSS j5r S r g)rKaRepresents an ONNX op instance (for example, the MatMul op from ONNX opset version 13). It belongs to a particular Opset and has a name. Attributes: opset: The Opset that this op belongs to. name: The name of the op. op_schema: The ONNX OpSchema for the op. NcXlX lU UbX0lO X=nbURUlOSUlSUlURc[ R SUU5 gg)NzRAn OpSchema was not provided for Op '%s' and there is not one found in opset '%s'.)_opset_name _op_schema op_schema _signatureloggerdebug)r>rr^rops rr? Op.__init__sh    'OKb , llDO"DO:> ?? " LL8   #r*cSSKJn UR5n[US5(aBURc[ SUR S35eURURX5$UR5RXU5$)Nr evaluatorevalzOpSchema not found for op 'z'.) onnxscript._internalrdefaulthasattrr ValueErrorr^reval_op)r>argskwargsrdefault_evaluators r__call__ Op.__call__sx2%--/ $f - -~~% #>tyyk!LMM$))$..$G G  "**4v>>r*cUR$r2)rrEs rr^Op.names zzr*cUR$r2)rrEs rrOp.opsets {{r*cUR$r2)rrEs rr Op.op_schemas r*cURb UR$URcg[RR UR5UlUR$)!Returns the signature of this op.N)rrr OpSignaturefrom_op_schemarEs rrOp.op_signaturesK ?? &?? " >> !"..==dnnMr*cXlgr2rr>values rrrr*)rrrrr2)rr,r^r.rOptional[onnx.defs.OpSchema]rlNonerkr)rlrrrz_schemas.OpSignature)rDrmrnrorpr?rrr^rrrsetterrrrr*rrKrKsRV"%2N , ?    r*rKF)repreqc8\rSrSr%SrS\S'S\S'S\S'S rg ) OnnxClosurezRRepresents a nested function used as a graph-valued attribute for an ONNX op call.irbuilder.IRFunction function_irztypes.FrameTypeframerfunctionrN)rDrmrnrorprqrrrr*rrrs\%% Mr*rc^\rSrSrSrS U4Sjjr\S Sj5r\RSSj5rSSjr SSjr SSjr S r SSS jjr S rU=r$) OnnxFunctioni aRepresents an ONNX op for which a function-body has been defined in onnxscript. Attributes: opset: Opset the function belongs to. name: Name of the function. function: Python function. function_ir: Python code parsed as an :class:`irbuilder.IRFunction`. source: Source code used to generate the function. kwargs: Additional properties used to construct a ModelProto. op_schema: Generated ONNX OpSchema for this op. c>U=(d [URS5n[TU] XR5 X lX0lURUR RS'X@l XPl [RRUR UR RURS9Ul[R "X5 SUlg)a5Constructs an OnnxFunction. Args: opset: opset the function belongs to pyfun: python function irfun: python code parsed by class :class:`onnxscript.converter.Converter` source: source code used to generate the function kwargs: additional properties used to construct a ModelProto  opset_versionr'r^FN)r,r'r4r?r^rrr&rsourcerrr from_functionr functoolsupdate_wrapper traceable)r>rpyfunirfunrrr;s rr?OnnxFunction.__init__s$/u||Q/  +  16o.  "..<< MM$"2"2"9"9 =    -r*cUR$rrrEs rrOnnxFunction.op_signature:r*cXlgr2rrs rrr?rr*cNSSKJn UR5RXU5$)z=Implements an eager-mode execution of an onnxscript function.rr)rrr eval_function)r>rrrs rrOnnxFunction.__call__Cs!2  "00VDDr*cPURRSUR<S3$NrBrC)r;rDrrEs rrFOnnxFunction.__repr__Is%..))*!DMM+>r*c6URR5$)z7Converts the function into :class:`onnx.FunctionProto`.)rto_function_protorEs rrOnnxFunction.to_function_protoLs1133r*c URR(a6[SURR55(a [S5e0UREUEnUR "S0UD6$)z4Converts the function into :class:`onnx.ModelProto`.c3<# UHoRSLv M g7fr2)r)rrYs rr.OnnxFunction.to_model_proto..Rs* +A4JJ$ +AszBA function with required attributes cannot be exported as a model.r)rattrsanyrr_to_model_proto)r>rmerged_kw_argss rto_model_protoOnnxFunction.to_model_protoPsq    ! !c* +/+;+;+A+A* ' ' T 3DKK262##5n55r*c URR5nUR5Vs/sHoRPM n nURRR 5n U R R 5n U HanURn U b#X;aURRSS5X'SU ;dM=SUR ;dMOUR SU S'Mc SU ;a&UbUO[RR5U S'SU;aURS5n O[U S5n [R "XS9nU HnXR"UR%5'M! [R&"U5nUR(SS2 UR(R+U R-5V Vs/sHun n[R."U US9PM snn 5 URnU(akUR0R+UR-5VVs/sH4unn[R2R5UUR755PM6 snn5 UbUR8HDnUR;S5(aMUR<R?UR755 MF UR@HDnUR;S5(aMUR<R?UR755 MF UbI[CUR8U5H/unnUR<R?UR755 M1 UbI[CUR@U5H/unnUR<R?UR755 M1 UR-5Hunn[EUUU5 M U$s snfs snn fs snnf) aCConverts this instance into a `onnx.ModelProto`. Args: io_types: When specified, all the inputs/outputs of the model are set to be of this type. input_types: When specified, all the inputs of the model are set to be of the corresponding type in this list. output_types: When specified, all the outputs of the model are set to be of the corresponding type in this list. value_infos: A dictionary mapping intermediate variable names to ONNX types. Used to set value_info for intermediate variables. opset_version: The standard opset version to use for the model if it cannot be inferred. Otherwise defaults to the current opset version. kwargs: Additional parameters given to function :func:`onnx.helper.make_model`. Returns: An instance of :class:`onnx.ModelProto`. Nrrr ir_version)r)r'r&rv)#rget_called_functionsvaluesgraphclone opset_importscopyr'rr3r!rIonnx_opset_versionrer)irModel functions identifierto_proto opset_importextendr%OperatorSetIdProto value_infor"make_value_info to_type_protoinputHasFieldrvCopyFromoutputzipsetattr)r>io_types input_types output_types value_infosrr sub_functionsfunc ir_functions main_graphrr'rmodel model_protor&rr^rvrrrrs rrOnnxFunction._to_model_protoasK8((==? HUH\H\H^*_H^+;+;H^ *_%%++113 "00557  D[[F!f&A(, oq(I %&21C1C+C$($6$6r$: b! ! ] ""/!:  @\@\@^ "  6 !L1J*=+<=J; D15OODOO- .!kk%(   $ $Q '  ''(5':':'< '(>(@A% ,,v..KK(()?)?)AB'  ""5;; < t ##D$6$6$89 =  # #ELL, ?  $$T%7%7%9:!@LLNDAq KA &#Q+`L sO"O &;O )rrrrrr) rzOptional[Opset]rrrrrr.rzdict[str, Any]rr)rz_P.argsrz _P.kwargsrlrrk)rlzonnx.FunctionProto)NNNNN) rzOptional[ONNXType]rOptional[Sequence[ONNXType]]rrr zdict[str, ONNXType] | Nonerz int | Nonerlzonnx.ModelProto)rDrmrnrorpr?rrrrrFrrrrrrsrts@rrr s """$ "  "  "H  E ?46&(,485926$( e$e2e3 e 0 e " e eer*rc^\rSrSrSrS U4SjjrSrSr\S Sj5r \S Sj5r \ RS Sj5r S r U=r $)TracedOnnxFunctionizaTracedOnnxFunction. Attributes: name: Name of the op. E.g. "aten::add". func: Function. c>[TU]XR5 X l[R R URSURS9Ul[R"X5 g)N_tracedr) r4r?rDr rrrr^rrr)r>rr r;s rr?TracedOnnxFunction.__init__sU  . "..<< IIidii=    ,r*c&UR"U0UD6$r2)r )r>rrs rrTracedOnnxFunction.__call__syy$)&))r*cPURRSUR<S3$r)r;rDr rEs rrFTracedOnnxFunction.__repr__s$..))*!DII=::r*c[R"UR5up[R"UR5n[R "UR5nUR R5nURUR5 [R"URUUS9nURU5$)zWReturn the function_ir. This function IR contains only the signature of the function. )r global_namesr)r get_src_and_astr inspect getmodulegetclosurevars__dict__rupdate nonlocalsconverter_module Converterrtranslate_function_signature)r>srcfunc_astmoduleclosurerrs rrTracedOnnxFunction.function_irs "11$))< ""499-((3++- G--.$..++% 55h??r*cUR$rrrEs rrTracedOnnxFunction.op_signaturerr*cXlgr2rrs rrr-rr*)rr )rr,r r)rlrrr)rDrmrnrorpr?rrFrrrrrrrsrts@rrrs[-*;@@$  r*rc"\rSrSrSrSSjrSrg) SymbolValueiaRepresents script-time value information about named variables used in a script. At translation-time, the (local) variables of a script, including its parameters, are bound to a SymbolValue. SymbolValues fall into the following categories: AttrRef: Function parameters of attribute-kind, also mapped to ONNX attributes Dynamic: values computed at runtime (of tensor type, for now) mapped to NodeArgs. Dynamic values include input-parameters of the script, as well intermediate values computed in the script. For example, consider the following script definition: :: @script() def ThresholdedRelu(X, alpha: float): zero = op.CastLike(0, X) return op.Where(X > alpha, X, zero) Here, `X` has a Dynamic value, `alpha` has an AttrRef value, and `zero` has a Dynamic value. Scripts may also contain references to global variables, but the translator does not associate a SymbolValue with them. The python value of global variables is used directly in the translation, and such global variables are intended to be used for limited purposes, namely: * To identify an opset * To represent constant-values, translated into ONNX constants. c[U[R5(d[S[ U5<S35eXlX lg)z Initializes SymbolValue. Arguments: value: The value bound to a python variable in a script. info: source-location information for error-messages/debugging z/info must be of type sourceinfo.SourceInfo not .N) isinstancer SourceInfo TypeErrorrvrinfo)r>rr6s rr?SymbolValue.__init__s<$ 5 566MdSWj^[\]^ ^  r*)r6rN)rrr6sourceinfo.SourceInforlr)rDrmrnrorpr?rrrr*rr0r0s @ r*r0c,^\rSrSrSU4SjjrSrU=r$)AttrRefi-c0>[TU]X5 X lg)zInitializes AttrRef. Arguments: attr: An ir.Attr representing the attribute-parameter as_bool: Whether the attribute is to be interpreted as a bool type (represented as int in ONNX) info: for debugging use. N)r4r?as_bool)r>rYr<r6r;s rr?AttrRef.__init__.s $ r*)r<)rYzir.Attrr<boolr6r8rlr)rDrmrnror?rrrsrts@rr:r:-s   r*r:)r)r&r/r'r.rlr/)r{zonnx.AttributeProtorlr)rzir.Valuerlr)I __future__r dataclassesrrloggingtypestypingrrrrrr r r r! onnx.defsonnx_irrtyping_extensionsr rr rrrr$ onnxscript.irronnxscript.onnx_typesrrrr)rIOpSchemaAttrTypeFLOATfloatINTr/STRINGr.TENSORGRAPH SPARSE_TENSOR TYPE_PROTOFLOATSINTSSTRINGSTENSORSGRAPHSSPARSE_TENSORS TYPE_PROTOS_ATTRIBUTE_TYPE_TO_PYTHON_TYPEobjectry getLoggerrDrr,r|rruntime_checkablerrK dataclassrrrr0r:rr*rr_s #     'AA>"* T]t_ ) II%%uII##SII&&II&&II%%tII--tII**DII&&II$$hsmII''#II''II&&II..II++T"$   8 $GGZ7$ BX B BG G T Ee,  - }2wr2v}@2 2 j,,^ k r*