提交 a75f4c8a authored 作者: Michael Osthege's avatar Michael Osthege 提交者: Brandon T. Willard

Pass args to `Scan`'s `typeConstructor` arg positionally

As advertised by the type hint.
上级 f30e7159
......@@ -752,9 +752,7 @@ class Scan(Op, ScanMethodsMixin, HasInnerGraph):
# output sequence
o = outputs[idx]
self.output_types.append(
typeConstructor(
shape=(False,) + o.type.broadcastable, dtype=o.type.dtype
)
typeConstructor((False,) + o.type.broadcastable, o.type.dtype)
)
idx += len(self.mit_mot_out_slices[jdx])
......@@ -765,9 +763,7 @@ class Scan(Op, ScanMethodsMixin, HasInnerGraph):
for o in outputs[idx:end]:
self.output_types.append(
typeConstructor(
shape=(False,) + o.type.broadcastable, dtype=o.type.dtype
)
typeConstructor((False,) + o.type.broadcastable, o.type.dtype)
)
# shared outputs + possibly the ending condition
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论