o h @sRUddlZddlmZejjZddlZddlZddlmZmZm Z ddl m Z ddl m Z iZeeejjfed<eZeeed<e dZe d Zd d Zd d Z ddejjdeeeejjfdeeeefgeeefffddZeejj   ddedeee dee de!def ddZ"eejj#d dede!defddZdS)!N)Tensor)CallableOptionalTypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_PcCs`tj}t|}|jD]}|j|ks Jd|jd|jq |j|ks.Jd|jdS)NzNo signature on param z for function z"No return annotation for function )inspect_empty signature parametersvalues annotationnamereturn_annotation)f inspect_emptysigparamrm/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationss    rc Cs|j}|j}t|t|krdSt||D]=\}}tj}dD]}|dkr2|jdkr2tdt ||t ||kr@dSq"|j }|j } ||krV| |krV|| krVdSq|j |j kS)NF)rrrselfz4PyTorch uses 'input' instead of 'self' on public api) rlenziprr r rwarningswarngetattrdefaultr) decomposition_sig torch_op_sig decomp_params op_params decomp_paramop_paramrfielddecomp_default op_defaultrrrsignatures_match#s&  r+aten_opregistryreturncs,dtttfdtttfffdd }|S)Nrr.csdurtttjjsJ|jtvsJd|jt|jtj |}tj |j t dD]}tj |j tj |j q3|tj<|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,r-rrdecomposition_decoratorFs     z7register_decomposition..decomposition_decorator)rr r )r,r-rCrrBrregister_decompositionBs(rDFinputdim correctionkeepdimc Cs|durg}|}t|ttfrt|dkr|}nd}|D] }||j|9}qt||d}||}||}t|||} |durIt |d} nt|t rUt ||} nt|t rat ||} nt d| t d| S)NrTzcorrection must be int or float) r0tuplelistrnumelshapeatenmeansumfloatint RuntimeErrormax) rErFrGrHdim_inrOsubsqrPdenomrrrvar_decompositioncs(   rZTunbiasedcCst||rddSddS)NrIr)rG)rZ)rEr[rrrvarsr\)N)NNF)T)$r1ropsrNr rtypingrrrtyping_extensionsr torch.typesrrdictr>r6ScriptFunction__annotations__setr r r rr+r2r3rDr\rGrKrRboolrZr!rrrrsN     ! !  #