Searched defs:zero_vec (Results 1 – 13 of 13) sorted by relevance
101 const auto zero_vec = Vectorized<float>(0); in smooth_l1_backward_cpu_kernel() local148 const auto zero_vec = Vectorized<scalar_t>(0); in smooth_l1_backward_cpu_kernel() local187 const auto zero_vec = Vectorized<scalar_t>(0); in huber_backward_cpu_kernel() local
98 auto zero_vec = Vectorized<float>(zero_val); in log_sigmoid_backward_cpu_kernel() local127 auto zero_vec = Vec(zero_val); in log_sigmoid_backward_cpu_kernel() local228 const Vec zero_vec(static_cast<scalar_t>(0)); in elu_kernel() local292 const Vec zero_vec(static_cast<scalar_t>(0)); in elu_backward_kernel() local963 auto zero_vec = Vectorized<float>((float)(0)); in leaky_relu_kernel() local982 auto zero_vec = Vec((scalar_t)(0)); in leaky_relu_kernel() local1002 auto zero_vec = Vectorized<float>((float)(0)); in leaky_relu_backward_kernel() local1022 auto zero_vec = Vec((scalar_t)(0)); in leaky_relu_backward_kernel() local
293 Vec zero_vec = Vec(scalar_t(0)); in _vec_softmax_backward() local504 Vec zero_vec = Vec(scalar_t(0)); in _vec_log_softmax_backward() local920 Vec zero_vec = Vec(scalar_t(0)); in _vec_logsoftmax() local
302 auto zero_vec = Vectorized<scalar_t>(static_cast<scalar_t>(0)); in sign_kernel() local
127 const auto zero_vec = _mm512_castsi512_pd(zero_vector); in angle() local367 auto zero_vec = _mm512_set1_epi64(0); in maximum() local380 auto zero_vec = _mm512_set1_epi64(0); in minimum() local
146 __m512 zero_vec = _mm512_set1_ps(0.f); in angle() local476 auto zero_vec = _mm512_set1_epi32(0); in maximum() local489 auto zero_vec = _mm512_set1_epi32(0); in minimum() local
454 auto zero_vec = _mm512_set1_epi64(0); in maximum() local469 auto zero_vec = _mm512_set1_epi64(0); in minimum() local
410 const auto zero_vec = _mm512_set1_ps(0.f); in angle() local861 __m512i zero_vec = _mm512_set1_epi32(0); in minimum() local1527 __m512i zero_vec = _mm512_set1_epi32(0); in minimum() local
42 xtfloatx2 zero_vec = XT_FLOAT_SX2(zero_point, 0); in quantize() local85 xtfloatx2 zero_vec = XT_FLOAT_SX2(zero_point, 0); in dequantize() local
121 const auto zero_vec = _mm256_set1_pd(0.f); in angle() local
129 const auto zero_vec = _mm256_set1_ps(0.f); in angle() local
329 const auto zero_vec = _mm256_set1_ps(0.f); in angle() local
628 Vec zero_vec = Vec(0.0f); in leaky_qrelu_out_kernel() local1107 fVec zero_vec(0.0f); in qhardswish_kernel() local1225 Vec zero_vec = Vec(0.0f); in qelu_kernel() local2794 fVec zero_vec(0.0f); local3082 fVec zero_vec(0.0f); local