提交 c9782590 编写于 作者: Y yuyang18

Add default value of op_role

上级 0e5a278c
...@@ -69,7 +69,9 @@ void OpProtoAndCheckerMaker::operator()(proto::OpProto* proto, ...@@ -69,7 +69,9 @@ void OpProtoAndCheckerMaker::operator()(proto::OpProto* proto,
static_cast<int>(OpRole::kOptimize), static_cast<int>(OpRole::kOptimize),
static_cast<int>(OpRole::kLoss) | static_cast<int>(OpRole::kForward), static_cast<int>(OpRole::kLoss) | static_cast<int>(OpRole::kForward),
static_cast<int>(OpRole::kLoss) | static_cast<int>(OpRole::kLoss) |
static_cast<int>(OpRole::kBackward)}); static_cast<int>(OpRole::kBackward),
static_cast<int>(OpRole::kNotSpecified)})
.SetDefault(static_cast<int>(OpRole::kNotSpecified));
AddAttr<std::vector<std::string>>(OpRoleVarAttrName(), AddAttr<std::vector<std::string>>(OpRoleVarAttrName(),
"Optimized for variable") "Optimized for variable")
.SetDefault({}); .SetDefault({});
......
...@@ -26,6 +26,9 @@ enum class OpRole { ...@@ -26,6 +26,9 @@ enum class OpRole {
kOptimize = 0x0002, kOptimize = 0x0002,
kLoss = 0x0100, kLoss = 0x0100,
// The default value of op's role. This should be only used for unittests and
// CreateOp inside a operator.
kNotSpecified = 0x1000,
}; };
// this class not only make proto but also init attribute checkers. // this class not only make proto but also init attribute checkers.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册