| 5 | using Mode = PoolingForward::Param::Mode; |
| 6 | #ifdef ENABLE_KERNEL_BENCHMARK |
| 7 | TEST(AARCH64, BenchmarkPoolingNCHW44) { |
| 8 | Benchmarker<PoolingForward> benchmarker(Arch::ARM64); |
| 9 | PoolingForward::Param param; |
| 10 | for (auto mode : {Mode::MAX, Mode::AVERAGE, Mode::AVERAGE_COUNT_EXCLUDE_PADDING}) { |
| 11 | printf("pooling mode=%d\n", mode); |
| 12 | param.mode = mode; |
| 13 | param.pad_h = 1; |
| 14 | param.pad_w = 1; |
| 15 | param.window_h = 3; |
| 16 | param.window_w = 3; |
| 17 | param.stride_h = 2; |
| 18 | param.stride_w = 2; |
| 19 | param.format = PoolingForward::Param::Format::NCHW44; |
| 20 | benchmarker.set_param(param); |
| 21 | benchmarker.execs({{1, 16, 112, 112, 4}, {}}).print(); |
| 22 | } |
| 23 | } |
| 24 | TEST(AARCH64, BenchmarkPoolingNchw44QInt8) { |
| 25 | Benchmarker<PoolingForward> benchmarker(Arch::ARM64); |
| 26 | PoolingForward::Param param; |