Searched refs:moment_r (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | rmsprop_fp32.c | 45 __m256 gradient_r, mean_square_r, moment_r, variable_r, avx_r1, avx_r2; in RMSPropUnuseCenterFp32() local
57 moment_r = _mm256_loadu_ps(moment_ptr); in RMSPropUnuseCenterFp32()
58 avx_r1 = _mm256_add_ps(_mm256_mul_ps(moment_r, momentum_r), avx_r2); in RMSPropUnuseCenterFp32()
95 __m256 grad_r, mean_grad_r, mean_square_r, moment_r, variable_r; in RMSPropUseCenterFp32() local
117 moment_r = _mm256_loadu_ps(moment_ptr); in RMSPropUseCenterFp32()
118 avx_r1 = _mm256_mul_ps(moment_r, momentum_r); in RMSPropUseCenterFp32()
120 moment_r = _mm256_blendv_ps(moment_r, avx_r1, cmp_r); in RMSPropUseCenterFp32()
121 _mm256_storeu_ps(moment_ptr, moment_r); in RMSPropUseCenterFp32()
124 avx_r1 = _mm256_sub_ps(variable_r, moment_r); in RMSPropUseCenterFp32()
|