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

Function run_xcorr_single_channel

dnn/src/fallback/convolution/run_conv.cpp:213–228  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

211}
212
213void run_xcorr_single_channel(
214 const float* src, const float* filter, float* dst, size_t IH, size_t IW,
215 size_t FH, size_t FW, size_t OH, size_t OW, size_t PH, size_t PW, size_t SH,
216 size_t SW, bool add_mode) {
217 if (can_run_xcorr_single_channel_templated(
218 IH, IW, FH, FW, OH, OW, PH, PW, SH, SW)) {
219 run_xcorr_single_channel_templated(
220 src, filter, dst, IH, IW, FH, FW, OH, OW, PH, PW, SH, SW, add_mode);
221 } else {
222 MIDOUT_BEGIN(megdnn_fallback_conv, void) {
223 run_xcorr_single_channel_nontemplated(
224 src, filter, dst, IH, IW, FH, FW, OH, OW, PH, PW, SH, SW, add_mode);
225 }
226 MIDOUT_END();
227 }
228}
229
230/*================ ConvolutionBackwardData =============*/
231

Callers 1

run_convFunction · 0.85

Tested by

no test coverage detected