提交 4d9afd30 authored 作者: Frederic's avatar Frederic

Make meta_conv try more case due to broadcasting problems

上级 c907bf60
...@@ -857,7 +857,9 @@ class LocalMetaOptimizer(LocalOptimizer): ...@@ -857,7 +857,9 @@ class LocalMetaOptimizer(LocalOptimizer):
elif hasattr(input.tag, 'test_value'): elif hasattr(input.tag, 'test_value'):
givens[input] = theano.shared( givens[input] = theano.shared(
input.type.filter(input.tag.test_value), input.type.filter(input.tag.test_value),
input.name, borrow=True) input.name,
broadcastable=input.broadcastable,
borrow=True)
else: else:
missing.add(input) missing.add(input)
if missing: if missing:
......
...@@ -1413,7 +1413,9 @@ class ConvMetaOptimizer(LocalCudaMetaOptimizer): ...@@ -1413,7 +1413,9 @@ class ConvMetaOptimizer(LocalCudaMetaOptimizer):
# allow_downcast=True), # allow_downcast=True),
numpy.require(numpy.random.randn(*shape), numpy.require(numpy.random.randn(*shape),
dtype=var.dtype), dtype=var.dtype),
var.name, borrow=True) var.name,
broadcastable=var.broadcastable,
borrow=True)
# return mapping # return mapping
return result return result
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论