U IºcPãZ@sÐUddlZddlZddlZddlZddlmmZddlmZm Z m Z m Z m Z ddl mZddlmZmZdaeeeefed<eejjejjejjejjejjejjfZe dfe dfed fe d fe d fed fed fejd fej dfej!dfej"dfej#dfej$dfej%dfej&dfej'dfej(dfej)dfej*dfej+dfej,dfej-dfej.dfej/dfej0dfej1dfej2d fej3d!fej4d"fej5d#fej6d$fej7d%fej8d&fej9d'fej:d(fej;d)fej0dfej1dfejd,fej?d-fej@d.fejAd/fejBd0fejCd1fej/dfejDd2fejAd/fejEd3fejFd4fej*dfej+dfej.dfej$dfej:d(fej;d)fej9d'fej6d$fej5d#fej7d%fej3d!fej2d fej4d"fej0dfejfejMjQjUd?fejjVd@fejWdAfejXjYdBfejXjZdCfejXj[dDfejXj\dEfejXj]dFfejXj^dGfejXj_dHfejXj`dIfejXjadJfgYZbdKdL„ZcecƒZddMdN„ZedOdP„ZfdQdR„ZgdSdT„ZhdS)UéNé)Ú_singleÚ_pairÚ_tripleÚ _quadrupleÚ_list_with_default)Ú OrderedDict)ÚDictÚOptionalÚ_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::radiansúaten::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_statez aten::warnz aten::stftz aten::istftz aten::cdistz aten::normzaten::unique_dimzaten::unique_consecutivezaten::nuclear_normzaten::frobenius_normzaten::tensordotcCs&ddddddddg}td d „|DƒƒS) NÚstftÚistftZluÚcdistÚnormÚuniqueÚunique_consecutiveÚ tensordotcss|]}ttj|ƒVqdS©N)ÚgetattrÚtorchÚ functional)Ú.0Úname©rú7/tmp/pip-unpacked-wheel-gikjz4vx/torch/jit/_builtins.pyÚ xsz7_gen_torch_functional_registered_ops..)Úset)ÚopsrrrÚ$_gen_torch_functional_registered_opsrsrcCs|tkSr)Ú_functional_registered_ops©ÚfnrrrÚ_is_special_functional_bound_op|sr#cCs²tdk r tSiadd„}tD] }||ƒqt tjdf¡t tjdf¡t tjdf¡ddlm m }|  ¡r”t |j df¡t |j df¡tD]\}}|tt|ƒ<q˜tS) NcSsRt|ƒD]D}t||ƒ}t|ƒrt|ƒs|tjk r|tjk rt |d|f¡qdS)Nzaten::) ÚdirrÚcallabler#rZno_gradZautocastÚ _builtin_opsÚappend)ÚmodrÚvrrrÚ register_all†s  $z(_get_builtin_table..register_allz aten::gcdr zaten::mathremainderrzaten::get_gradientszaten::dist_backward)r Ú_modules_containing_builtinsr&r'ÚmathÚgcdÚisfiniteÚ remainderZtorch.distributed.autogradZ distributedÚautogradZ is_availableZ get_gradientsÚbackwardÚid)r*r(Z dist_autogradÚbuiltinZaten_oprrrÚ_get_builtin_table€s   r4cCs|tƒt|ƒ<dSr)r4r2)r"ÚoprrrÚ_register_builtinžsr6cCstƒ t|ƒ¡Sr)r4Úgetr2r!rrrÚ _find_builtin¢sr8)ir,ZcmathÚwarningsrZtorch.backends.cudnnÚbackendsZcudnnZnn.modules.utilsrrrrrÚ collectionsrÚtypingr r r ÚintÚstrÚ__annotations__Z_CZ_nnZ_fftZ_linalgZ_nestedZ_sparseÚ_specialr+ÚdictZ is_acceptableÚceilÚcopysignÚerfÚerfcÚexpÚexpm1ÚfabsÚfloorÚgammaÚlgammaÚlogÚlog10Úlog1pÚpowÚsqrtÚisnanÚasinhÚatanhÚcoshÚsinhÚtanhÚacosÚasinÚatanÚatan2ÚcosÚsinÚtanÚacoshÚfmodÚmodfÚ factorialÚfrexpÚisinfÚdegreesÚradiansr.ÚphaseZrectÚldexpZ_assertr0Zgradr1Z _infer_sizeÚnnrZ_no_grad_embedding_renorm_Zassert_int_or_pairÚinitZ_no_grad_fill_Z_no_grad_normal_Z_no_grad_uniform_Z_no_grad_zero_Z_get_tracing_stateÚwarnZ_VFr rrrZ unique_dimrZ nuclear_normZfrobenius_normrr&rr r#r4r6r8rrrrÚsÔ *                   ¦a