mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
ea3078aa56
* Update custom ops namespace * rename register, rename generated library name * rename domain to mmcv
12 lines
294 B
C++
Executable File
12 lines
294 B
C++
Executable File
#include <pybind11/pybind11.h>
|
|
|
|
#include "../ops/ncnn_ops_register.h"
|
|
#include "net.h"
|
|
|
|
PYBIND11_MODULE(ncnn_ext, m) {
|
|
m.def(
|
|
"register_mmdeploy_custom_layers",
|
|
[](ncnn::Net &net) { return register_mmdeploy_custom_layers(net); },
|
|
"register mmdeploy custom ncnn layers.");
|
|
}
|