提交 98cad04c authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Flake8

上级 9b4238a8
......@@ -3124,6 +3124,7 @@ def local_dnn_reduction(node):
node.op.dtype,
False)(node.inputs[0]),)
@register_opt('cudnn')
@local_optimizer([GpuMaxAndArgmax])
def local_cudnn_maxandargmax(node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论