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

Function get_nchw44_args

dnn/test/common/resize.h:118–135  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

116}
117
118static inline std::vector<TestArg> get_nchw44_args() {
119 std::vector<TestArg> args;
120
121 param::Resize param;
122 param.format = param::Resize::Format::NCHW44;
123 param.imode = param::Resize::InterpolationMode::LINEAR;
124 rep(n, 4ul) rep(c, 4ul) rep(ih, 4ul) rep(iw, 4ul) rep(oh, 4ul) rep(ow, 4ul)
125 args.emplace_back(
126 param, TensorShape{n + 1ul, c + 1ul, ih + 1ul, iw + 1ul, 4ul},
127 TensorShape{n + 1ul, c + 1ul, oh + 1ul, ow + 1ul, 4ul});
128
129 param.imode = param::Resize::InterpolationMode::NEAREST;
130 rep(n, 4ul) rep(c, 4ul) rep(ih, 4ul) rep(iw, 4ul) rep(oh, 4ul) rep(ow, 4ul)
131 args.emplace_back(
132 param, TensorShape{n + 1ul, c + 1ul, ih + 1ul, iw + 1ul, 4ul},
133 TensorShape{n + 1ul, c + 1ul, oh + 1ul, ow + 1ul, 4ul});
134 return args;
135}
136
137static inline std::vector<TestArg> get_nchw88_args() {
138 std::vector<TestArg> args;

Callers 2

TEST_FFunction · 0.85
TEST_FFunction · 0.85

Calls 2

emplace_backMethod · 0.80
repFunction · 0.70

Tested by

no test coverage detected