提交 65af716c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Revert some changes in scalar.

上级 08f3bf01
...@@ -2299,8 +2299,8 @@ class RoundHalfToEven(UnaryScalarOp): ...@@ -2299,8 +2299,8 @@ class RoundHalfToEven(UnaryScalarOp):
(x,) = inputs (x,) = inputs
(z,) = outputs (z,) = outputs
typ = node.outputs[0].type.dtype typ = node.outputs[0].type.dtype
if not typ.startswith('float'): if typ not in ['float32', 'float64']:
raise NotImplementedError("The output should a float") raise NotImplementedError("The output should be float32 or float64")
return dedent(""" return dedent("""
#ifndef ROUNDING_EPSILON #ifndef ROUNDING_EPSILON
...@@ -2398,7 +2398,7 @@ class RoundHalfAwayFromZero(UnaryScalarOp): ...@@ -2398,7 +2398,7 @@ class RoundHalfAwayFromZero(UnaryScalarOp):
if node.outputs[0].type.dtype in ['float32', 'float64']: if node.outputs[0].type.dtype in ['float32', 'float64']:
return "%(z)s = round(%(x)s);" % locals() return "%(z)s = round(%(x)s);" % locals()
else: else:
raise NotImplementedError("The output should be a float") raise NotImplementedError("The output should be float32 or float64")
round_half_away_from_zero = RoundHalfAwayFromZero(same_out_float_only) round_half_away_from_zero = RoundHalfAwayFromZero(same_out_float_only)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论