Node

说明

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

定义

struct Node {
    Operation* operation = nullptr;
    SVector<uint32_t> inTensorIds;
    SVector<uint32_t> outTensorIds;
    SVector<ReshapeFunc> inTensorReshapeFuncs;
};

成员

成员名称	描述
operation	Node对应的operation或者graphOperation。
inTensorIds	Node对应的operation或者graphOperation的输入tensorId SVector。
outTensorIds	Node对应的operation或者graphOperation的输出tensorId SVector。
inTensorReshapeFuncs	Node对应的operation或者graphOperation的每个输入Tensor的reshape函数SVector。

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

父主题： atb/types.h