Home
last modified time | relevance | path

Searched defs:beta2 (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFusedAdam.cpp28 const double beta2, in _fused_adam_kernel_cpu_()
85 const double beta2, in _fused_adam_kernel_cpu_()
104 const double beta2, in _fused_adamw_kernel_cpu_()
161 const double beta2, in _fused_adamw_kernel_cpu_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DFusedAdamKernel.cpp32 double beta2, in adam_math()
175 double beta2, in adam_math()
275 const double beta2, in adam_fused_step_impl()
346 const double beta2, in fused_adam_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/
H A Doptimization_parameters.proto112 float beta2 = 4; field
195 float beta2 = 4; field
299 float beta2 = 3; field
326 float beta2 = 4; field
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFusedAdamKernel.cu25 const double beta2, in _fused_adam_kernel_cuda_()
84 const double beta2, in _fused_adam_kernel_cuda_()
H A DFusedAdamWKernel.cu26 const double beta2, in _fused_adamw_kernel_cuda_()
85 const double beta2, in _fused_adamw_kernel_cuda_()
H A Dfused_adam_impl.cu19 const double beta2, in _fused_adam_cuda_impl_()
65 const double beta2, in _fused_adam_cuda_impl_()
H A Dfused_adamw_impl.cu20 const double beta2, in _fused_adamw_cuda_impl_()
66 const double beta2, in _fused_adamw_cuda_impl_()
H A Dfused_adam_amsgrad_impl.cu20 const double beta2, in _fused_adam_amsgrad_cuda_impl_()
71 const double beta2, in _fused_adam_amsgrad_cuda_impl_()
H A Dfused_adamw_amsgrad_impl.cu21 const double beta2, in _fused_adamw_amsgrad_cuda_impl_()
72 const double beta2, in _fused_adamw_amsgrad_cuda_impl_()
H A Dfused_adam_utils.cuh31 const double& beta2, in adam_math()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dadam_test.py42 beta2=0.999, argument
182 beta2 = lambda: 0.999 function
H A Dadam.py109 beta2=0.999, argument
H A Dtraining_ops_test.py445 def _adamUpdateNumpy(self, param, g_t, t, m, v, alpha, beta1, beta2, epsilon): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dadam_test.py37 beta2=0.999, argument
/aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/
H A Dadam.cpp113 auto beta2 = std::get<1>(options.betas()); in step() local
H A Dadamw.cpp118 auto beta2 = std::get<1>(options.betas()); in step() local
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/numeric/odeint/include/boost/numeric/odeint/stepper/detail/
Dpid_step_adjuster.hpp42 const double beta2; member
/aosp_15_r20/external/speex/libspeexdsp/
H A Dscal.c156 float beta, beta2; in speex_decorrelate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dtraining_ops.cc526 xla::XlaOp beta2 = ctx->Input(7); in Compile() local
616 xla::XlaOp beta2 = ctx->Input(6); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtraining_ops_test.cc262 auto beta2 = Scalar(g, 0.99); in Adam() local
H A Dtraining_ops.cc3556 const Tensor& beta2 = ctx->input(7); in Compute() local
3702 const Tensor& beta2 = ctx->input(8); in Compute() local
3836 const Tensor& beta2 = ctx->input(6); in Compute() local
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py3164 beta2, argument
3190 beta2, argument
/aosp_15_r20/out/target/common/obj/JAVA_LIBRARIES/framework-location.impl_intermediates/
Dclasses.jarMETA-INF/ META-INF/MANIFEST.MF android/ android/location/ android/ ...
/aosp_15_r20/out/soong/.intermediates/frameworks/base/location/framework-location.impl/android_common/javac/
Dframework-location.jarMETA-INF/ META-INF/MANIFEST.MF android/ android/location/ android/ ...
/aosp_15_r20/out/target/common/obj/JAVA_LIBRARIES/framework-location_intermediates/
Dclasses.jarMETA-INF/ META-INF/MANIFEST.MF android/ android/location/ android/ ...

12