基于算子的IR定义,声明实例化时每种输入的实际个数
ContextBuilder &IrInstanceNum(std::vector<uint32_t> instanceNum);
参数 |
输入/输出 |
说明 |
---|---|---|
instanceNum |
输入 |
基于算子IR原型定义,按照输入的index顺序声明实例化个数。 |
当前ContextBuilder的对象
必须配合NodeIoNum一同使用。
context_ascendc::ContextBuilder builder; (void)builder.NodeIoNum(5,3) // 算子有5种输入,3种输出 .IrInstanceNum({1, 1, 3, 1, 1}); // 算子实例化时,index为2的动态类型输入tensor有3个实例