Overview
Describes an input tensor of a subgraph.
- A tensor may have multiple consumers. Therefore, a subgraph input may correspond to multiple node inputs.
tensor1 / \ op1 op2In the preceding example, tensor1 is a subgraph input and corresponds to two nodes op1 and op2 on the boundary. tensor1 is outside the subgraph boundary, and op1 and op2 are inside the subgraph boundary.
- Assume that tensor1 has two consumers, where op1 is inside the subgraph boundary and op2 is outside the subgraph boundary.
Therefore, when describing the subgraph input, it is necessary to specify the node input on the boundary.
Header Files to Be Included
1 | #include <ge/fusion/subgraph_boundary.h> |
Public Member Functions
SubgraphInput() explicit SubgraphInput(std::vector<NodeIo> node_inputs) SubgraphInput(const SubgraphInput &other) noexcept SubgraphInput &operator=(SubgraphInput &&other) noexcept SubgraphInput &operator=(const SubgraphInput &other) noexcept Status AddInput(const NodeIo &node_input) [[nodiscard]] std::vector<NodeIo> GetAllInputs() const
Parent topic: SubgraphInput