|
|
@ -27,7 +27,7 @@ namespace operators {
|
|
|
|
AddInput("X", "Input of " #OP_NAME " operator"); \
|
|
|
|
AddInput("X", "Input of " #OP_NAME " operator"); \
|
|
|
|
AddOutput("Out", "Output of " #OP_NAME " operator").Reuse("X"); \
|
|
|
|
AddOutput("Out", "Output of " #OP_NAME " operator").Reuse("X"); \
|
|
|
|
AddAttr<bool>("use_mkldnn", \
|
|
|
|
AddAttr<bool>("use_mkldnn", \
|
|
|
|
"(bool, default false) Only used in mkldnn kernel") \
|
|
|
|
"(default false) Only used in mkldnn kernel") \
|
|
|
|
.SetDefault(false); \
|
|
|
|
.SetDefault(false); \
|
|
|
|
AddComment(OP_COMMENT); \
|
|
|
|
AddComment(OP_COMMENT); \
|
|
|
|
} \
|
|
|
|
} \
|
|
|
|