Searched defs:one_vec (Results 1 – 8 of 8) sorted by relevance
100 auto one_vec = Vectorized<float>(one_val); in log_sigmoid_backward_cpu_kernel() local129 auto one_vec = Vec(one_val); in log_sigmoid_backward_cpu_kernel() local227 const Vec one_vec(static_cast<scalar_t>(1)); in elu_kernel() local964 auto one_vec = Vectorized<float>((float)(1)); in leaky_relu_kernel() local983 auto one_vec = Vec((scalar_t)(1)); in leaky_relu_kernel() local1003 auto one_vec = Vectorized<float>((float)(1)); in leaky_relu_backward_kernel() local1023 auto one_vec = Vec((scalar_t)(1)); in leaky_relu_backward_kernel() local1114 const Vec one_vec(static_cast<scalar_t>(1.0)); in softplus_backward_kernel() local1151 const Vec one_vec(one_val); in glu_kernel() local1207 const Vec one_vec(one_val); in glu_backward_kernel() local
814 auto one_vec = Vectorized<scalar_t>(scalar_t{1}); in sigmoid_backward_kernel() local825 auto one_vec = Vectorized<float>((float)(1)); in sigmoid_backward_kernel() local843 auto one_vec = Vectorized<scalar_t>((scalar_t)(1)); in sigmoid_backward_kernel() local909 auto one_vec = Vectorized<scalar_t>(scalar_t{1}); in tanh_backward_kernel() local922 auto one_vec = Vectorized<float>(float{1}); in tanh_backward_kernel() local940 auto one_vec = Vectorized<scalar_t>(scalar_t{1}); in tanh_backward_kernel() local
303 auto one_vec = Vectorized<scalar_t>(static_cast<scalar_t>(1)); in sign_kernel() local
148 const auto one_vec = _mm256_set1_ps(1.0f); in erf() local
173 const auto one_vec = _mm256_set1_ps(1.0f); in erf() local
193 const auto one_vec = _mm512_set1_ps(1.0f); in erf() local
621 auto one_vec = builder.AddInstruction( in BuildWhileInstruction_InitPointsToNonDistinct() local636 auto one_vec = builder.AddInstruction( in BuildWhileInstruction_InitPointsToInterfering() local
629 Vec one_vec = Vec(1.0f); in leaky_qrelu_out_kernel() local1226 Vec one_vec = Vec(1.0f); in qelu_kernel() local2793 fVec one_vec(1.0f); local3081 fVec one_vec(1.0f); local