提交 4e67e4d4 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Ignore nodes that are already on the GPU.

上级 f279798b
...@@ -806,6 +806,9 @@ theano.tensor.nnet.conv2d() ...@@ -806,6 +806,9 @@ theano.tensor.nnet.conv2d()
AbstractConv2d_gradWeights, AbstractConv2d_gradWeights,
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):
# Don't handle this node here.
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],
context_name=context_name) context_name=context_name)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论