Node

功能

图算子中的Operation节点,每个Node表示一个Operation或者GraphOperation,所有的Node组成一个完整的图算子。

定义

1
2
3
4
5
6
7
struct Node {
    Operation* operation = nullptr;
    SVector<uint32_t> inTensorIds;
    SVector<uint32_t> outTensorIds;
    SVector<ReshapeFunc> inTensorReshapeFuncs;
    SVector<Chunk> inTensorChunks;
};

参数列表

成员名称

数据类型

默认值

描述

operation

Operation

nullptr

Node对应的operation或者graphOperation。

inTensorIds

SVector< uint32_t >

-

Node对应的operation或者graphOperation的输入tensorId SVector。

outTensorIds

SVector< uint32_t >

-

Node对应的operation或者graphOperation的输出tensorId SVector。

inTensorReshapeFuncs

SVector< ReshapeFunc >

-

Node对应的operation或者graphOperation的每个输入tensor的reshape函数SVector。

inTensorChunks

SVector< Chunk >

-

存放chunk。

inTensorIds、outTensorIds和inTensorReshapeFuncs均为SVector,每个元素的顺序需要和对应Tensor的顺序保持一致。