提交 96095dee authored 作者: Frederic Bastien's avatar Frederic Bastien

added the unroll version by kern. If you put the ConvOp option unroll_batch or…

added the unroll version by kern. If you put the ConvOp option unroll_batch or unroll_kern>0, we will those version with the provided unrool size.
上级 57489fbe
差异被折叠。
......@@ -208,8 +208,8 @@ class TestConvOp(unittest.TestCase):
#test speed
bsize = 10 # batch size
imshp_start = (1,50,50)
kshps = ([12,12],[12,12])
imshp_start = (1,50,49)
kshps = ([11,12],[12,11])
nkerns = [20,20] # per output pixel
ssizes = [(1,1),]#(1,1)]#(2,2) bugged
convmodes = ['valid','full']
......@@ -297,7 +297,7 @@ class TestConvOp(unittest.TestCase):
hidval1=outval.copy()
# ConvOp
conv_op = ConvOp(imshp, kshp, nkern, bsize, 1,1, conv_mode, unroll_batch=10)(inputs4, kerns4)
conv_op = ConvOp(imshp, kshp, nkern, bsize, 1,1, conv_mode, unroll_kern=10)(inputs4, kerns4)
l1shp=N.hstack((nkern,
getFilterOutShp(imshp, kshp, ss, conv_mode)))
propup2 = function([inputs4, kerns4], conv_op)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论