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

Make meta_conv try more case due to broadcasting problems

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