Yh \UddlZddlmZejjZddlZddlZddlmZmZm Z ddl m Z ddl m Z iZeeejjfed<eZeeed<e dZe d Zd Zd Z dd ejjd eeeejjfdeeeefgeeefffdZeejj ddedeee dee de!def dZ"eejj#ddede!defdZdS)N)Tensor)CallableOptionalTypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_Pctj}tj|}|jD]'}|j|ksJd|jd|j(|j|ksJd|jdS)NzNo signature on param z for function z"No return annotation for function )inspect_empty signature parametersvalues annotationnamereturn_annotation)f inspect_emptysigparams k/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationsrsNM  A  C&&((  =000 GUZ G Gqv G G1000  M 1 1 15QV55 2 1 1 1 1c|j}|j}t|t|krdSt||D]\}}tj}dD]M}|dkr|jdkrtjdt||t||krdSN|j }|j } ||kr| |kr || krdS|j |j kS)NF)rrrselfz4PyTorch uses 'input' instead of 'self' on public api) rlenziprrrrwarningswarngetattrdefaultr) decomposition_sig torch_op_sig decomp_params op_params decomp_paramop_paramrfielddecomp_default op_defaults rsignatures_matchr.#s %0M'I =S^^++u"%m&:&:&<&N>N>P>P"Q"Q h   +  E<#4#>#> TUUU|U++wx/G/GGGuuuH&-%  ] * *z]/J/J++uu  .,2P PPraten_opregistryreturnc|dtttfdtttfffd }|S)Nrr1cBtttjjsJ|jt vsJd|jt |jtj |}tj |j tdD]J}tj |j tj |j K|t!j<|S)NzDuplicated function name )r isinstancetorch_ops OpOverload__name__r addjitscript_C_jit_pass_inlinegraphrange_jit_pass_peephole_jit_pass_constant_propagationstr_schema)r scripted_func_r/r0s rdecomposition_decoratorz7register_decomposition..decomposition_decoratorFs  *H'5:#899999z!2222 4 4 4322 aj))) ((++  !!-"5666q I IA H ' ' (; < < < H 3 3M4G H H H H)6W_%%&r)rr r )r/r0rGs`` rregister_decompositionrHBsO8BF#3R8H. #"rFinputdim correctionkeepdimcV|g}|}t|ttfr(t|dkr|}nd}|D]}||j|z}t ||d}||z }||z}t |||} |t|dz } n_t|trt||z } n7t|trt||z } ntd| td| z S)NrTzcorrection must be int or float) r5tuplelistrnumelshapeatenmeansumfloatint RuntimeErrormax) rIrJrKrLdim_inrTsubsqrUdenoms rvar_decompositionr_cs- {#t}%%$#c((a-- KKMM  $ $E U# #AA 99UC & &D $,C sB ((2sG $ $Ca!e  j# & & B!j.))EE  E * * B!HHz)EE@AA A Q rTunbiasedc,t||rdndS)NrNr)rK)r_)rIr`s rvarrbs UX0D1 F F FFr)N)NNF)T)$r6ropsrSrr!typingrrrtyping_extensionsr torch.typesrr dictrCr;ScriptFunction__annotations__setr r r rr.r7r8rHrbrKrPrWboolr_r$rrrmsI  y~..........''''''<>T#uy778===!cee3s8### WT]]Yt__   QQQB?C## Z "#tC!99:;#xB (2r6"223####B+,, $#'     $s)         -, F())GGvGGGGG*)GGGr