Node

功能说明

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

定义

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的顺序保持一致。

父主题： atb/types.h