Searched defs:kOneVec (Results 1 – 4 of 4) sorted by relevance
348 auto kOneVec = Vectorized<float>((float)(1)); in GeluKernelImpl() local377 const Vec kOneVec(scalar_t(1)); in GeluKernelImpl() local400 auto kOneVec = Vectorized<float>((float)(1)); in GeluKernelImpl() local421 const Vec kOneVec(scalar_t(1)); in GeluKernelImpl() local445 auto kOneVec = Vectorized<float>((float)(1)); in GeluBackwardKernelImpl() local508 const Vec kOneVec(scalar_t(1)); in GeluBackwardKernelImpl() local561 auto kOneVec = Vectorized<float>((float)(1)); in GeluBackwardKernelImpl() local592 const Vec kOneVec(scalar_t(1)); in GeluBackwardKernelImpl() local1224 const Vectorized<float> kOneVec(1.0f); in silu_kernel() local1240 const Vectorized<scalar_t> kOneVec(scalar_t(1)); in silu_kernel() local[all …]
148 const Vectorized<scalar_t> kOneVec(scalar_t(1)); in logit_kernel() local162 const Vectorized<scalar_t> kOneVec(scalar_t(1)); in logit_kernel() local
861 const Vectorized<scalar_t> kOneVec(scalar_t(1)); in logit_backward_kernel() local
753 const auto kOneVec = Vectorized<float>(1); in qgelu_kernel() local