Searched defs:vec_max (Results 1 – 7 of 7) sorted by relevance
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve/ |
| D | impl.cpp | 51 auto vec_max = wrapper::svdup_n(support::cpp11::lowest<ScalarType>()); in sve_logits_1d_max() local
98 const auto vec_max = wrapper::svdup_n(max_val); in sve_softmax_logits_1d_float() local
|
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/ |
| D | impl.cpp | 56 auto vec_max = wrapper::vdup_n(support::cpp11::lowest<T>(), ExactTagType{}); in neon_logits_1d_max() local
123 const auto vec_max = wrapper::vdup_n(max_val, wrapper::traits::vector_128_tag{}); in neon_softmax_logits_1d_quantized() local
298 const auto vec_max = wrapper::vdup_n(max_val, ExactTagType{}); in neon_softmax_logits_1d_float() local
|
| /external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve2/ |
| D | impl.cpp | 66 const auto vec_max = wrapper::svdup_n(max_val); in sve2_softmax_logits_1d_quantized() local
|
| /external/clang/lib/Headers/ |
| D | vecintrin.h | 6408 vec_max(vector signed char __a, vector signed char __b) { in vec_max() function
6413 vec_max(vector signed char __a, vector bool char __b) { in vec_max() function
6419 vec_max(vector bool char __a, vector signed char __b) { in vec_max() function
6425 vec_max(vector unsigned char __a, vector unsigned char __b) { in vec_max() function
6430 vec_max(vector unsigned char __a, vector bool char __b) { in vec_max() function
6436 vec_max(vector bool char __a, vector unsigned char __b) { in vec_max() function
6442 vec_max(vector signed short __a, vector signed short __b) { in vec_max() function
6447 vec_max(vector signed short __a, vector bool short __b) { in vec_max() function
6453 vec_max(vector bool short __a, vector signed short __b) { in vec_max() function
6459 vec_max(vector unsigned short __a, vector unsigned short __b) { in vec_max() function
[all …]
|
| D | altivec.h | 3028 vec_max(vector signed char __a, vector signed char __b) { in vec_max() function
3033 vec_max(vector bool char __a, vector signed char __b) { in vec_max() function
3038 vec_max(vector signed char __a, vector bool char __b) { in vec_max() function
3043 vec_max(vector unsigned char __a, vector unsigned char __b) { in vec_max() function
3048 vec_max(vector bool char __a, vector unsigned char __b) { in vec_max() function
3053 vec_max(vector unsigned char __a, vector bool char __b) { in vec_max() function
3057 static __inline__ vector short __ATTRS_o_ai vec_max(vector short __a, in vec_max() function
3062 static __inline__ vector short __ATTRS_o_ai vec_max(vector bool short __a, in vec_max() function
3067 static __inline__ vector short __ATTRS_o_ai vec_max(vector short __a, in vec_max() function
3073 vec_max(vector unsigned short __a, vector unsigned short __b) { in vec_max() function
[all …]
|
| /external/executorch/extension/llm/custom_ops/ |
| D | op_sdpa.cpp | 92 auto vec_max = vec::Vectorized<T1>(val); in _exp_reduce_sum_fusion_kernel() local
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | FlashAttentionKernel.cpp | 61 auto vec_max = vec::Vectorized<T1>(val); in _exp_reduce_sum_fusion_kernel() local
|