13 KiB

Raw Blame History

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

13 KiB Raw Blame History Unescape Escape

VARP

枚举类

Dimensionformat

InputType

成员函数

VARP

VARP

get

VARP

VARP

operator+

operator-

operator*

operator/

mean

sum

operator==

operator<

operator<=

operator=

operator=

operator->

fix

operator==

operator!=

Variable

成员函数

name

setName

expr

getInfo

resize

readMap

writeMap

input

replace

create

load

loadMap

load

loadMap

getInputAndOutput

mapToSequence

getExecuteOrder

save

save

save

prepareCompute

compute

linkNumber

toExprs

setExpr

Expr

枚举类

MemoryType

成员函数

create

create

create

create

create

setName

get

inputs

outputSize

replace

requireInfo

visitOutputs

visit

outputs

~Expr()

visited

setVisited

name

outputName

inputType

outputInfo

extra

inside

13 KiB

Raw Blame History