提交 0478780c 编写于 作者: D dzhwinter 提交者: GitHub

Merge pull request #3166 from dzhwinter/no_gradient

"add no gradient"
...@@ -400,6 +400,14 @@ class GradOpRegisterHelper { ...@@ -400,6 +400,14 @@ class GradOpRegisterHelper {
return 0; \ return 0; \
} }
/**
* Macro to Forbid user register Gradient Operator.
*/
#define NO_GRADIENT(__op_type) \
STATIC_ASSERT_GLOBAL_NAMESPACE( \
__reg_gradient_op__##__op_type##__op_type##_grad, \
"NO_GRADIENT must be in global namespace")
/** /**
* Macro to Register OperatorKernel. * Macro to Register OperatorKernel.
*/ */
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册