昇腾社区首页
中文
注册
开发者
下载

简介

用于描述subgraph的一个输入tensor。

  • 考虑一个tensor可能有多个消费者,因此一个subgraph input可能对应多个node input
           tensor1
            /  \
          op1   op2

    如上示例,tensor1是一个subgraph的input, 其在边界上对应op1和op2两个node。tensor1在subgraph边界外,op1和op2在subgraph边界内。

  • 再考虑tensor1有2个消费者,其中op1在subgraph边界内,op2在边界外。

    因此当描述subgraph input的时候,明确边界上的node input是有必要的

需要包含的头文件

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