简介
用于描述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