mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
* Update custom ops namespace * rename register, rename generated library name * rename domain to mmcv
20 lines
380 B
C++
Executable File
20 lines
380 B
C++
Executable File
#ifndef LAYER_EXPAND_H
|
|
#define LAYER_EXPAND_H
|
|
|
|
#include "layer.h"
|
|
|
|
namespace mmdeploy {
|
|
|
|
class Expand : public ncnn::Layer {
|
|
public:
|
|
Expand();
|
|
|
|
virtual int forward(const std::vector<ncnn::Mat>& bottom_blobs,
|
|
std::vector<ncnn::Mat>& top_blobs,
|
|
const ncnn::Option& opt) const;
|
|
};
|
|
|
|
} // namespace mmdeploy
|
|
|
|
#endif // LAYER_EXPAND_H
|