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

Function get_opr

dnn/src/cuda/group_local/forward/opr_impl.cpp:18–22  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

16namespace {
17
18std::unique_ptr<LocalForward> get_opr(Handle* handle, param::Convolution param) {
19 auto&& opr = handle->create_operator<LocalForward>();
20 opr->param() = param;
21 return std::move(opr);
22}
23
24TensorLayout prepare_src_dst(const TensorLayout& input, size_t g) {
25 TensorLayout ret = input;

Callers 2

execMethod · 0.85

Calls 1

paramMethod · 0.45

Tested by

no test coverage detected