25 #ifdef ARM_COMPUTE_ENABLE_SVE
27 #include "../std_transforms_sve.hpp"
28 #include "../bfloat.hpp"
29 #include "../performance_parameters.hpp"
32 unsigned int, const unsigned int *, \
33 IndirectInputArg<bfloat16>, \
36 IndirectOutputArg<float>, \
37 const float *, Activation, bool
42 void sve_hybrid_bf16fp32_dot_6x4VL( ARGLIST );
44 class cls_sve_hybrid_bf16fp32_dot_6x4VL
49 typedef float result_type;
51 typedef void (*kern_type)( ARGLIST );
54 static constexpr
unsigned int out_height()
59 static unsigned int out_width()
61 return get_vector_length<float>() * 4;
64 static constexpr
unsigned int k_unroll()
69 static constexpr
bool supports_accumulate()
74 StdTransformsSVE<rhs_operand_type, result_type, 6, 4, 2> transforms = {};
76 static inline PerformanceParameters get_performance_parameters(
const CPUInfo *
ci)
78 if (std::is_same<T, bfloat16>::value) {
93 kern_type kernel=sve_hybrid_bf16fp32_dot_6x4VL;
94 cls_sve_hybrid_bf16fp32_dot_6x4VL(
const CPUInfo *)
102 #endif // ARM_COMPUTE_ENABLE_SVE