提交 f4588d23 authored 作者: Jesse Livezey's avatar Jesse Livezey

added to test from review

上级 a1301828
...@@ -591,7 +591,9 @@ class test_DnnSoftMax(test_nnet.test_SoftMax): ...@@ -591,7 +591,9 @@ class test_DnnSoftMax(test_nnet.test_SoftMax):
xp = x.reshape((5, 5)) xp = x.reshape((5, 5))
y = T.nnet.softmax(xp.flatten()).sum() y = T.nnet.softmax(xp.flatten()).sum()
g = T.grad(y, x) g = T.grad(y, x)
theano.function(inputs=[x], outputs=g, mode=self.mode) f = theano.function(inputs=[x], outputs=g, mode=self.mode)
assert(any(n for n in f.maker.fgraph.toposort() if
isinstance(n.op, dnn.GpuDnnSoftmaxGrad)))
def test_cudnn_softmax_grad_opt(self): def test_cudnn_softmax_grad_opt(self):
# Verify that the SoftmaxGrad -> GpuDnnSoftmaxGrad optimization is # Verify that the SoftmaxGrad -> GpuDnnSoftmaxGrad optimization is
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论