提交 01f4490c authored 作者: sebastien-j's avatar sebastien-j

Fix flake8

上级 2aadc8ee
...@@ -1672,6 +1672,7 @@ def local_pool_dnn_grad_stride(node): ...@@ -1672,6 +1672,7 @@ def local_pool_dnn_grad_stride(node):
gpu_contiguous(inp_grad), gpu_contiguous(inp_grad),
desc) desc)
@register_opt('cudnn') @register_opt('cudnn')
@op_lifter([AveragePoolGrad]) @op_lifter([AveragePoolGrad])
def local_avg_pool_dnn_grad_stride(node): def local_avg_pool_dnn_grad_stride(node):
...@@ -1687,10 +1688,11 @@ def local_avg_pool_dnn_grad_stride(node): ...@@ -1687,10 +1688,11 @@ def local_avg_pool_dnn_grad_stride(node):
desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)() desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)()
return GpuDnnPoolGrad()(gpu_contiguous(inp), return GpuDnnPoolGrad()(gpu_contiguous(inp),
gpu_contiguous(numpy.empty((1,1,1,1), dtype=numpy.float32)), gpu_contiguous(numpy.empty((1, 1, 1, 1), dtype=numpy.float32)),
gpu_contiguous(inp_grad), gpu_contiguous(inp_grad),
desc) desc)
@register_opt('cudnn') @register_opt('cudnn')
@local_optimizer([GpuSoftmax]) @local_optimizer([GpuSoftmax])
def local_softmax_dnn(node): def local_softmax_dnn(node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论