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

Function get_nchw88_args

dnn/test/common/resize.h:137–154  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

135}
136
137static inline std::vector<TestArg> get_nchw88_args() {
138 std::vector<TestArg> args;
139
140 param::Resize param;
141 param.format = param::Resize::Format::NCHW88;
142 param.imode = param::Resize::InterpolationMode::LINEAR;
143 rep(n, 4ul) rep(c, 4ul) rep(ih, 4ul) rep(iw, 4ul) rep(oh, 4ul) rep(ow, 4ul)
144 args.emplace_back(
145 param, TensorShape{n + 1ul, c + 1ul, ih + 1ul, iw + 1ul, 8ul},
146 TensorShape{n + 1ul, c + 1ul, oh + 1ul, ow + 1ul, 8ul});
147
148 param.imode = param::Resize::InterpolationMode::NEAREST;
149 rep(n, 4ul) rep(c, 4ul) rep(ih, 4ul) rep(iw, 4ul) rep(oh, 4ul) rep(ow, 4ul)
150 args.emplace_back(
151 param, TensorShape{n + 1ul, c + 1ul, ih + 1ul, iw + 1ul, 8ul},
152 TensorShape{n + 1ul, c + 1ul, oh + 1ul, ow + 1ul, 8ul});
153 return args;
154}
155
156static inline std::vector<TestArg> get_cv_args() {
157 std::vector<TestArg> args;

Callers 1

TEST_FFunction · 0.85

Calls 2

emplace_backMethod · 0.80
repFunction · 0.70

Tested by

no test coverage detected