Searched defs:beta_vec (Results 1 – 14 of 14) sorted by relevance
75 … const auto beta_vec = (input_beta != nullptr) ? svld1_f32(pg, input_beta + x) : const_0; in fp32_sve_batch_normalization() local
75 … const auto beta_vec = (input_beta != nullptr) ? svld1_f16(pg, input_beta + x) : const_0; in fp16_sve_batch_normalization() local
49 auto beta_vec = Vec(beta_val); in addr_kernel() local
1045 const Vec beta_vec(beta); in softplus_kernel() local1066 const Vec beta_vec(beta); in softplus_kernel() local1088 const Vec beta_vec(beta); in softplus_backward_kernel() local1112 const Vec beta_vec(beta); in softplus_backward_kernel() local
106 const Vec beta_vec(beta_data[c]); in batch_norm_cpu_contiguous_impl() local159 Vec beta_vec = Vec::loadu(beta_data + d); in batch_norm_cpu_channels_last_impl() local166 Vec beta_vec = Vec::loadu(beta_data + d, n_channel - d); in batch_norm_cpu_channels_last_impl() local
63 auto beta_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_conv() local
63 auto beta_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nhwc() local
81 …const auto beta_vec = (input_beta != nullptr) ? wrapper::vloadq(input_beta + x) : wrapper::vdup_n… in batch_normalization() local
80 …const auto beta_vec = (input_beta != nullptr) ? wrapper::vloadq(input_beta + x) : wrapper::vdup_n… in batch_normalization() local
64 auto beta_vec = wrapper::vdup_n(ScalarType(0), ExactTagType{}); in fused_batch_normalization_dwc_nchw() local
196 const auto beta_vec = wrapper::vdup_n(static_cast<T>(_norm_info.beta()), ExactTagType{}); in normalize_float() local
186 auto beta_vec = wrapper::vdup_n(beta, ExactTagType{}); in batch_normalization_nchw() local
239 const vec_t * beta_vec = (beta != nullptr) ? reinterpret_cast<const vec_t*>(beta) : nullptr; in vectorized_layer_norm_kernel_impl() local
2849 fVec beta_vec(beta); local