|
|
@ -24,12 +24,12 @@ namespace operators {
|
|
|
|
: public ::paddle::framework::OpProtoAndCheckerMaker { \
|
|
|
|
: public ::paddle::framework::OpProtoAndCheckerMaker { \
|
|
|
|
public: \
|
|
|
|
public: \
|
|
|
|
void Make() override { \
|
|
|
|
void Make() override { \
|
|
|
|
AddInput("X", "Input of " #OP_NAME "operator"); \
|
|
|
|
AddInput("X", "Input of " #OP_NAME " operator"); \
|
|
|
|
AddOutput("Out", "Output of" #OP_NAME "operator"); \
|
|
|
|
AddOutput("Out", "Output of " #OP_NAME " operator"); \
|
|
|
|
AddAttr<bool>("use_mkldnn", \
|
|
|
|
AddAttr<bool>("use_mkldnn", \
|
|
|
|
"(bool, default false) Only used in mkldnn kernel") \
|
|
|
|
"(bool, default false) Only used in mkldnn kernel") \
|
|
|
|
.SetDefault(false); \
|
|
|
|
.SetDefault(false); \
|
|
|
|
AddComment(#OP_COMMENT); \
|
|
|
|
AddComment(OP_COMMENT); \
|
|
|
|
} \
|
|
|
|
} \
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|