o h'@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do |d S)N___) startswith)namerM/var/www/html/ai/venv/lib/python3.10/site-packages/torch/jit/supported_ops.py_hidden sr 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_argssrcCs 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'cCsP|dur|}n|d|}d|tt|d||j|dt|j}|S)N.z {}({}) -> {}r")formatrr%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_ops9s 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_modulescriptedr,builtinr@rrr_get_nn_functional_opsPs<      rUcCsrg}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)rZr+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^rZrrrsz+_get_torchscript_builtins..rBrCzTorchScript Builtin Functions) filterr\listrHrJrKrr4r5r<r=r1rF) rPrY builtins_listrZr[r+rTr@r,rrr_get_torchscript_builtinss   rfc Csg}tddt}t|}|D]8\}}t|}|s#td|dt|}|durHtj |}|D]}t |j |j |} d| vrBq3| |q3qd|fS)NcSs t|dSr_r`rarrrrbs z$_get_math_builtins..rBrCr:z``math`` Module) rcr\rdrHrJrKrr4r5r<r1rFr=) rPrYrerZr[r+rTr@r,r0rrr_get_math_builtinss$    rgc 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 rGrW isinstancer%hexoctroundhashminmaxabsalldivmodrdordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rmrlrjrkrurtrsr{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)rhrirGrWrnr|rr{))rk __complex__)rj __float__)rl__int__)rm__bool__)r __str__)r%__len__)ro__hex__)rp__oct__"z", "``z``"r9r)r.z":any:`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_rowsrZ magic_methodschematized_opsschemaless_opsop_namer@s table_rowschematized_ops_strschemaless_ops_strmagic_methods_rows_strsectionrrr_get_global_builtinssd                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..)r*r)declsrrr emit_block<sz'_list_supported_ops..emit_blockr`-rr~z.. _:r) rArUrfrrgreplacelowerrnr r%)rbodyop_gathering_fnsrZheaderitems link_targetrrrr_list_supported_ops;s"      r)rr()rHr torch.jitr4torch.jit._builtinsrr r rrr!r'r1rArUr\r^rfrgrr__doc__rrrrs(    +w