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