U ʹj'@sddlZddlZddlZddlmZddZddZddZd d Z d d Z d dZ d"ddZ ddZ ddZddZddZddZddZddZd d!ZeZdS)#N) _find_builtincCs|do|d S)N___) startswith)namerV/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/jit/supported_ops.py_hidden sr cCst|SN)str)typerrr _emit_typesr cCsT|jdt|j}|j}|dk r6|dt|}|dkrPdd||}|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"sr!cCs4t|dkrt|dSdddd|DdS)NrzTuple[z, css|]}t|VqdSr )r!)rrrrrr)sz_emit_rets..])lenr!r)returnsrrr _emit_rets&s  r'cCsP|dkr|}n|d|}d|tt|d||j|dt|j}|S)N.z {}({}) -> {}r")formatrr%rr'r&)modrschema arg_startpaddingZqualified_name schema_strrrr _emit_schema,sr0c 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,r1rrris_tensor_method:s  z)_get_tensor_ops..is_tensor_methodaten::Tensorr")r-zSupported Tensor Methods)dirr2r7r r3_jit_get_schemas_for_operatorappendr0)r5methodselemschemasr,rrr_get_tensor_ops9s r>c 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)r2nnZ functional__name__r8getattrinspect isfunctionr getmodule RuntimeErrorjitscriptr,r:r0 _builtinsZ_modules_containing_builtinsrr3r9) functionsr+rr<attrZ attr_moduleZscriptedr,builtinr=rrr_get_nn_functional_opsPs8      rNcCsrg}tjjjD]^\}}t|}t|ds,q|s2qt|jst|j st|jrRqd|jkr^q| ||fq|S)NrBztorch._C) r2rHrJZ _builtin_opsrDrFhasattrr rB __qualname__r:)builtinsfn _builtin_namer+rrr_get_builtins_helper{s   rTcCs(t|}|std|d|jdkS)Nr?r@math)rDrFrGrB)rRr+rrr _is_math_fns rVc Csg}tddt}t|}|D]b\}}t|}|sFtd|dt|}|dk r tj |}|D]}| t |j |j |qfq d|fS)NcSst|d SNrrVrRrrrz+_get_torchscript_builtins..r?r@zTorchScript Builtin Functions) filterrTlistrDrFrGrr2r3r9r:r0rB) rKrQ builtins_listrRrSr+rMr=r,rrr_get_torchscript_builtinss   r_c Csg}tddt}t|}|D]p\}}t|}|sFtd|dt|}|dk r tj |}|D](}t |j |j |} d| krqf| |qfq d|fS)NcSs t|dSrWrXrYrrrrZr[z$_get_math_builtins..r?r@r7z``math`` Module) r\rTr]rDrFrGrr2r3r9r0rBr:) rKrQr^rRrSr+rMr=r,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|d6|d7qg}g}|D]}d8|} ||kr||} tj| } | D]} |td|| d9d:qt| d9kr|d;qd<|d=||d5} || qd>|} d>|}d>|}t| d?} t|d?}t|d?}d@|dA|dB| dC}dD|fS)ENprinttuplefloatcomplexintboolr rCrO 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)rarbrCrOrgrurrt)rd __complex__)rc __float__)re__int__)rf__bool__)r __str__)r%__len__)rhZ__hex__)riZ__oct__"z", "``z``"r6r)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:r2r3r9r0r%rtextwrapr)Zsupported_builtinsZ op_renamesZschemaless_op_explanationsZ magic_methodsZmagic_methods_rowsrRZ magic_methodZschematized_opsZschemaless_opsZop_namer=sZ table_rowZschematized_ops_strZschemaless_ops_strZmagic_methods_rows_strsectionrrr_get_global_builtinss                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>sz:_list_supported_ops..emit_block..)r*r)Zdeclsrrr emit_block<sz'_list_supported_ops..emit_blockr~`-rr~z.. _:r) r>rNr_rr`replacelowerrgr r%)rbodyZop_gathering_fnsrRheaderitemsZ link_targetrrrr_list_supported_ops;s"      r)rr()rDrZ torch.jitr2Ztorch.jit._builtinsrr r rrr!r'r0r>rNrTrVr_r`rr__doc__rrrrs&   +w