Searched defs:sum_vec (Results 1 – 8 of 8) sorted by relevance
313 Vec sum_vec = Vec::loadu(tmp_sum_data + d1); in _vec_softmax_backward() local334 Vec sum_vec = Vec::loadu(tmp_sum_data + d2); in _vec_softmax_backward() local522 Vec sum_vec = Vec::loadu(tmp_sum_data + d1); in _vec_log_softmax_backward() local543 Vec sum_vec = Vec::loadu(tmp_sum_data + d2); in _vec_log_softmax_backward() local829 Vec sum_vec = Vec(0.0); in _vec_softmax() local959 Vec sum_vec = Vec::loadu(tmp_sum_data + d2); in _vec_logsoftmax() local984 Vec sum_vec = Vec::loadu(tmp_sum_data + d3); in _vec_logsoftmax() local
113 Vec sum_vec = Vec::loadu(state_sum_ptr + d) + grad_vec * grad_vec; in adagrad_math() local
54 auto sum_vec = wrapper::vdup_n(static_cast<ScalarType>(0.f), ExactTagType{}); in mean_stddev_normalization() local126 float16x8_t sum_vec = vdupq_n_f16(static_cast<float16_t>(0.0f)); in mean_stddev_normalization() local
78 uint32x4_t sum_vec = vdupq_n_u32(0); in neon_qasymm8_meanstddevnorm() local
417 std::vector<uint64_t> sum_vec; in TEST() local511 std::vector<uint64_t> sum_vec; in TEST() local
288 std::vector<uint64_t> sum_vec = {1, 16, 0, 2}; in TEST() local
45 __m256i sum_vec = _mm256_setzero_si256(); in aom_compute_mean_stddev_avx2() local
52 __m128i sum_vec = _mm_setzero_si128(); in aom_compute_mean_stddev_sse4_1() local