提交 2708e558 authored 作者: Frederic's avatar Frederic

Fix warning reported on the mailing list

上级 6f01cf86
......@@ -1718,11 +1718,11 @@ if True:
@register_opt('cudnn')
@local_optimizer([SoftmaxGrad])
def local_softmax_dnn_grad(node):
if (
isinstance(node.op, SoftmaxGrad)
and (isinstance(node.inputs[0].owner.op, HostFromGpu)
or isinstance(node.inputs[1].owner.op, HostFromGpu))
):
if (isinstance(node.op, SoftmaxGrad) and
((node.inputs[0].owner and
isinstance(node.inputs[0].owner.op, HostFromGpu))
or (node.inputs[1].owner and
isinstance(node.inputs[1].owner.op, HostFromGpu)))):
if not dnn_available():
return
ins = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论