o i@sUddlZddlZddlZddlmZddlmZmZddlZddl m m Z ddl m Z mZmZmZmZdaeeeefed<eejjejjejjejjejjejjfZgedfedfedfed fe d fed fed fe jd fej d fej!dfej"dfej#dfej$dfej%dfej&dfej'dfej(dfej)dfej*dfej+dfej,dfej-dfej.dfej/dfej0dfej1dfej2dfej3d fej4d!fej5d"fej6d#fej7d$fej8d%fej9d&fej:d'fej;d(fej0dfej1dfejd+fej?d,fej@d-fejAd.fejBd/fejCd0fej/dfejDd1fejAd.fejEd2fejFd3fej*dfej+dfej.dfej$dfej:d'fej;d(fej9d&fej6d#fej5d"fej7d$fej3d fej2dfej4d!fej0dfejfejjVd?fejjWd@fejXdAfejYjZdBfejYj[dCfejYj\dDfejYj]dEfejYj^dFfejYj_dGfejYj`dHfejYjadIfejYjbdJfZcdKdLZdedZedMdNZfdOdPZgdQdRZhdSdTZidS)UN) OrderedDict)DictOptional)_list_with_default_pair _quadruple_single_triple_builtin_tablez aten::_pairzaten::_quadruplez aten::_singlez aten::_triplezaten::list_with_defaultz aten::dictzaten::cudnn_is_acceptablez aten::ceilzaten::copysignz aten::erfz aten::erfcz aten::expz aten::expm1z aten::fabsz aten::floorz aten::gammaz aten::lgammaz aten::logz aten::log10z aten::log1pz aten::powz aten::sqrtz aten::isnanz aten::asinhz aten::atanhz aten::coshz aten::sinhz aten::tanhz aten::acosz aten::asinz aten::atanz aten::atan2z aten::cosz aten::sinz aten::tanz aten::acoshz aten::fmodz aten::modfzaten::factorialz aten::frexpz aten::isinfz aten::degreesz aten::radiansaten::isfinitez aten::anglez aten::polarz aten::ldexpz aten::_assertz aten::gradzaten::backwardzaten::_infer_sizez aten::_no_grad_embedding_renorm_zaten::_assert_int_or_pairzaten::_no_grad_fill_zaten::_no_grad_normal_zaten::_no_grad_uniform_zaten::_no_grad_zero_zaten::_get_tracing_statezaten::_get_cpu_capabilityz aten::warnz aten::stftz aten::istftz aten::cdistz aten::normzaten::unique_dimzaten::unique_consecutivezaten::nuclear_normzaten::frobenius_normzaten::tensordotcCsgd}dd|DS)N)stftistftlucdistnormuniqueunique_consecutive tensordotcSsh|]}ttj|qS)getattrtorch functional).0namerrQ/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/jit/_builtins.py sz7_gen_torch_functional_registered_ops..r)opsrrr$_gen_torch_functional_registered_opszs rcCs|tvSN)_functional_registered_opsfnrrr_is_special_functional_bound_opsr"cCstdurtSiadd}tD]}||qttjdfttjdfttjdfddlm m }| rJt|j dft|j dftD] \}}|tt|<qLtS) NcSsft|D],}t||}t|r0t|s0|tjur0|tjur0|dkr'|dd}t|d|fqdS)N_segment_reducezaten::) dirrcallabler"rno_gradautocast _builtin_opsappend)modrvrrr register_alls     z(_get_builtin_table..register_allz aten::gcdr zaten::mathremainderrzaten::get_gradientszaten::dist_backward)r _modules_containing_builtinsr)r*mathgcdisfinite remaindertorch.distributed.autograd distributedautograd is_available get_gradientsbackwardid)r-r+ dist_autogradbuiltinaten_oprrr_get_builtin_tables   r=cCs|tt|<dSr)r=r9)r!oprrr_register_builtinsr?cCstt|Sr)r=getr9r rrr _find_builtinsrA)jcmathr/warnings collectionsrtypingrrrtorch.backends.cudnnbackendscudnntorch.nn.modules.utilsrrrrr r intstr__annotations___C_nn_fft_linalg_nested_sparse_specialr.dict is_acceptableceilcopysignerferfcexpexpm1fabsfloorgammalgammaloglog10log1ppowsqrtisnanasinhatanhcoshsinhtanhacosasinatanatan2cossintanacoshfmodmodf factorialfrexpisinfdegreesradiansr1phaserectldexp_assertr5gradr8 _infer_sizennr_no_grad_embedding_renorm_assert_int_or_pairinit_no_grad_fill__no_grad_normal__no_grad_uniform__no_grad_zero__get_tracing_state_get_cpu_capabilitywarn_VFr r rr unique_dimr nuclear_normfrobenius_normrr)rrr"r=r?rArrrrs   *      !"#$%&'()*+,-./0123456789:;<=>?@ABCDEF G H I J K L M N O P QR S T U V W X Y Z [c(