mnn-sys 0.1.2

Rust bindings for MNN, a lightweight deep neural network inference engine.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//
//  ModuleInside.hpp
//  MNN
//
//  Created by MNN on b'2025/01/13'.
//  Copyright © 2018, Alibaba Group Holding Limited
//

#ifndef ModuleInside_hpp
#define ModuleInside_hpp
#include <unordered_map>
#include <MNN/expr/Module.hpp>
#include <MNN/expr/Executor.hpp>
namespace MNN {
namespace Express {
class Module::CloneContext {
public:
    CloneContext() = default;
    explicit CloneContext(const bool shareParams)
        : mShareParams(shareParams) {}
    virtual ~CloneContext() = default;

    const bool shareParams() const { return mShareParams; }

    EXPRP getOrClone(const EXPRP expr);
    VARP getOrClone(const VARP var);
    std::shared_ptr<Executor::RuntimeManager> pRuntimeManager;
private:
    bool mShareParams = false;
    std::unordered_map<const Expr*, EXPRP> mExprMap;
    std::unordered_map<const Variable*, VARP> mVarMap;
};
};
};
#endif