更新时间:2021-03-18 GMT+08:00
分享

AINNNodeRegisterar

NN Node注册类。AINNNodeRegisterar类在ai_nn_node.h中定义。

    class AINNNodeRegisterar
    {
    public:
        AINNNodeRegisterar(const AINNNodeDescription &nnnode_desc, AINNNODE_CREATE_FUN create_func)
        {
            AINNNodeFactory::GetInstance()->RegisterNNNodeCreator(nnnode_desc, create_func);
            nnnode_desc_ = nnnode_desc;
            nnnode_str_.erase(nnnode_str_.begin(),nnnode_str_.end());
        }

        ~AINNNodeRegisterar(){
                AINNNodeFactory::GetInstance()->UnRegisterNNNode(nnnode_desc_);
        }
    private:
        AINNNodeDescription nnnode_desc_;
        string nnnode_str_;
    };
分享:

    相关文档

    相关产品