mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
12 lines
283 B
C++
12 lines
283 B
C++
|
#include <pybind11/pybind11.h>
|
||
|
|
|
||
|
|
#include "../ops/ncnn_ops_register.h"
|
||
|
|
#include "net.h"
|
||
|
|
|
||
|
|
PYBIND11_MODULE(ncnn_ext, m) {
|
||
|
|
m.def(
|
||
|
|
"register_mm_custom_layers",
|
||
|
|
[](ncnn::Net &net) { return register_mm_custom_layers(net); },
|
||
|
|
"register all mmlab custom ncnn layers.");
|
||
|
|
}
|