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

make the optimizer local_gpu_reshape work in more case.

上级 f0312080
......@@ -1211,7 +1211,7 @@ class GpuSum(Op):
class GpuReshape(tensor.Reshape):
# __hash__, __eq__, __str__ come from tensor.Subtensor
def make_node(self, x, shp):
host_reshaped = host_from_gpu(x).reshape(shp)
host_reshaped = host_from_gpu(x).reshape(shp,ndim=self.ndim)
return Apply(self, [x, shp], [CudaNdarrayType(host_reshaped.broadcastable)()])
def perform(self, node, (x, shp), (out,)):
if (len(shp) != self.ndim):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论