|
|
|
@ -69,7 +69,9 @@ void OpProtoAndCheckerMaker::operator()(proto::OpProto* proto,
|
|
|
|
|
static_cast<int>(OpRole::kOptimize),
|
|
|
|
|
static_cast<int>(OpRole::kLoss) | static_cast<int>(OpRole::kForward),
|
|
|
|
|
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(),
|
|
|
|
|
"Optimized for variable")
|
|
|
|
|
.SetDefault({});
|
|
|
|
|