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