Searched defs:vec_scale (Results 1 – 5 of 5) sorted by relevance
36 auto vec_scale = at::vec::VectorizedN<T1, T1_n>(val); in _scale_attn_mask_fusion_kernel() local96 auto vec_scale = vec::Vectorized<scalar_t>(scale); in _mul_reduce_max_fusion_kernel() local
127 auto vec_scale = vec::Vectorized<scalar_t>(scale); in _mul_reduce_max_fusion_kernel() local
2561 __m128i vec_scale = __lsx_vreplgr2vr_w(scale); in ARGBQuantizeRow_LSX() local
3164 v4i32 vec_scale = __msa_fill_w(scale); in ARGBQuantizeRow_MSA() local
3079 v4i32 vec_scale = __msa_fill_w(scale); in ARGBQuantizeRow_MSA() local