| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedAdam.cpp | 27 const double beta1, in _fused_adam_kernel_cpu_()
84 const double beta1, in _fused_adam_kernel_cpu_()
103 const double beta1, in _fused_adamw_kernel_cpu_()
160 const double beta1, in _fused_adamw_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedAdamKernel.cu | 24 const double beta1, in _fused_adam_kernel_cuda_()
83 const double beta1, in _fused_adam_kernel_cuda_()
|
| H A D | FusedAdamWKernel.cu | 25 const double beta1, in _fused_adamw_kernel_cuda_()
84 const double beta1, in _fused_adamw_kernel_cuda_()
|
| H A D | fused_adam_impl.cu | 18 const double beta1, in _fused_adam_cuda_impl_()
64 const double beta1, in _fused_adam_cuda_impl_()
|
| H A D | fused_adamw_impl.cu | 19 const double beta1, in _fused_adamw_cuda_impl_()
65 const double beta1, in _fused_adamw_cuda_impl_()
|
| H A D | fused_adam_amsgrad_impl.cu | 19 const double beta1, in _fused_adam_amsgrad_cuda_impl_()
70 const double beta1, in _fused_adam_amsgrad_cuda_impl_()
|
| H A D | fused_adamw_amsgrad_impl.cu | 20 const double beta1, in _fused_adamw_amsgrad_cuda_impl_()
71 const double beta1, in _fused_adamw_amsgrad_cuda_impl_()
|
| H A D | fused_adam_utils.cuh | 30 const double& beta1, in adam_math()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | adam_test.py | 41 beta1=0.9, argument
181 beta1 = lambda: 0.9 function
|
| H A D | adam.py | 108 beta1=0.9, argument
|
| H A D | training_ops_test.py | 445 def _adamUpdateNumpy(self, param, g_t, t, m, v, alpha, beta1, beta2, epsilon): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | FusedAdamKernel.cpp | 274 const double beta1, in adam_fused_step_impl()
345 const double beta1, in fused_adam_kernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/ |
| H A D | optimization_parameters.proto | 194 float beta1 = 3; field
323 float beta1 = 3; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | adam_test.py | 36 beta1=0.9, argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/kernels/a64_sgemv_pretransposed/ |
| H A D | generic.cpp | 37 const bool beta1 = (beta==1.0f); in a64_sgemv_pretransposed() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | adam.cpp | 112 auto beta1 = std::get<0>(options.betas()); in step() local
|
| H A D | adamw.cpp | 117 auto beta1 = std::get<0>(options.betas()); in step() local
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/numeric/odeint/include/boost/numeric/odeint/stepper/detail/ |
| D | pid_step_adjuster.hpp | 41 const double beta1; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | training_ops.cc | 525 xla::XlaOp beta1 = ctx->Input(6); in Compile() local
615 xla::XlaOp beta1 = ctx->Input(5); in Compile() local
|
| /aosp_15_r20/external/aac/libSACdec/src/ |
| H A D | sac_reshapeBBEnv.cpp | 353 FIXP_DBL beta1 = ((FIXP_DBL)MAXVAL_DBL - beta) << SF_BETA1; in extractBBEnv() local
|
| /aosp_15_r20/external/apache-commons-math/src/main/java/org/apache/commons/math3/linear/ |
| H A D | SymmLQ.java | 238 private double beta1; field in SymmLQ.State
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops_test.cc | 261 auto beta1 = Scalar(g, 0.9); in Adam() local
|
| H A D | training_ops.cc | 3555 const Tensor& beta1 = ctx->input(6); in Compute() local
3701 const Tensor& beta1 = ctx->input(7); in Compute() local
3835 const Tensor& beta1 = ctx->input(5); in Compute() local
|
| H A D | training_ops_gpu.cu.cc | 195 const T beta1 = (*beta1_); in ApplyAdamKernel() local
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_generated_ops.cc | 225 auto beta1 = 2; in TEST() local
263 auto beta1 = 2; in TEST() local
637 auto beta1 = 2; in TEST() local
3288 auto beta1 = 2; in TEST() local
3979 auto beta1 = 2; in TEST() local
6013 auto beta1 = 2; in TEST() local
6049 auto beta1 = 2; in TEST() local
|