Searched refs:momentum_r (Results 1 – 1 of 1) sorted by relevance
42 __m256 momentum_r = _mm256_set1_ps(momentum); in RMSPropUnuseCenterFp32() local58 avx_r1 = _mm256_add_ps(_mm256_mul_ps(moment_r, momentum_r), avx_r2); in RMSPropUnuseCenterFp32()92 __m256 momentum_r = _mm256_set1_ps(momentum); in RMSPropUseCenterFp32() local118 avx_r1 = _mm256_mul_ps(moment_r, momentum_r); in RMSPropUseCenterFp32()