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

Function prepare_sub_opr

dnn/src/cuda/convolution3d/forward/group_conv.cpp:37–46  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

35}
36
37std::pair<TensorLayoutArray, std::unique_ptr<Convolution3DForward>> prepare_sub_opr(
38 const Convolution3DForwardImpl::AlgoBase::SizeArgs& args) {
39 auto conv3d_opr = args.handle->create_operator<Convolution3D>();
40 set_execution_policy<Convolution3DForward, Convolution3DForward*>(
41 args.opr, conv3d_opr.get());
42 auto&& config = sub_opr_config(args);
43 conv3d_opr->param() = config.second;
44
45 return {config.first, std::move(conv3d_opr)};
46}
47} // namespace
48
49std::vector<Algorithm::SearchItem> Convolution3DForwardImpl::AlgoGroupConvGeneral::

Callers 3

is_availableMethod · 0.70
get_workspace_bundleMethod · 0.70
execMethod · 0.70

Calls 3

sub_opr_configFunction · 0.70
getMethod · 0.45
paramMethod · 0.45

Tested by

no test coverage detected