提交 808296a1 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Expand on the comment.

上级 4e67e4d4
...@@ -807,7 +807,7 @@ theano.tensor.nnet.conv2d() ...@@ -807,7 +807,7 @@ theano.tensor.nnet.conv2d()
AbstractConv2d_gradInputs]) AbstractConv2d_gradInputs])
def local_lift_abstractconv2d(node, context_name): def local_lift_abstractconv2d(node, context_name):
if isinstance(node.outputs[0].type, GpuArrayType): if isinstance(node.outputs[0].type, GpuArrayType):
# Don't handle this node here. # Don't handle this node here, it's already on the GPU.
return return
inps = list(node.inputs) inps = list(node.inputs)
inps[0] = as_gpuarray_variable(node.inputs[0], inps[0] = as_gpuarray_variable(node.inputs[0],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论