o fi(@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do |d S)N___) startswith)namera/var/www/addictedbytheproject.nl/epg/venv/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_startpaddingqualified_name schema_strrrr _emit_schema.sr1c Csjdtfdd}g}ttjD]"}t|s0tjd|}|D]}||r/|td||ddqqd|fS) NreturncSsFt|jdkr dS|jd}|jdkrdS|jtjjs!dSdS)NrFselfT) r%rrr isSubtypeOftorch_C TensorTypeget)r,r3rrris_tensor_method<s  z)_get_tensor_ops..is_tensor_methodaten::Tensorr")r-zSupported Tensor Methods) booldirr5r;r r6_jit_get_schemas_for_operatorappendr1)r9methodselemschemasr,rrr_get_tensor_ops;s rCc 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)r5nn functional__name__r=getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr,r?r1 _builtins_modules_containing_builtinsrr6r>) functionsr+rrAattr attr_modulescriptedscripted_schemabuiltinrBr,rrr_get_nn_functional_opsRs<      rXcCsrg}tjjjD]/\}}t|}t|dsq|sqt|js(t|j s(t|jr)qd|jvr/q| ||fq|S)NrHztorch._C) r5rNrP _builtin_opsrJrLhasattrr rH __qualname__r?)builtinsfn _builtin_namer+rrr_get_builtins_helper}s   r_cCs(t|}|std|d|jdkS)NrDrEmath)rJrLrMrH)r]r+rrr _is_math_fns  rac Csg}tddt}t|}|D]1\}}t|}|s#td|dt|}|durAtj |}|D] }| t |j |j |q3qd|fS)NcSst|d SNrrar]rrrsz+_get_torchscript_builtins..rDrEzTorchScript Builtin Functions) filterr_listrJrLrMrr5r6r>r?r1rH) rRr\ builtins_listr]r^r+rWrBr,rrr_get_torchscript_builtinss   ric Csg}tddt}t|}|D]8\}}t|}|s#td|dt|}|durHtj |}|D]}t |j |j |} d| vrBq3| |q3qd|fS)NcSs t|dSrbrcrdrrrres z$_get_math_builtins..rDrEr;z``math`` Module) rfr_rgrJrLrMrr5r6r>r1rHr?) rRr\rhr]r^r+rWrBr,r0rrr_get_math_builtinss"    rjc 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)printtuplefloatcomplexintr<r rIrZ isinstancer%hexoctroundhashminmaxabsalldivmodrgordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)r<rormrnrwrvrur}zPrint 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 staticzArguments must be iterable.z-Can only be used as an iterator in a for loop)rkrlrIrZrpr~rr}))rn __complex__)rm __float__)ro__int__)r<__bool__)r __str__)r%__len__)rq__hex__)rr__oct__"z", "``z``"r:r)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 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?r5r6r>r1r%rtextwrapr)supported_builtins op_renamesschemaless_op_explanations magic_methodsmagic_methods_rowsr] magic_methodschematized_opsschemaless_opsop_namerBs table_rowschematized_ops_strschemaless_ops_strmagic_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 :: {} rcss|] }d|dVqdS)z  Nr)rdrrrrBsz:_list_supported_ops..emit_block..)formatr)declsrrr emit_block@sz'_list_supported_ops..emit_blockr`-rr~z.. _:r) rCrXrirrjreplacelowerrpr r%)rbodyop_gathering_fnsr]headeritems link_targetrrrr_list_supported_ops?s"      r)rr()rJr torch.jitr5torch.jit._builtinsrr r rrr!r'r1rCrXr_rarirjrr__doc__rrrrs(    +{