提交 c3210104 authored 作者: Frederic Bastien's avatar Frederic Bastien

small bugfix. The old version was causing an assert when this was not needed.

上级 06700932
......@@ -220,8 +220,8 @@ using namespace std;
elif node.inputs[0].type.dtype=="float64": d["type"]="double"
else: raise Exception("Type %s not implemented"%node.inputs[0].type.dtype)
if self.unroll_batch>0 or self.unroll_kern>0:
if self.unroll_batch<0: self.unroll_batch=1
if self.unroll_kern<0: self.unroll_kern=1
if self.unroll_batch<=0: self.unroll_batch=1
if self.unroll_kern<=0: self.unroll_kern=1
# print "return unrolled batch and kern code by",self.unroll_batch, self.unroll_kern
return gen_conv_code_unroll_batch_kern(d, self.unroll_batch,
self.unroll_kern)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论