Home
last modified time | relevance | path

Searched refs:vec_scale (Results 1 – 5 of 5) sorted by relevance

/external/libyuv/source/
Drow_lsx.cc2561 __m128i vec_scale = __lsx_vreplgr2vr_w(scale); in ARGBQuantizeRow_LSX() local
2584 DUP4_ARG2(__lsx_vmul_w, tmp0, vec_scale, tmp1, vec_scale, tmp2, vec_scale, in ARGBQuantizeRow_LSX()
2585 tmp3, vec_scale, tmp0, tmp1, tmp2, tmp3); in ARGBQuantizeRow_LSX()
2586 DUP4_ARG2(__lsx_vmul_w, tmp4, vec_scale, tmp5, vec_scale, tmp6, vec_scale, in ARGBQuantizeRow_LSX()
2587 tmp7, vec_scale, tmp4, tmp5, tmp6, tmp7); in ARGBQuantizeRow_LSX()
2600 DUP4_ARG2(__lsx_vmul_w, tmp0, vec_scale, tmp1, vec_scale, tmp2, vec_scale, in ARGBQuantizeRow_LSX()
2601 tmp3, vec_scale, tmp0, tmp1, tmp2, tmp3); in ARGBQuantizeRow_LSX()
2602 DUP4_ARG2(__lsx_vmul_w, tmp4, vec_scale, tmp5, vec_scale, tmp6, vec_scale, in ARGBQuantizeRow_LSX()
2603 tmp7, vec_scale, tmp4, tmp5, tmp6, tmp7); in ARGBQuantizeRow_LSX()
Drow_msa.cc3164 v4i32 vec_scale = __msa_fill_w(scale); in ARGBQuantizeRow_MSA() local
3199 tmp0 *= vec_scale; in ARGBQuantizeRow_MSA()
3200 tmp1 *= vec_scale; in ARGBQuantizeRow_MSA()
3201 tmp2 *= vec_scale; in ARGBQuantizeRow_MSA()
3202 tmp3 *= vec_scale; in ARGBQuantizeRow_MSA()
3203 tmp4 *= vec_scale; in ARGBQuantizeRow_MSA()
3204 tmp5 *= vec_scale; in ARGBQuantizeRow_MSA()
3205 tmp6 *= vec_scale; in ARGBQuantizeRow_MSA()
3206 tmp7 *= vec_scale; in ARGBQuantizeRow_MSA()
3207 tmp8 *= vec_scale; in ARGBQuantizeRow_MSA()
[all …]
/external/pytorch/aten/src/ATen/native/cpu/
DFlashAttentionKernel.cpp36 auto vec_scale = at::vec::VectorizedN<T1, T1_n>(val); in _scale_attn_mask_fusion_kernel() local
42 auto res = a_n * vec_scale + b_n_convert; in _scale_attn_mask_fusion_kernel()
96 auto vec_scale = vec::Vectorized<scalar_t>(scale); in _mul_reduce_max_fusion_kernel() local
101 auto tmp1 = tmp0 * vec_scale; in _mul_reduce_max_fusion_kernel()
/external/libvpx/third_party/libyuv/source/
Drow_msa.cc3079 v4i32 vec_scale = __msa_fill_w(scale); in ARGBQuantizeRow_MSA() local
3114 tmp0 *= vec_scale; in ARGBQuantizeRow_MSA()
3115 tmp1 *= vec_scale; in ARGBQuantizeRow_MSA()
3116 tmp2 *= vec_scale; in ARGBQuantizeRow_MSA()
3117 tmp3 *= vec_scale; in ARGBQuantizeRow_MSA()
3118 tmp4 *= vec_scale; in ARGBQuantizeRow_MSA()
3119 tmp5 *= vec_scale; in ARGBQuantizeRow_MSA()
3120 tmp6 *= vec_scale; in ARGBQuantizeRow_MSA()
3121 tmp7 *= vec_scale; in ARGBQuantizeRow_MSA()
3122 tmp8 *= vec_scale; in ARGBQuantizeRow_MSA()
[all …]
/external/executorch/extension/llm/custom_ops/
Dop_sdpa.cpp127 auto vec_scale = vec::Vectorized<scalar_t>(scale); in _mul_reduce_max_fusion_kernel() local
132 auto tmp1 = tmp0 * vec_scale; in _mul_reduce_max_fusion_kernel()