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   op2

    In 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