提交 4ec07673 authored 作者: affanv14's avatar affanv14

small fix for tests

上级 65ea94c5
......@@ -1859,8 +1859,12 @@ class Grouped_conv_noOptim(unittest.TestCase):
filter_dilation=self.filter_dilation,
num_groups=groups)
if self.flip_filter:
grouped_conv_output = grouped_convgrad_op(kern_sym[:, :, ::-1, ::-1], top_sym,
tensor.as_tensor_variable(imshp[-self.convdim:]))
if self.convdim == 2:
grouped_conv_output = grouped_convgrad_op(kern_sym[:, :, ::-1, ::-1], top_sym,
tensor.as_tensor_variable(imshp[-self.convdim:]))
else:
grouped_conv_output = grouped_convgrad_op(kern_sym[:, :, ::-1, ::-1, ::-1], top_sym,
tensor.as_tensor_variable(imshp[-self.convdim:]))
else:
grouped_conv_output = grouped_convgrad_op(kern_sym,
top_sym,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论