简介
用于描述Subgraph的一个输入Tensor。
- 考虑一个Tensor可能有多个消费者,因此一个Subgraph input可能对应多个Node input。
tensor1 / \ op1 op2如上示例,tensor1是一个Subgraph的input,其在边界上对应op1和op2两个node。tensor1在Subgraph边界外,op1和op2在Subgraph边界内。
需要包含的头文件
1 | #include <ge/fusion/subgraph_boundary.h> |
Public成员函数
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
父主题: SubgraphInput