Searched defs:alpha_vec (Results 1 – 4 of 4) sorted by relevance
50 auto alpha_vec = Vec(alpha_val); in addr_kernel() local
105 const Vec alpha_vec(alpha_data[c]); in batch_norm_cpu_contiguous_impl() local158 Vec alpha_vec = Vec::loadu(alpha_data + d); in batch_norm_cpu_channels_last_impl() local165 Vec alpha_vec = Vec::loadu(alpha_data + d, n_channel - d); in batch_norm_cpu_channels_last_impl() local
47 auto alpha_vec = Vectorized<scalar_t>(alpha); in add_clamp_kernel() local
1227 Vec alpha_vec = Vec(alpha_float); in qelu_kernel() local