o F h(@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do |d S)N___) startswith)namerK/var/www/vscode/kcb/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 Csddd}g}ttjD]"}t|s-tjd|}|D]}||r,|td||ddqq d|fS)NcSsFt|jdkr dS|jd}|jdkrdS|jtjjs!dSdS)NrFselfT) r%rrr isSubtypeOftorch_C TensorTypeget)r,r2rrris_tensor_method<s  z)_get_tensor_ops..is_tensor_methodaten::Tensorr")r-zSupported Tensor Methods)dirr4r:r r5_jit_get_schemas_for_operatorappendr1)r8methodselemschemasr,rrr_get_tensor_ops;s rAc 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)r4nn functional__name__r;getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr,r=r1 _builtins_modules_containing_builtinsrr5r<) functionsr+rr?attr attr_modulescriptedscripted_schemabuiltinr@r,rrr_get_nn_functional_opsRs<      rVcCsrg}tjjjD]/\}}t|}t|dsq|sqt|js(t|j s(t|jr)qd|jvr/q| ||fq|S)NrFztorch._C) r4rLrN _builtin_opsrHrJhasattrr rF __qualname__r=)builtinsfn _builtin_namer+rrr_get_builtins_helper}s   r]cCs(t|}|std|d|jdkS)NrBrCmath)rHrJrKrF)r[r+rrr _is_math_fns  r_c Csg}tddt}t|}|D]1\}}t|}|s#td|dt|}|durAtj |}|D] }| t |j |j |q3qd|fS)NcSst|d SNrr_r[rrrsz+_get_torchscript_builtins..rBrCzTorchScript Builtin Functions) filterr]listrHrJrKrr4r5r<r=r1rF) rPrZ builtins_listr[r\r+rUr@r,rrr_get_torchscript_builtinss   rgc Csg}tddt}t|}|D]8\}}t|}|s#td|dt|}|durHtj |}|D]}t |j |j |} d| vrBq3| |q3qd|fS)NcSs t|dSr`rarbrrrrcs z$_get_math_builtins..rBrCr:z``math`` Module) rdr]rerHrJrKrr4r5r<r1rFr=) rPrZrfr[r\r+rUr@r,r0rrr_get_math_builtinss"    rhc 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 rGrX isinstancer%hexoctroundhashminmaxabsalldivmodreordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rnrmrkrlrvrurtr|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 staticzMArguments must be iterable. See :ref:`Iterables ` for details.z-Can only be used as an iterator in a for loop)rirjrGrXror}rr|))rl __complex__)rk __float__)rm__int__)rn__bool__)r __str__)r%__len__)rp__hex__)rq__oct__"z", "``z``"r9r)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=r4r5r<r1r%rtextwrapr)supported_builtins op_renamesschemaless_op_explanations magic_methodsmagic_methods_rowsr[ magic_methodschematized_opsschemaless_opsop_namer@s 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)rdrrrr@sz:_list_supported_ops..emit_block..)formatr)declsrrr emit_block>sz'_list_supported_ops..emit_blockr`-rr~z.. _:r) rArVrgrrhreplacelowerror r%)rbodyop_gathering_fnsr[headeritems link_targetrrrr_list_supported_ops=s"      r)rr()rHr torch.jitr4torch.jit._builtinsrr r rrr!r'r1rArVr]r_rgrhrr__doc__rrrrs(    +y