提交 ad25509e authored 作者: Frederic Bastien's avatar Frederic Bastien

flake8

上级 aefd29db
......@@ -1472,7 +1472,7 @@ def local_abstractconv_cudnn_graph(op, context_name, inputs, outputs):
return [rval]
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@local_optimizer([AbstractConv2d])
def local_abstractconv_cudnn(node):
ctx = infer_context_name(*node.inputs)
......@@ -1481,7 +1481,7 @@ def local_abstractconv_cudnn(node):
return local_abstractconv_cudnn_graph(node.op, ctx, node.inputs, node.outputs)
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@local_optimizer([AbstractConv2d_gradWeights])
def local_abstractconv_gw_cudnn(node):
ctx = infer_context_name(*node.inputs)
......@@ -1490,7 +1490,7 @@ def local_abstractconv_gw_cudnn(node):
return local_abstractconv_cudnn_graph(node.op, ctx, node.inputs, node.outputs)
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@register_opt('fast_compile', 'conv_dnn', 'cudnn')
@local_optimizer([AbstractConv2d_gradInputs])
def local_abstractconv_gi_cudnn(node):
ctx = infer_context_name(*node.inputs)
......
......@@ -14,7 +14,6 @@ from theano.compile.ops import shape_i
from theano.gof import (local_optimizer, EquilibriumDB, TopoOptimizer,
SequenceDB, Optimizer, DB, toolbox, graph)
from theano.gof.opt import NavigatorOptimizer
from theano.gof.optdb import LocalGroupDB
from theano.ifelse import IfElse
from theano.misc.ordered_set import OrderedSet
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论