提交 a87abbc9 authored 作者: orhanf's avatar orhanf

pep8 fixes

上级 8ce2395b
...@@ -61,6 +61,7 @@ def register_opt(*tags, **kwargs): ...@@ -61,6 +61,7 @@ def register_opt(*tags, **kwargs):
register_opt('fast_compile')(theano.tensor.opt.local_track_shape_i) register_opt('fast_compile')(theano.tensor.opt.local_track_shape_i)
register_opt('unsafe')(theano.tensor.opt.local_remove_all_assert) register_opt('unsafe')(theano.tensor.opt.local_remove_all_assert)
def safe_to_gpu(x): def safe_to_gpu(x):
if isinstance(x.type, tensor.TensorType): if isinstance(x.type, tensor.TensorType):
return gpu_from_host(x) return gpu_from_host(x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论