o i @sUddlZddlmZejjZddlZddlZddlmZmZm Z m Z ddl m Z iZ eeejjfed<eZe eed<ddZd d Zdd d Zeejj   ddede eede e dedef ddZeejjddededefddZdS)N)Tensor)DictListOptionalSet)Numberdecomposition_tablefunction_name_setcCs`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_emptysigparamrh/var/www/html/eduruby.in/lip-sync/lip-sync-env/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) r lenziprr r rwarningswarngetattrdefaultr) Zdecomposition_sigZ torch_op_sigZ decomp_paramsZ op_paramsZ decomp_paramZop_paramrfieldZdecomp_defaultZ op_defaultrrrsignatures_matchs&  r!csfdd}|S)NcsdurtttjjsJ|jtvsJd|jt|jtj |}tj |j t dD]}tj |j tj |j q3|tj<|S)NzDuplicated function name )r isinstancetorchZ_opsZ OpOverload__name__r addjitscriptZ_CZ_jit_pass_inlinegraphrangeZ_jit_pass_peepholeZ_jit_pass_constant_propagationstrZ_schema)rZ scripted_func_aten_opregistryrrdecomposition_decorator<s     z7register_decomposition..decomposition_decoratorr)r.r/r0rr-rregister_decomposition;sr1Finputdim correctionkeepdimreturnc 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) r#tuplelistrZnumelshapeatenmeansumfloatint RuntimeErrormax) r2r3r4r5Zdim_inr<subsqr=denomrrrvar_decompositionYs(   rFTunbiasedcCst||rddSddS)Nr7r)r4)rF)r2rGrrrvar}srH)N)NNF)T)r$ropsr;r rtypingrrrrZ torch.typesrrr+r'ZScriptFunction__annotations__setr rr!r1rHr4r?boolrFrrrrrs:      #