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

Method set

dnn/src/cuda/convolution/helper.h:33–40  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31 FilterDesc<param::Convolution> filter_desc;
32 ConvDesc conv_desc;
33 void set(
34 const TensorLayout& src, const CanonizedFilterMeta& filter,
35 const TensorLayout& dst, const param::Convolution& param) {
36 src_desc.set(src, param.format);
37 filter_desc.set(filter);
38 dst_desc.set(dst, param.format);
39 conv_desc.set(src.dtype, param, filter.group);
40 }
41};
42
43struct CUDNNBwdDataDescs {

Callers 11

execMethod · 0.45
setMethod · 0.45
setMethod · 0.45
init_descMethod · 0.45
init_descMethod · 0.45
execMethod · 0.45
exec_internalMethod · 0.45
execMethod · 0.45
set_conv_biasMethod · 0.45
set_convMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected