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

condition checking merged

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