提交 273912a6 authored 作者: notoraptor's avatar notoraptor

Just use bool type for op param AdvancedIncSubtensor1.set_instead_of_inc.

上级 2378dd6d
......@@ -1875,11 +1875,11 @@ class AdvancedIncSubtensor1(Op):
__props__ = ('inplace', 'set_instead_of_inc')
check_input = False
params_type = ParamsType(inplace=scal.bool,
set_instead_of_inc=scal.int8)
set_instead_of_inc=scal.bool)
def __init__(self, inplace=False, set_instead_of_inc=False):
self.inplace = bool(inplace)
self.set_instead_of_inc = int(bool(set_instead_of_inc))
self.set_instead_of_inc = bool(set_instead_of_inc)
if inplace:
self.destroy_map = {0: [0]}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论