SetOpNameType

Function

Sets the operator name and type.

Prototype

1
ContextBuilder &SetOpNameType(const std::string& opName, const std::string& opType)

Parameters

Parameter

Input/Output

Description

opName

Input

Operator name.

opType

Input

Operator type.

Returns

Object of the current ContextBuilder

Constraints

None

Examples

1
2
3
4
std::string opName = "tmpNode";
std::string opType = "FlashAttentionScore";
context_ascendc::ContextBuilder builder;
(void)builder.SetOpNameType(opName, opType);