a h(@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do|d S)N___) startswith)namerE/var/www/auris/lib/python3.9/site-packages/torch/jit/supported_ops.py_hiddensr cCst|SN)str)typerrr _emit_typesr cCsT|jdt|j}|j}|dur6|dt|}|dkrPdd||}|S)Nz : =r  )rr r default_valuer )indentiargvdefaultrrr _emit_argsrcsdfddt|DS)N,c3s|]\}}t||VqdSr )r).0rrrrr !z_emit_args..)join enumerate)r argumentsrrr _emit_args sr cCs t|jSr )r r )retrrr _emit_ret$sr"cCs4t|dkrt|dSdddd|DdS)NrzTuple[z, css|]}t|VqdSr )r")rrrrrr+rz_emit_rets..])lenr"r)returnsrrr _emit_rets(s  r(cCsV|dur|}n|d|}|dtt|d||j|ddt|j}|S)N.(r#z) -> )r r&rr(r')modrschema arg_startpaddingZqualified_name schema_strrrr _emit_schema.sr1c 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-r2rrris_tensor_method<s  z)_get_tensor_ops..is_tensor_methodaten::Tensorr#)r.zSupported Tensor Methods)dirr3r8r r4_jit_get_schemas_for_operatorappendr1)r6methodselemschemasr-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)r3nnZ functional__name__r9getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr-r;r1 _builtinsZ_modules_containing_builtinsrr4r:) functionsr,rr=attrZ attr_moduleZscriptedZscripted_schemabuiltinr>r-rrr_get_nn_functional_opsRs8      rOcCsrg}tjjjD]^\}}t|}t|ds,q|s2qt|jst|j st|jrRqd|jvr^q| ||fq|S)NrCztorch._C) r3rIrKZ _builtin_opsrErGhasattrr rC __qualname__r;)builtinsfn _builtin_namer,rrr_get_builtins_helper}s   rUcCs(t|}|std|d|jdkS)Nr@rAmath)rErGrHrC)rSr,rrr _is_math_fns rWc Csg}tddt}t|}|D]b\}}t|}|sFtd|dt|}|dur tj |}|D]}| t |j |j |qfq d|fS)NcSst|d SNrrWrSrrrrz+_get_torchscript_builtins..r@rAzTorchScript Builtin Functions) filterrUlistrErGrHrr3r4r:r;r1rC) rLrR builtins_listrSrTr,rNr>r-rrr_get_torchscript_builtinss   r_c Csg}tddt}t|}|D]p\}}t|}|sFtd|dt|}|dur tj |}|D](}t |j |j |} d| vrqf| |qfq d|fS)NcSs t|dSrXrYrZrrrr[rz$_get_math_builtins..r@rAr8z``math`` Module) r\rUr]rErGrHrr3r4r:r1rCr;) rLrRr^rSrTr,rNr>r-r0rrr_get_math_builtinss    r`c CsXgd}dddddddd d }d d d d ddddd}gd}g}|D] \}}|d|d|dqDg}g}|D]}d|} ||vr||} tj| } | D]} |td|| ddqt| dkr|dqrd|d||d} || qrd|} d|}d|}t| d} t|d}t|d}d|d|d | d!}d"|fS)#N)printtuplefloatcomplexintboolr rDrP isinstancer&hexoctroundhashminmaxabsalldivmodr]ordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rfrercrdrnrmrlrtzPrint 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)rarbrDrPrgrurrt))rd __complex__)rc __float__)re__int__)rf__bool__)r __str__)r&__len__)rhZ__hex__)riZ__oct__"z", "``z``"r7r)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;r3r4r:r1r&rtextwrapr)Zsupported_builtinsZ op_renamesZschemaless_op_explanationsZ magic_methodsZmagic_methods_rowsrSZ magic_methodZschematized_opsZschemaless_opsZop_namer>sZ table_rowZschematized_ops_strZschemaless_ops_strZmagic_methods_rows_strsectionrrr_get_global_builtinssf                 rcCsdd}d}tttttf}|D]}|\}}|dddddd}t|trv|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@rz:_list_supported_ops..emit_block..)formatr)Zdeclsrrr emit_block>sz'_list_supported_ops..emit_blockr~`-rr~z.. _:r) r?rOr_rr`replacelowerrgr r&)rbodyZop_gathering_fnsrSheaderitemsZ link_targetrrrr_list_supported_ops=s"      r)rr))rErZ torch.jitr3Ztorch.jit._builtinsrr r rr r"r(r1r?rOrUrWr_r`rr__doc__rrrrs&   +y