提交 89f488e8 authored 作者: Frederic's avatar Frederic

remove parameter not used

上级 992f9046
...@@ -304,8 +304,7 @@ class test_SoftMax(unittest.TestCase): ...@@ -304,8 +304,7 @@ class test_SoftMax(unittest.TestCase):
x_gpu, x_gpu,
f_z, f_z,
f_gpu_z, f_gpu_z,
cmp, cmp
gpu_mode
): ):
""" """
This is basic test for GpuSoftmax and GpuDnnSoftmax This is basic test for GpuSoftmax and GpuDnnSoftmax
...@@ -366,8 +365,7 @@ class test_SoftMax(unittest.TestCase): ...@@ -366,8 +365,7 @@ class test_SoftMax(unittest.TestCase):
x, x,
z, z,
z, z,
self._cmp, self._cmp
self.mode,
) )
# cuDNN R1 cannot handle these test cases but the Theano softmax can so # cuDNN R1 cannot handle these test cases but the Theano softmax can so
...@@ -383,8 +381,7 @@ class test_SoftMax(unittest.TestCase): ...@@ -383,8 +381,7 @@ class test_SoftMax(unittest.TestCase):
x, x,
z, z,
z, z,
self._cmp, self._cmp
self.mode,
) )
# Theano can handle that case, but cudnn can't # Theano can handle that case, but cudnn can't
self._cmp(0, 10, f, f_gpu) self._cmp(0, 10, f, f_gpu)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论