提交 5fdf304d authored 作者: Tim Cooijmans's avatar Tim Cooijmans

GpuBatchedDot: add stream_threshold to __props__ to avoid incorrect merges

上级 d2236cff
...@@ -16,7 +16,7 @@ from theano.tensor import as_tensor_variable ...@@ -16,7 +16,7 @@ from theano.tensor import as_tensor_variable
class GpuBatchedDot(GpuOp): class GpuBatchedDot(GpuOp):
__props__ = () __props__ = ("stream_threshold",)
def __init__(self, stream_threshold=128): def __init__(self, stream_threshold=128):
self.stream_threshold = stream_threshold self.stream_threshold = stream_threshold
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论