27 #include "../std_transforms_fixed.hpp"
28 #include "../kernel_weight_format.hpp"
29 #include "../performance_parameters.hpp"
32 unsigned int, const unsigned int *, \
33 IndirectInputArg<__fp16>, \
37 IndirectOutputArg<__fp16>, \
38 const __fp16 *, Activation, bool
43 void a64_ffhybrid_fp16_mla_6x32( ARGLIST );
45 class cls_a64_ffhybrid_fp16_mla_6x32
48 typedef __fp16 lhs_operand_type;
49 typedef __fp16 rhs_operand_type;
50 typedef __fp16 result_type;
52 typedef void (*kern_type)( ARGLIST );
55 static constexpr
unsigned int out_height()
59 static unsigned int stripe_width()
69 static unsigned int out_width()
74 static constexpr
unsigned int k_unroll()
79 static constexpr
bool supports_accumulate()
84 StdTransformsFixed<rhs_operand_type, result_type, 6, 32, 1> transforms = {};
86 static inline PerformanceParameters get_performance_parameters(
const CPUInfo *
ci)
88 if (std::is_same<T, __fp16>::value) {
99 kern_type kernel=a64_ffhybrid_fp16_mla_6x32;
100 cls_a64_ffhybrid_fp16_mla_6x32(
const CPUInfo *)
108 #endif // __aarch64__