提交 bcfc8c4e authored 作者: Olivier Breuleux's avatar Olivier Breuleux

merge

...@@ -324,7 +324,8 @@ class Abs(UnaryScalarOp): ...@@ -324,7 +324,8 @@ class Abs(UnaryScalarOp):
def grad(self, (x, ), (gz, )): def grad(self, (x, ), (gz, )):
return gz * sgn(x), return gz * sgn(x),
def c_code(self, (x, ), (z, ), sub): def c_code(self, (x, ), (z, ), sub):
return "%(z)s = abs(%(x)s);" % locals() return "%(z)s = %(prefix)sabs(%(x)s);" \
% dict(locals(), prefix = 'float' in self.inputs[0].dtype and 'f' or '')
class Sgn(UnaryScalarOp): class Sgn(UnaryScalarOp):
def impl(self, x): def impl(self, x):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论