提交 b887bd78 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Align numba reciprocal with C backend

上级 4e3b5bed
......@@ -258,11 +258,10 @@ def numba_funcify_Second(op, node, **kwargs):
def numba_funcify_Reciprocal(op, node, **kwargs):
@numba_basic.numba_njit
def reciprocal(x):
# TODO FIXME: This isn't really the behavior or `numpy.reciprocal` when
# `x` is an `int`
return 1 / x
# This is how the C-backend implementation works
return np.divide(np.float32(1.0), x)
return reciprocal, scalar_op_cache_key(op)
return reciprocal, scalar_op_cache_key(op, cache_version=1)
@register_funcify_and_cache_key(Sigmoid)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论