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

Function TEST_F

dnn/test/arm_common/pooling_multi_thread.cpp:16–48  ·  view source on GitHub ↗

mutli threads *********************************/

Source from the content-addressed store, hash-verified

14
15/*********************** mutli threads *********************************/
16TEST_F(ARM_COMMON_MULTI_THREADS, POOLING) {
17 using Param = param::Pooling;
18 for (size_t ih : {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
19 for (size_t iw : {2, 3, 5, 7, 11, 13, 17, 19, 23, 24, 25, 26, 27, 28, 29, 30})
20 for (size_t p : {1, 2}) {
21 Param param;
22 param.mode = Param::Mode::MAX;
23 param.window_h = param.window_w = 3;
24 param.stride_h = param.stride_w = 2;
25 param.pad_h = param.pad_w = p;
26 Checker<Pooling> checker(handle());
27 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
28
29 param.mode = Param::Mode::AVERAGE;
30 param.window_h = param.window_w = 3;
31 param.stride_h = param.stride_w = 2;
32 param.pad_h = param.pad_w = p;
33 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
34
35 param.mode = Param::Mode::MAX;
36 param.window_h = param.window_w = 4;
37 param.stride_h = param.stride_w = 2;
38 param.pad_h = param.pad_w = p;
39 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
40
41 param.mode = Param::Mode::MAX;
42 param.window_h = param.window_w = 5;
43 param.stride_h = param.stride_w = 2;
44 param.pad_h = param.pad_w = p;
45 if (ih + p * 2 >= 5 && iw + p * 2 >= 5)
46 checker.set_param(param).exec({{2, 3, ih, iw}, {}});
47 }
48}
49
50TEST_F(ARM_COMMON_MULTI_THREADS, POOLING_RECORD) {
51 using Param = param::Pooling;

Callers

nothing calls this directly

Calls 7

set_dtypeMethod · 0.80
run_pooling_checkFunction · 0.70
get_nchw44_pool_argsFunction · 0.70
execMethod · 0.45
enumvMethod · 0.45
push_backMethod · 0.45
clearMethod · 0.45

Tested by

no test coverage detected