声明算子定义的输入与输出个数
ContextBuilder &NodeIoNum(size_t inputNum, size_t outputNum);
参数 |
输入/输出 |
说明 |
|---|---|---|
inputNum |
输入 |
算子IR原型定义中的输入个数 |
outputNum |
输入 |
算子IR原型定义中的输出个数 |
当前ContextBuilder的对象
必须配合IrInstanceNum一同使用。
auto builder = ContextBuilder().NodeIoNum(5, 3); // 该算子有5个输入，3个输出