o Zh(@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do |d S)N___) startswith)namerF/var/www/auris/lib/python3.10/site-packages/torch/jit/supported_ops.py_hiddensr cCst|SN)str)typerrr _emit_typesr cCsT|jdt|j}|j}|dur|dt|}|dkr(dd||}|S)Nz : =r  )rr r default_valuer )indentiargvdefaultrrr _emit_argsrcsdfddt|DS)N,c3s |] \}}t||VqdSr )r).0rrrrr !sz_emit_args..)join enumerate)r argumentsrrr _emit_args srcCs t|jSr )r r )retrrr _emit_ret$s r!cCs4t|dkr t|dSdddd|DdS)NrzTuple[z, css|]}t|VqdSr )r!)rrrrrr+sz_emit_rets..])lenr!r)returnsrrr _emit_rets(s  r'cCsV|dur|}n|d|}|dtt|d||j|ddt|j}|S)N.(r"z) -> )rr%rr'r&)modrschema arg_startpaddingZqualified_name schema_strrrr _emit_schema.sr0c Csddd}g}ttjD]"}t|s-tjd|}|D]}||r,|td||ddqq d|fS)NcSsFt|jdkr dS|jd}|jdkrdS|jtjjs!dSdS)NrFselfT) r%rrr Z isSubtypeOftorch_CZ TensorTypeget)r,r1rrris_tensor_method<s  z)_get_tensor_ops..is_tensor_methodaten::Tensorr")r-zSupported Tensor Methods)dirr2r7r r3_jit_get_schemas_for_operatorappendr0)r5methodselemschemasr,rrr_get_tensor_ops;s r>c Csg}tjj}|j}ttjjD]A}t||}t|r!t|dr"qt |}|s1t d|dd|jvr7qztj |}|j }|t|||WqYqtj jjD]/}|j}t|D]%}tt||}|durtj|} | D]} t|s|t||| qtq_qVd|fS)Nr Module for not foundztorch.nn.functionalzSupported PyTorch Functions)r2nnZ functional__name__r8getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr,r:r0 _builtinsZ_modules_containing_builtinsrr3r9) functionsr+rr<attrZ attr_moduleZscriptedZscripted_schemabuiltinr=r,rrr_get_nn_functional_opsRs<      rNcCsrg}tjjjD]/\}}t|}t|dsq|sqt|js(t|j s(t|jr)qd|jvr/q| ||fq|S)NrBztorch._C) r2rHrJZ _builtin_opsrDrFhasattrr rB __qualname__r:)builtinsfn _builtin_namer+rrr_get_builtins_helper}s   rTcCs(t|}|std|d|jdkS)Nr?r@math)rDrFrGrB)rRr+rrr _is_math_fns  rVc Csg}tddt}t|}|D]1\}}t|}|s#td|dt|}|durAtj |}|D] }| t |j |j |q3qd|fS)NcSst|d SNrrVrRrrrsz+_get_torchscript_builtins..r?r@zTorchScript Builtin Functions) filterrTlistrDrFrGrr2r3r9r:r0rB) rKrQ builtins_listrRrSr+rMr=r,rrr_get_torchscript_builtinss   r^c Csg}tddt}t|}|D]8\}}t|}|s#td|dt|}|durHtj |}|D]}t |j |j |} d| vrBq3| |q3qd|fS)NcSs t|dSrWrXrYrrrrZs z$_get_math_builtins..r?r@r7z``math`` Module) r[rTr\rDrFrGrr2r3r9r0rBr:) rKrQr]rRrSr+rMr=r,r/rrr_get_math_builtinss"    r_c CsXgd}dddddddd d }d d d d ddddd}gd}g}|D]\}}|d|d|dq"g}g}|D]A}d|} ||vrH||} tj| } | D] } |td|| ddqPt| dkrj|dq9d|d||d} || q9d|} d|}d|}t| d} t|d}t|d}d|d|d | d!}d"|fS)#N)printtuplefloatcomplexintboolr rCrO isinstancer%hexoctroundhashminmaxabsalldivmodr\ordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rerdrbrcrmrlrkrszPrint 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`rarCrOrfrtrrs))rc __complex__)rb __float__)rd__int__)re__bool__)r __str__)r%__len__)rgZ__hex__)rhZ__oct__"z", "``z``"r6r)r.z":external+python:py:obj:`z`", "r z The functions in the following table are supported but do not have a static schema .. csv-table:: :header: "Function", "Note" z The following functions will use the corresponding magic method on :any:`TorchScript classes` .. csv-table:: :header: "Function", "Magic Method" zX These built-in functions use the schema .. rst-class:: codeblock-height-limiter :: z zPython Built-in Functions) r:r2r3r9r0r%rtextwrapr)Zsupported_builtinsZ op_renamesZschemaless_op_explanationsZ magic_methodsZmagic_methods_rowsrRZ magic_methodZschematized_opsZschemaless_opsZop_namer=sZ table_rowZschematized_ops_strZschemaless_ops_strZmagic_methods_rows_strsectionrrr_get_global_builtinssf                rcCsdd}d}tttttf}|D]J}|\}}|dddddd}t|tr;|ddt |d|d}n|ddt |d||}d |d d |}||7}q|S) NcSsdddd|DS)Nz1 .. rst-class:: codeblock-height-limiter :: {} r}css|] }d|dVqdS)z  Nr)rdrrrr@sz:_list_supported_ops..emit_block..)formatr)Zdeclsrrr emit_block>sz'_list_supported_ops..emit_blockr}`-rr~z.. _:r) r>rNr^rr_replacelowerrfr r%)rbodyZop_gathering_fnsrRheaderitemsZ link_targetrrrr_list_supported_ops=s"      r)rr()rDrZ torch.jitr2Ztorch.jit._builtinsrr r rrr!r'r0r>rNrTrVr^r_rr__doc__rrrrs(    +y