Searched defs:mean_vec (Results 1 – 14 of 14) sorted by relevance
20 const double* mean_vec, in RemoveMean()47 const double* mean_vec = gmm_parameters.mean; in EvaluateGmm() local
87 auto mean_vec = wrapper::vdup_n(mean, ExactTagType{}); in mean_stddev_normalization() local162 float16x8_t mean_vec = vdupq_n_f16(mean); in mean_stddev_normalization() local
77 const float32x4_t mean_vec = vdupq_n_f32(-mean); in ComputeStdDevNeon() local
72 const auto mean_vec = svld1_f16(pg, input_mean + x); in fp16_sve_batch_normalization() local
72 const auto mean_vec = svld1_f32(pg, input_mean + x); in fp32_sve_batch_normalization() local
60 auto mean_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_conv() local
60 auto mean_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nhwc() local
78 const auto mean_vec = wrapper::vloadq(input_mean + x); in batch_normalization() local
77 const auto mean_vec = wrapper::vloadq(input_mean + x); in batch_normalization() local
61 auto mean_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nchw() local
183 auto mean_vec = wrapper::vdup_n(mean, ExactTagType{}); in batch_normalization_nchw() local
237 const int32x4_t mean_vec = vdup_n(mean, wrapper::traits::vector_128_tag{}); in normalize_qasymm16() local
179 const Vec mean_vec = Vec(mean); in normal_fill_VSX() local
735 auto mean_vec = Vec(mean[n * G + g]); in GammaBackward() local748 auto mean_vec = Vec(mean[n * G + g]); in GammaBackward() local786 fVec mean_vec = fVec(opmath_t(mean[n * G + g])); in GammaBackward() local809 fVec mean_vec = fVec(opmath_t(mean[n * G + g])); in GammaBackward() local