提交 b4f64b29 authored 作者: Ziye Fan's avatar Ziye Fan

condition checking merged

上级 b7411329
......@@ -34,14 +34,14 @@ _logger = logging.getLogger(_logger_name)
def as_cuda_ndarray_variable(x):
if x.owner and isinstance(x.owner.op, HostFromGpu):
return x.owner.inputs[0]
if \
x.owner and \
isinstance(x.owner.op, GpuFromHost) and \
x.owner.inputs[0].owner and \
isinstance(x.owner.inputs[0].owner.op, HostFromGpu):
return x.owner.inputs[0].owner.inputs[0]
if x.owner:
if isinstance(x.owner.op, HostFromGpu):
return x.owner.inputs[0]
elif \
isinstance(x.owner.op, GpuFromHost) and \
x.owner.inputs[0].owner and \
isinstance(x.owner.inputs[0].owner.op, HostFromGpu):
return x.owner.inputs[0].owner.inputs[0]
if hasattr(x, '_as_CudaNdarrayVariable'):
return x._as_CudaNdarrayVariable()
tensor_x = tensor.as_tensor_variable(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论