U dr(@sddlZddlmZddlZddlZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do|d S)N___) startswith)namer;/tmp/pip-unpacked-wheel-ua33x9lu/torch/jit/supported_ops.py_hidden sr cCst|SN)str)typerrr _emit_type sr cCsNd|jt|j}|j}|dk r2d|t|}|dkrJdd||}|S)Nz{} : {}z{}={}rz {}{} )formatrr 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_retsr cCs2t|dkrt|dSdddd|DS)Nrz Tuple[{}]z, css|]}t|VqdSr )r )rrrrrr!sz_emit_rets..)lenr rr)returnsrrr _emit_retss  r%cCsN|dkr|}n d||}d|tt|d||j|dt|j}|S)Nz{}.{}z {}({}) -> {}r!)rrr#rr%r$)modrschema arg_startpaddingZqualified_name schema_strrrr _emit_schema#s r,c 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(r-rrris_tensor_method.s  z)_get_tensor_ops..is_tensor_methodzaten::Tensorr!)r)zSupported Tensor Methods)dirr.r2r r/_jit_get_schemas_for_operatorappendr,)r1methodselemschemasr(rrr_get_tensor_ops-s r9c Csg}tjj}|j}ttjjD]}t||}t|rt|drDqt |}|sbt d|dd|jkrnqz(tj |}|j }|t|||WqYqXqtj jjD]^}|j}t|D]J}tt||}|dk rtj|} | D]}t|s|t|||qqqd|fS)Nr Module for not foundztorch.nn.functionalzSupported PyTorch Functions)r.nnZ functional__name__r3getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr(r5r, _builtinsZ_modules_containing_builtinsrr/r4) functionsr'rr7attrZ attr_moduleZscriptedr(builtinr8rrr_get_nn_functional_opsCs8      rIcCsrg}tjjjD]^\}}t|}t|ds,q|s2qt|jst|j st|jrRqd|jkr^q| ||fq|S)Nr=ztorch._C) r.rCrEZ _builtin_opsr?rAhasattrr r= __qualname__r5)builtinsfn _builtin_namer'rrr_get_builtins_helperms   rOcCs(t|}|std|d|jdkS)Nr:r;math)r?rArBr=)rMr'rrr _is_math_fns rQc Csg}tddt}t|}|D]b\}}t|}|sFtd|dt|}|dk r tj |}|D]}| t |j |j |qfq d|fS)NcSst|d SNrrQrMrrrz+_get_torchscript_builtins..r:r;zTorchScript Builtin Functions) filterrOlistr?rArBrr.r/r4r5r,r=) rFrL builtins_listrMrNr'rHr8r(rrr_get_torchscript_builtinss   rZc Csg}tddt}t|}|D]p\}}t|}|sFtd|dt|}|dk r tj |}|D](}t |j |j |} d| krqf| |qfq d|fS)NcSs t|dSrRrSrTrrrrUrVz$_get_math_builtins..r:r;r2z``math`` Module) rWrOrXr?rArBrr.r/r4r,r=r5) rFrLrYrMrNr'rHr8r(r+rrr_get_math_builtinss"    r[cCsddddddddd d d d d dddddddddddddddg}dddd d!d"d#d$d%}d&d'd(d(d)d*d*d+d,}d-d.d/d0d1d2d3d4g}g}|D]\}}|d5||qg}g}|D]~}d6|} ||kr||} tj| } | D]} |td|| d7d8qt| d7kr|d9qd:|||} || qd;|} d;|}d;|}t | d<} t |d<}t |d<}d=||| }d>|fS)?Nprinttuplefloatcomplexintboolr r>rJ isinstancer#hexoctroundhashminmaxabsalldivmodrXordchrbinrangeziprsortedz aten::Boolz aten::Intz aten::Floatz aten::Complexz prim::absz prim::maxz prim::minzfake::does_not_exist)rar`r^r_rirhrgrozPrint 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\r]r>rJrbrprro)r_ __complex__)r^ __float__)r`__int__)ra__bool__)r __str__)r#__len__)rcZ__hex__)rdZ__oct__z"{}", "``{}``"zaten::{}r)r*z":any:`{}`", "{}"  a The functions in the following table are supported but do not have a static schema .. csv-table:: :header: "Function", "Note" {} The following functions will use the corresponding magic method on :any:`TorchScript classes` .. csv-table:: :header: "Function", "Magic Method" {} These built-in functions use the schema .. rst-class:: codeblock-height-limiter :: {} zPython Built-in Functions) r5rr.r/r4r,r#rtextwrapr)Zsupported_builtinsZ op_renamesZschemaless_op_explanationsZ magic_methodsZmagic_methods_rowsrMZ magic_methodZschematized_opsZschemaless_opsZop_namer8sZ table_rowZschematized_ops_strZschemaless_ops_strZmagic_methods_rows_strsectionrrr_get_global_builtinss               r~cCsdd}d}tttttf}|D]}|\}}|dddddd}t|trnd |dt ||}nd |dt |||}d |d |}||7}q|S) NcSsdddd|DS)Nz1 .. rst-class:: codeblock-height-limiter :: {} rxcss|]}d|VqdS)z {} N)r)rdrrrr.sz:_list_supported_ops..emit_block..)rr)Zdeclsrrr emit_block-sz'_list_supported_ops..emit_blockrx`-rz {} {} {} ~z{} {} {}z.. _{}:z ) r9rIrZr~r[replacelowerrbr rr#)rbodyZop_gathering_fnsrMheaderitemsZ link_targetr}rrr_list_supported_ops,s"    r)rr&)Z torch.jitr.Ztorch.jit._builtinsrr?r{r r rrr r%r,r9rIrOrQrZr[r~r__doc__rrrrs&   *w