Node

Description

Operation node in a graph operator. Each node indicates an Operation or GraphOperation, and all nodes together form a complete graph operator.

Definition

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

Parameters

Member	Data Type	Default Value	Description
operation	Operation	nullptr	Operation or GraphOperation corresponding to a node.
inTensorIds	SVector< uint32_t >	-	Input tensor ID SVector of the Operation or GraphOperation corresponding to a node.
outTensorIds	SVector< uint32_t >	-	Output tensor ID SVector of the Operation or GraphOperation corresponding to a node.
inTensorReshapeFuncs	SVector< ReshapeFunc >	-	Reshape function SVector of each input tensor of the Operation or GraphOperation corresponding to a node
inTensorChunks	SVector< Chunk >	-	Stores chunks.

inTensorIds, outTensorIds, and inTensorReshapeFuncs are all SVectors. The sequence of each element must be the same as that of the corresponding tensor.

Parent topic: atb/types.h