提交 448f39fd authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix in gpuflatten optimization

上级 ace63e7a
...@@ -175,7 +175,7 @@ def local_gpu_flatten(node): ...@@ -175,7 +175,7 @@ def local_gpu_flatten(node):
host_input = node.inputs[0] host_input = node.inputs[0]
if host_input.owner and isinstance(host_input.owner.op, tensor.Flatten): if host_input.owner and isinstance(host_input.owner.op, tensor.Flatten):
outdim = host_input.owner.op.outdim outdim = host_input.owner.op.outdim
return [GpuFlatten(outdim)(gpu_from_host(host_input.inputs[0]))] return [GpuFlatten(outdim)(gpu_from_host(host_input.owner.inputs[0]))]
if isinstance(node.op, tensor.Flatten): if isinstance(node.op, tensor.Flatten):
x, = node.inputs x, = node.inputs
outdim = node.op.outdim outdim = node.op.outdim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论