提交 cdc564fd authored 作者: f0k's avatar f0k

Make test_conv_cuda_ndarray:test_conv_grads compare all ops against legacy implementation

上级 04cf96ba
......@@ -1005,7 +1005,7 @@ def conv_grad(mode, bs, ch, nf, rImg1, rImg2, rFlt1, rFlt2, subsample, op):
# skip if the reference implementation can't do it
pass
f = theano.function([i, k], outputs, mode=theano_mode)
f = theano.function([i, k], outputs, mode=theano_mode.excluding('conv_dnn', 'conv_gemm'))
allvals = f(npy_img, npy_kern)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论