Expr

VARP

class VARP

枚举类

Dimensionformat

enum Dimensionformat {
    NHWC,
    NC4HW4,
    NCHW
};

value	name	说明
0	NHWC
1	NC4HW4
2	NCHW

---

InputType

enum InputType {
    INPUT = 0,
    CONSTANT = 1,
    TRAINABLE = 2,
};

value	name	说明
0	INPUT	默认输入变量
1	CONSTANT	常量
2	TRAINABLE	可训练变量

成员函数

---

VARP

VARP(std::shared_ptr<Variable> c) {
    mContent = std::move(c);
};

调用移动构造函数，将c的对象移动至mContent对象

参数：

• c Variable类型变量

返回：void

---

VARP

VARP(Variable* c) {
    mContent.reset(c);
};

重置mContent对象中的Variable对象

参数：

• c Variable类型变量

返回：void

---

get

Variable* get() const  {
    return mContent.get();
};

获取mContent的Variable对象

参数：无

返回：Variable对象

---

VARP

VARP(const VARP& var) {
    mContent = var.mContent;
};

把var的content对象赋值给mContent变量

参数：

• var 输入变量

返回：void

---

VARP

VARP(VARP&& var) {
    mContent = std::move(var.mContent);
};

调用移动构造函数，将var.mContent的对象移动至mContent对象

参数：

• var 输入变量

返回：void

---

operator+

VARP operator+(VARP var) const;

VARP类型对象加法计算

参数：

• var 输入变量

返回：VARP对象

---

operator-

VARP operator-(VARP var) const;

VARP类型对象减法计算

参数：

• var 输入变量

返回：VARP对象

---

operator*

VARP operator*(VARP var) const;

VARP类型对象乘法计算

参数：

• var 输入变量

返回：VARP对象

---

operator/

VARP operator/(VARP var) const;

VARP类型对象除法计算

参数：

• var 输入变量

返回：VARP对象

---

mean

VARP mean(INTS dims) const;

计算均值

参数：

• dims 一个向量

返回：VARP对象

---

sum

VARP sum(INTS dims) const;

计算和

参数：

• dims 一个向量

返回：VARP对象

---

operator==

bool operator==(const VARP& var) const {
    return var.mContent == mContent;
};

重载相等运算符，判断var.mContent和mContent是否相等

参数：

• var 输入变量

返回：true/false

---

operator<

bool operator<(const VARP& var) const {
    return mContent < var.mContent;
};

重载小于运算符，判断mContent是否小于var.mContent

参数：

• var 输入变量

返回：true/false

---

operator<=

bool operator<=(const VARP& var) const {
    return mContent <= var.mContent;
};

重载小于等于运算符，判断mContent是否小于等于var.mContent

参数：

• var 输入变量

返回：true/false

---

operator=

VARP& operator=(const VARP& var) {
    mContent = var.mContent;
    return *this;
};

拷贝var.mContent对象到mContent

参数：

• var 输入变量

返回：当前对象的拷贝

---

operator=

VARP& operator=(Variable* var) {
    mContent.reset(var);
    return *this;
};

重置mContent对象中的Variable对象，并拷贝

参数：

• var 输入变量

返回：当前对象的拷贝

---

operator->

Variable* operator->() const  {
    return mContent.get();
};

获取mContent对象的值

参数：无

返回：Variable对象

---

fix

bool fix(InputType type) const;

朝零方向取整

参数：

• type 输入数据类型

返回：true/false

---

operator==

inline bool operator==(Variable* src, VARP dst) {
    return src == dst.get();
};

重载相等运算符，判断src和dst.get()是否相等

参数：

• src Variable类型输入变量
• dst VARP类型输入变量

返回：true/false

---

operator!=

inline bool operator!=(Variable* src, VARP dst) {
    return src != dst.get();
};

重载不相等运算符，判断src和dst.get()是否不相等

参数：

• src Variable类型输入变量
• dst VARP类型输入变量

返回：true/false

Variable

class Variable

成员函数

---

name

const std::string& name() const;

获取Variable对象的名称

参数：无

返回：名称

---

setName

void setName(const std::string& name);

设置名称

参数：

• name 名称

返回：void

---

expr

std::pair<EXPRP, int> expr() const {
    return std::make_pair(mFrom, mFromIndex);
};

创建一个EXPRP对象，需要mFrom作为参数，位置为mFromIndex

参数：无

返回：EXPRP对象

---

getInfo

const Info* getInfo();

获取Variable对象的相关信息

参数：无

返回：如果计算信息错误，返回nullptr

---

resize

bool resize(INTS dims);

调整Variable对象的大小

参数：

• dims 一个向量

返回：true/false

---

readMap

template <typename T>
const T* readMap() {
    return (const T*)readInternal();
};

读取内部信息

参数：无

返回：信息

---

writeMap

template <typename T>
T* writeMap() {
    return (T*)writeInternal();
};

写入内部信息

参数：无

返回：信息

---

input

bool input(VARP src);

输入信息

参数：

• src 输入变量

返回：true/false

---

replace

static void replace(VARP dst, VARP src);

替换信息

参数：

• dst 输入变量
• src 输入变量

返回：void

---

create

static VARP create(EXPRP expr, int index = 0);

在index位置创建expr对象

参数：

• expr 输入变量
• index 位置下标，默认为0

返回：VARP对象

---

load

static std::vector<VARP> load(const char* fileName);

通过文件名加载对象

参数：

• fileName 文件名

返回：VARP对象矩阵

---

loadMap

static std::map<std::string, VARP> loadMap(const char* fileName);

通过文件名读取模型对象

参数：

• fileName 文件名

返回：模型对象

---

load

static std::vector<VARP> load(const uint8_t* buffer, size_t length);

加载存储的，长度为length的模型对象

参数：

• buffer 存储数据
• length 数据长度

返回：模型对象

---

loadMap

static std::map<std::string, VARP> loadMap(const uint8_t* buffer, size_t length);

读取存储的，长度为length的模型对象

参数：

• buffer 存储数据
• length 数据长度

返回：模型对象

---

getInputAndOutput

static std::pair<std::map<std::string, VARP>, std::map<std::string, VARP>> getInputAndOutput(const std::map<std::string, VARP>& allVariable);

获取模型输入输出节点

参数：

• allVariable 输入变量

返回：模型输入输出节点

---

mapToSequence

static std::vector<VARP> mapToSequence(const std::map<std::string, VARP>& source);

模型的输出节点及其名称

参数：

• source 输入变量

返回：输出节点及其名称

---

getExecuteOrder

static std::vector<EXPRP> getExecuteOrder(const std::vector<VARP>& output);

获取操作指令

参数：

• output 输入变量

返回：指令集

---

save

static void save(const std::vector<VARP>& vars, const char* fileName);

保存vars到指定位置

参数：

• vars 输入变量
• fileName 文件名

返回：void

---

save

static std::vector<int8_t> save(const std::vector<VARP>& vars);

保存vars到默认位置

参数：

• vars 输入变量

返回：void

---

save

static void save(const std::vector<VARP>& vars, NetT* dest);

保存vars到dest位置

参数：

• vars 输入变量
• dest 目标地址

返回：void

---

prepareCompute

static void prepareCompute(const std::vector<VARP>& vars, bool forceCPU = false);

将几个变量打包在一个管道中进行计算

参数：

• vars 输入变量
• forceCPU 是否强制使用CPU，默认为false

返回：void

---

compute

static void compute(const std::vector<VARP>& vars, bool forceCPU = false);

计算变量

参数：

• vars 输入变量
• forceCPU 是否强制使用CPU，默认为false

返回：void

---

linkNumber

size_t linkNumber() const;

获取输出信息的size

参数：无

返回：size

---

toExprs

const std::vector<WeakEXPRP>& toExprs() const;

返回模型对象信息

参数：无

返回：模型对象信息

---

setExpr

void setExpr(EXPRP expr, int index) {
    mFrom = expr;
    mFromIndex = index;
};

在index位置设置EXPRP对象

参数：无

返回：EXPRP对象

Expr

class Expr

枚举类

---

MemoryType

enum MemoryType {
    COPY,
    MOVE,
    REF
};

value	name	说明
0	COPY	拷贝
1	MOVE	移动
2	REF	引用

成员函数

---

create

static EXPRP create(Tensor* tensor, bool own = false);

创建一个包含tensor的EXPRP变量

参数：

• tensor 输入变量
• own 默认为false

返回：EXPRP变量

---

create

static EXPRP create(Variable::Info&& info, const void* ptr, VARP::InputType type, MemoryType copy = COPY);

创建一个包含info对象的EXPRP变量

参数：

• info Variable类型输入变量
• ptr 目标对象地址
• type 输入数据类型
• copy 内存拷贝

返回：EXPRP变量

---

create

static EXPRP create(const OpT* op, std::vector<VARP> inputs, int outputSize = 1);

创建EXPRP变量

参数：

• op 输入变量
• inputs 输入变量
• outputSize 输出信息大小，默认为1

返回：EXPRP变量

---

create

static EXPRP create(std::shared_ptr<BufferStorage> extra, std::vector<VARP>&& inputs, int outputSize = 1);

创建EXPRP变量

参数：

• extra 输入变量
• inputs 输入变量
• outputSize 输出信息大小，默认为1

返回：EXPRP变量

---

create

static EXPRP create(std::unique_ptr<OpT>&& op, std::vector<VARP> inputs, int outputSize = 1) {
    return create(op.get(), inputs, outputSize);
};

创建EXPRP变量

参数：

• op 输入变量
• inputs 输入变量
• outputSize 输出信息大小，默认为1

返回：EXPRP变量

---

setName

void setName(const std::string& name);

设置名称

参数：

• name 名称

返回：void

---

get

const Op* get() const {
    return mOp;
};

获取对象信息

参数：无

返回：对象信息

---

inputs

const std::vector<VARP>& inputs() const {
    return mInputs;
};

获取输入节点信息

参数：无

返回：输入节点信息

---

outputSize

int outputSize() const {
    return (int)mOutputNames.size();
};

返回输出节点信息大小

参数：无

返回：size

---

replace

static void replace(EXPRP oldExpr, EXPRP newExpr);

用newExpr替换oldExpr

参数：

• oldExpr 输入变量，源对象
• newExpr 输入变量，目标对象

返回：void

---

requireInfo

bool requireInfo();

获取需要的信息

参数：无

返回：信息

---

visitOutputs

void visitOutputs(const std::function<bool(EXPRP, int)>& visit);

访问输出节点的信息

参数：

• visit 访问方法

返回：void

---

visit

static void visit(EXPRP expr, const std::function<bool(EXPRP)>& before, const std::function<bool(EXPRP)>& after);

访问某一个范围的信息

参数：

• expr 输入变量
• before before指针
• after after指针

返回：void

---

outputs

const std::vector<WeakEXPRP>& outputs() const {
    return mTo;
};

返回输出节点信息

参数：无

返回：信息

---

~Expr()

析构函数

---

visited

bool visited() const {
    return mVisited;
};

是否已经访问过

参数：无

返回：true/false

---

setVisited

void setVisited(bool visited) {
    mVisited = visited;
};

设置已经被访问

参数：

• visited 是否访问

返回：void

---

name

const std::string& name() const {
    return mName;
};

获取名称

参数：无

返回：名称

---

outputName

const std::string& outputName(int index) {
    return mOutputNames[index];
};

输出指定index的名称

参数：

• index 下标

返回：名称

---

inputType

VARP::InputType inputType() const {return mType;};

返回当前输入类型

参数：无

返回：输入类型

---

outputInfo

Variable::Info* outputInfo(int index) const;

返回指定下标的输出信息

参数：

• index 下标值

返回：输出信息

---

extra

std::shared_ptr<BufferStorage> extra() const {
    return mStorage;
};

返回附加信息

参数：无

返回：附加信息

---

inside

std::shared_ptr<Inside> inside() const {
    return mInside;
};

返回内部信息

参数：无

返回：内部信息

---

valid

bool valid() const {
    return mValid;
};

是否有效

参数：无

返回：true/false