MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / TEST_F

Function TEST_F

dnn/test/arm_common/pooling.cpp:12–59  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

10namespace test {
11
12TEST_F(ARM_COMMON, POOLING) {
13 using Param = param::Pooling;
14 // clang-format off
15 for (size_t ih: {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
16 for (size_t iw: {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
17 for (size_t p: {1, 2})
18 {
19 Param param;
20 param.mode = Param::Mode::MAX;
21 param.window_h = param.window_w = 3;
22 param.stride_h = param.stride_w = 2;
23 param.pad_h = param.pad_w = p;
24 Checker<Pooling> checker(handle());
25 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
26
27 param.mode = Param::Mode::AVERAGE;
28 param.window_h = param.window_w = 3;
29 param.stride_h = param.stride_w = 2;
30 param.pad_h = param.pad_w = p;
31 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
32
33 param.mode = Param::Mode::MAX;
34 param.window_h = param.window_w = 4;
35 param.stride_h = param.stride_w = 2;
36 param.pad_h = param.pad_w = p;
37 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
38
39 param.mode = Param::Mode::MAX;
40 param.window_h = param.window_w = 5;
41 param.stride_h = param.stride_w = 2;
42 param.pad_h = param.pad_w = p;
43 if (ih + p * 2 >= 5 && iw + p * 2 >= 5)
44 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
45 }
46 for (size_t ih: {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
47 for (size_t iw: {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
48 for (size_t p: {1, 2})
49 {
50 Param param;
51 param.mode = Param::Mode::AVERAGE_COUNT_EXCLUDE_PADDING;
52 param.window_h = param.window_w = 3;
53 param.stride_h = param.stride_w = 1;
54 param.pad_h = param.pad_w = p;
55 Checker<Pooling> checker(handle());
56 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
57 }
58 // clang-format on
59}
60
61TEST_F(ARM_COMMON, POOLING_RECORD) {
62 using Param = param::Pooling;

Callers

nothing calls this directly

Calls 12

benchmark_nchw88_fp16Function · 0.85
set_dtypeMethod · 0.80
emplace_backMethod · 0.80
benchmark_nchw44_fp32Function · 0.70
runFunction · 0.50
execMethod · 0.45
enumvMethod · 0.45
getMethod · 0.45
execlMethod · 0.45
paramMethod · 0.45
deduce_layoutMethod · 0.45
total_nr_elemsMethod · 0.45

Tested by

no test coverage detected