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