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