a dr(@sddlZddlmZddlZddlZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do|d S)N___) startswith)namer`/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/jit/supported_ops.py_hidden sr cCst|SN)str)typerrr _emit_type sr cCsNd|jt|j}|j}|dur2d|t|}|dkrJdd||}|S)Nz{} : {}z{}={}rz {}{} )formatrr r default_valuer )indentiargvdefaultrrr _emit_argsrcsdfddt|DS)N,c3s|]\}}t||VqdSr )r).0rrrrr z_emit_args..)join enumerate)r argumentsrrr _emit_argssrcCs t|jSr )r r )retrrr _emit_retsr!cCs2t|dkrt|dSdddd|DS)Nrz Tuple[{}]z, css|]}t|VqdSr )r!)rrrrrr!rz_emit_rets..)lenr!rr)returnsrrr _emit_retss  r&cCsN|dur|}n d||}d|tt|d||j|dt|j}|S)Nz{}.{}z {}({}) -> {}r")rrr$rr&r%)modrschema arg_startpaddingZqualified_name schema_strrrr _emit_schema#s r-c Csddd}g}ttjD]D}t|stjd|}|D]"}||r6|td||ddq6qd|fS)NcSsFt|jdkrdS|jd}|jdkr*dS|jtjjsBdSdS)NrFselfT) r$rrr Z isSubtypeOftorch_CZ TensorTypeget)r)r.rrris_tensor_method.s  z)_get_tensor_ops..is_tensor_methodzaten::Tensorr")r*zSupported Tensor Methods)dirr/r3r r0_jit_get_schemas_for_operatorappendr-)r2methodselemschemasr)rrr_get_tensor_ops-s r:c Csg}tjj}|j}ttjjD]}t||}t|rt|drDqt |}|sbt d|dd|jvrnqz(tj |}|j }|t|||WqYq0qtj jjD]^}|j}t|D]J}tt||}|durtj|} | D]}t|s|t|||qqqd|fS)Nr Module for not foundztorch.nn.functionalzSupported PyTorch Functions)r/nnZ functional__name__r4getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr)r6r- _builtinsZ_modules_containing_builtinsrr0r5) functionsr(rr8attrZ attr_moduleZscriptedr)builtinr9rrr_get_nn_functional_opsCs8      rJcCsrg}tjjjD]^\}}t|}t|ds,q|s2qt|jst|j st|jrRqd|jvr^q| ||fq|S)Nr>ztorch._C) r/rDrFZ _builtin_opsr@rBhasattrr r> __qualname__r6)builtinsfn _builtin_namer(rrr_get_builtins_helperms   rPcCs(t|}|std|d|jdkS)Nr;r<math)r@rBrCr>)rNr(rrr _is_math_fns rRc Csg}tddt}t|}|D]b\}}t|}|sFtd|dt|}|dur tj |}|D]}| t |j |j |qfq d|fS)NcSst|d SNrrRrNrrrrz+_get_torchscript_builtins..r;r<zTorchScript Builtin Functions) filterrPlistr@rBrCrr/r0r5r6r-r>) rGrM builtins_listrNrOr(rIr9r)rrr_get_torchscript_builtinss   rZc Csg}tddt}t|}|D]p\}}t|}|sFtd|dt|}|dur tj |}|D](}t |j |j |} d| vrqf| |qfq d|fS)NcSs t|dSrSrTrUrrrrVrz$_get_math_builtins..r;r<r3z``math`` Module) rWrPrXr@rBrCrr/r0r5r-r>r6) rGrMrYrNrOr(rIr9r)r,rrr_get_math_builtinss"    r[c CsBgd}dddddddd d }d d d d ddddd}gd}g}|D]\}}|d||qDg}g}|D]|}d|} ||vr||} tj| } | D]} |td|| ddqt| dkr|dqld|||} || qld|} d|}d|}t | d} t |d}t |d}d||| }d|fS)N)printtuplefloatcomplexintboolr r?rK isinstancer$hexoctroundhashminmaxabsalldivmodrXordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rar`r^r_rirhrgrozPrint any valuez]Lists cannot be converted to tuples with this method since their size is not statically knownz'Attribute name must be a literal stringzResult is staticzMArguments must be iterable. See :ref:`Iterables ` for details.z-Can only be used as an iterator in a for loop)r\r]r?rKrbrprro))r_ __complex__)r^ __float__)r`__int__)ra__bool__)r __str__)r$__len__)rcZ__hex__)rdZ__oct__z"{}", "``{}``"zaten::{}r)r+z":any:`{}`", "{}"  a The functions in the following table are supported but do not have a static schema .. csv-table:: :header: "Function", "Note" {} The following functions will use the corresponding magic method on :any:`TorchScript classes` .. csv-table:: :header: "Function", "Magic Method" {} These built-in functions use the schema .. rst-class:: codeblock-height-limiter :: {} zPython Built-in Functions) r6rr/r0r5r-r$rtextwrapr)Zsupported_builtinsZ op_renamesZschemaless_op_explanationsZ magic_methodsZmagic_methods_rowsrNZ magic_methodZschematized_opsZschemaless_opsZop_namer9sZ table_rowZschematized_ops_strZschemaless_ops_strZmagic_methods_rows_strsectionrrr_get_global_builtinss\                r~cCsdd}d}tttttf}|D]}|\}}|dddddd}t|trnd |dt ||}nd |dt |||}d |d |}||7}q|S) NcSsdddd|DS)Nz1 .. rst-class:: codeblock-height-limiter :: {} rxcss|]}d|VqdS)z {} N)r)rdrrrr.rz:_list_supported_ops..emit_block..)rr)Zdeclsrrr emit_block-sz'_list_supported_ops..emit_blockrx`-rz {} {} {} ~z{} {} {}z.. _{}:z ) r:rJrZr~r[replacelowerrbr rr$)rbodyZop_gathering_fnsrNheaderitemsZ link_targetr}rrr_list_supported_ops,s"    r)rr')Z torch.jitr/Ztorch.jit._builtinsrr@r{r r rrr!r&r-r:rJrPrRrZr[r~r__doc__rrrrs&   *w