设置算子的名字与类型
ContextBuilder &SetOpNameType(const std::string& opName, const std::string& opType);
参数 |
输入/输出 |
说明 |
---|---|---|
opName |
输入 |
算子名字 |
opType |
输入 |
算子类型 |
当前ContextBuilder的对象。
无
std::string opName = "tmpNode"; std::string opType = "FlashAttentionScore"; context_ascendc::ContextBuilder builder; (void)builder.SetOpNameType(opName, opType);