Searched refs:kAlphaVec (Results 1 – 2 of 2) sorted by relevance
399 auto kAlphaVec = Vectorized<float>((float)(M_SQRT1_2)); in GeluKernelImpl() local410 auto res0 = x0 * kPointFiveVec * (kOneVec + (x0 * kAlphaVec).erf()); in GeluKernelImpl()411 auto res1 = x1 * kPointFiveVec * (kOneVec + (x1 * kAlphaVec).erf()); in GeluKernelImpl()420 const Vec kAlphaVec(scalar_t(M_SQRT1_2)); in GeluKernelImpl() local431 (kOneVec + (x_vec * kAlphaVec).erf()); in GeluKernelImpl()559 auto kAlphaVec = Vectorized<float>((float)(M_SQRT1_2)); in GeluBackwardKernelImpl() local577 auto cdf_vec0 = kPointFiveVec * (kOneVec + (x0 * kAlphaVec).erf()); in GeluBackwardKernelImpl()578 auto cdf_vec1 = kPointFiveVec * (kOneVec + (x1 * kAlphaVec).erf()); in GeluBackwardKernelImpl()590 const Vec kAlphaVec(scalar_t(M_SQRT1_2)); in GeluBackwardKernelImpl() local607 kPointFiveVec * (kOneVec + (x_vec * kAlphaVec).erf()); in GeluBackwardKernelImpl()
750 const auto kAlphaVec = Vectorized<float>(M_SQRT1_2); in qgelu_kernel() local819 value = value * kPointFiveVec * (kOneVec + (value * kAlphaVec).erf()); in qgelu_kernel()