提交 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): ...@@ -208,8 +208,8 @@ class TestConvOp(unittest.TestCase):
#test speed #test speed
bsize = 10 # batch size bsize = 10 # batch size
imshp_start = (1,50,50) imshp_start = (1,50,49)
kshps = ([12,12],[12,12]) kshps = ([11,12],[12,11])
nkerns = [20,20] # per output pixel nkerns = [20,20] # per output pixel
ssizes = [(1,1),]#(1,1)]#(2,2) bugged ssizes = [(1,1),]#(1,1)]#(2,2) bugged
convmodes = ['valid','full'] convmodes = ['valid','full']
...@@ -297,7 +297,7 @@ class TestConvOp(unittest.TestCase): ...@@ -297,7 +297,7 @@ class TestConvOp(unittest.TestCase):
hidval1=outval.copy() hidval1=outval.copy()
# ConvOp # 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, l1shp=N.hstack((nkern,
getFilterOutShp(imshp, kshp, ss, conv_mode))) getFilterOutShp(imshp, kshp, ss, conv_mode)))
propup2 = function([inputs4, kerns4], conv_op) propup2 = function([inputs4, kerns4], conv_op)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论