提交 e63cea9b authored 作者: notoraptor's avatar notoraptor 提交者: notoraptor

Fix runtime tests for 3D convs

上级 87641e6f
......@@ -2421,6 +2421,7 @@ class TestDnnConv2DRuntimeAlgorithms(object):
class TestDnnConv3DRuntimeAlgorithms(TestDnnConv2DRuntimeAlgorithms):
ndim = 3
cpu_conv_class = theano.tensor.nnet.corr3d.Corr3dMM
runtime_shapes = [
(3, [(2, 3, 5, 10, 9), (5, 3, 4, 7, 7)]),
(1, [(1, 1, 5, 100, 200), (1, 1, 4, 50, 200)]),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论