Searched refs:decay_r (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | rmsprop_fp32.c | 41 __m256 decay_r = _mm256_set1_ps(1.0 - decay); in RMSPropUnuseCenterFp32() local
50 avx_r2 = _mm256_mul_ps(avx_r1, decay_r); in RMSPropUnuseCenterFp32()
91 __m256 decay_r = _mm256_set1_ps(1.0 - decay); in RMSPropUseCenterFp32() local
101 avx_r2 = _mm256_mul_ps(avx_r1, decay_r); in RMSPropUseCenterFp32()
106 avx_r1 = _mm256_mul_ps(_mm256_sub_ps(grad_r, mean_grad_r), decay_r); in RMSPropUseCenterFp32()
|
| D | adam_fp32.c | 167 __m512 decay_r = _mm512_set1_ps(decay); in AdamWeightDecayFp32() local
188 avx_r0 = _mm512_fmadd_ps(var_r, decay_r, avx_r0); in AdamWeightDecayFp32()
219 __m512 decay_r = _mm512_set1_ps(decay); in FusedCastAdamFp32() local
240 avx_r0 = _mm512_fmadd_ps(var_r, decay_r, avx_r0); in FusedCastAdamFp32()
265 __m512 decay_r = _mm512_set1_ps(decay); in FusedCastAdamFp16() local
286 avx_r0 = _mm512_fmadd_ps(var_r, decay_r, avx_r0); in FusedCastAdamFp16()
|