27 #if defined(ARM_COMPUTE_ENABLE_SME) && defined(__ARM_FP16_ARGS)
32 void sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst_impl(
unsigned int,
const __fp16 *
const *
const, __fp16 *
const *
const,
bool,
unsigned int,
unsigned int,
unsigned int,
unsigned int);
34 struct sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst :
public DepthfirstStrategy<__fp16, __fp16>
36 using Parent = DepthfirstStrategy<__fp16, __fp16>;
39 const static auto pool_rows = 2u, pool_cols = 2u;
40 const static auto stride_rows = 1u, stride_cols = 1u;
42 sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst(
const CPUInfo *)
43 : Parent(pool_rows, pool_cols, stride_rows, stride_cols, 2, 2) {}
45 Parent::KernelType get_kernel(
void)
const {
return sme_fp16_nhwc_max_2x2_s1_output2x2_depthfirst_impl; }
51 #endif // defined(ARM_COMPUTE_ENABLE_SME) && defined(__ARM_FP16_ARGS)