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

More readable names for numba set_subtensor funcs

上级 d878e6a9
...@@ -77,10 +77,10 @@ def numba_funcify_default_subtensor(op, node, **kwargs): ...@@ -77,10 +77,10 @@ def numba_funcify_default_subtensor(op, node, **kwargs):
y_name = input_names[1] y_name = input_names[1]
if op.set_instead_of_inc: if op.set_instead_of_inc:
function_name = "setsubtensor" function_name = "set_subtensor"
index_body = f"z[indices] = {y_name}" index_body = f"z[indices] = {y_name}"
else: else:
function_name = "incsubtensor" function_name = "inc_subtensor"
index_body = f"z[indices] += {y_name}" index_body = f"z[indices] += {y_name}"
else: else:
function_name = "subtensor" function_name = "subtensor"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论