提交 e9796582 authored 作者: affanv14's avatar affanv14

add op attibutes to other tests

上级 d8a2e822
......@@ -2161,5 +2161,8 @@ class Cudnn_grouped_conv(Grouped_conv_noOptim):
conv2d = staticmethod(dconv2d)
conv2d_gradw = staticmethod(dconv2dw)
conv2d_gradi = staticmethod(dconv2di)
conv2d_op = staticmethod(dnn.GpuDnnConv)
conv2d_gradw_op = staticmethod(dnn.GpuDnnConvGradW)
conv2d_gradi_op = staticmethod(dnn.GpuDnnConvGradI)
flip_filter = False
is_dnn = True
......@@ -227,5 +227,8 @@ class TestGroupGpuCorr2d(Grouped_conv_noOptim):
conv2d = staticmethod(GpuCorrMM)
conv2d_gradw = staticmethod(GpuCorrMM_gradWeights)
conv2d_gradi = staticmethod(GpuCorrMM_gradInputs)
conv2d_op = staticmethod(GpuCorrMM)
conv2d_gradw_op = staticmethod(GpuCorrMM_gradWeights)
conv2d_gradi_op = staticmethod(GpuCorrMM_gradInputs)
flip_filter = True
is_dnn = False
......@@ -425,6 +425,9 @@ class TestGroupCorr2d(Grouped_conv_noOptim):
conv2d = staticmethod(corr.CorrMM)
conv2d_gradw = staticmethod(corr.CorrMM_gradWeights)
conv2d_gradi = staticmethod(corr.CorrMM_gradInputs)
conv2d_op = staticmethod(corr.CorrMM)
conv2d_gradw_op = staticmethod(corr.CorrMM_gradWeights)
conv2d_gradi_op = staticmethod(corr.CorrMM_gradInputs)
flip_filter = True
is_dnn = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论