29 #if defined(ARM_COMPUTE_ENABLE_SME) && defined(__ARM_FP16_ARGS)
34 void sme_fp16_nhwc_avg_generic_depthfirst_impl(
const uint64_t window_cells,
const uint64_t n_valid_cells, uint64_t n_channels,
const __fp16 *
const *
const inptrs, __fp16 *outptr);
36 struct sme_fp16_nhwc_avg_generic_depthfirst : IGenericDepthfirstStrategy<__fp16, __fp16>
38 using Parent = IGenericDepthfirstStrategy<__fp16, __fp16>;
39 sme_fp16_nhwc_avg_generic_depthfirst(
const CPUInfo *) {}
40 typename Parent::KernelType get_kernel(
void)
const override {
return sme_fp16_nhwc_avg_generic_depthfirst_impl; }
46 #endif // defined(ARM_COMPUTE_ENABLE_SME) && defined(__ARM_FP16_ARGS)