提交 b8fe4eb6 authored 作者: Frederic's avatar Frederic

Enable local_pool_dnn opt. It was disabled by mistake in a rebase.

上级 5c9e0b31
...@@ -1274,8 +1274,7 @@ if True: ...@@ -1274,8 +1274,7 @@ if True:
border_mode=border_mode, subsample=subsample, border_mode=border_mode, subsample=subsample,
direction_hint=direction_hint)] direction_hint=direction_hint)]
# DISABLED as there is problems in the handling of borders @register_opt('cudnn')
# @register_opt('cudnn')
@local_optimizer([GpuDownsampleFactorMax]) @local_optimizer([GpuDownsampleFactorMax])
def local_pool_dnn(node): def local_pool_dnn(node):
if not dnn_available(): if not dnn_available():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论