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

Function kern_naive

dnn/src/fallback/convolution/algos.h:58–64  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

56
57template <typename ftype, typename dtype, typename gtype>
58void kern_naive(const ConvolutionBackwardDataImpl::NCBKernParam& p) {
59 TensorND diff(const_cast<void*>(p.diff_ptr.get_ptr()), p.diff_layout),
60 filter(const_cast<void*>(p.filter_ptr.get_ptr()), p.filter_layout),
61 grad(p.grad_ptr.get_ptr(), p.grad_layout);
62 naive::convolution::backward_data<ftype, dtype, gtype>(
63 filter, diff, grad, p.filter_meta);
64}
65
66class ConvolutionImpl::AlgoFallback final : public AlgoBase {
67public:

Callers 2

cbFunction · 0.50
execMethod · 0.50

Calls 1

get_ptrMethod · 0.80

Tested by

no test coverage detected