SetOpNameType
功能说明
设置算子的名字与类型
函数原型
1 | ContextBuilder &SetOpNameType(const std::string& opName, const std::string& opType) |
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
opName |
输入 |
算子名字 |
opType |
输入 |
算子类型 |
返回值说明
当前ContextBuilder的对象。
约束说明
无
调用示例
1 2 3 4 | std::string opName = "tmpNode"; std::string opType = "FlashAttentionScore"; context_ascendc::ContextBuilder builder; (void)builder.SetOpNameType(opName, opType); |
父主题： ContextBuilder