Searched refs:variable_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
61 variable_r = _mm256_loadu_ps(variable_ptr); in RMSPropUnuseCenterFp32()
62 variable_r = _mm256_sub_ps(variable_r, avx_r1); in RMSPropUnuseCenterFp32()
63 _mm256_storeu_ps(variable_ptr, variable_r); in RMSPropUnuseCenterFp32()
95 __m256 grad_r, mean_grad_r, mean_square_r, moment_r, variable_r; in RMSPropUseCenterFp32() local
123 variable_r = _mm256_loadu_ps(variable_ptr); in RMSPropUseCenterFp32()
124 avx_r1 = _mm256_sub_ps(variable_r, moment_r); in RMSPropUseCenterFp32()
125 variable_r = _mm256_blendv_ps(variable_r, avx_r1, cmp_r); in RMSPropUseCenterFp32()
126 _mm256_storeu_ps(variable_ptr, variable_r); in RMSPropUseCenterFp32()
|