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

make the GpuAlloc work in more case.

上级 5cc01161
......@@ -1835,7 +1835,6 @@ class GpuAlloc(Op):
#the optimizer will remove them.
v = gpu_from_host(tensor.as_tensor_variable(value))
sh = [tensor.as_tensor_variable(s) for s in shape]
assert all(v.broadcastable)
bcast = []
for s in sh:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论