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

add num_groups to __props__

上级 7848292a
...@@ -502,7 +502,7 @@ class GpuDnnConv(DnnBase): ...@@ -502,7 +502,7 @@ class GpuDnnConv(DnnBase):
""" """
_f16_ok = True _f16_ok = True
__props__ = ('algo', 'inplace') __props__ = ('algo', 'inplace', 'num_groups')
check_input = False check_input = False
params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionFwdAlgo_t, params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionFwdAlgo_t,
...@@ -638,7 +638,7 @@ class GpuDnnConvGradW(DnnBase): ...@@ -638,7 +638,7 @@ class GpuDnnConvGradW(DnnBase):
""" """
_f16_ok = True _f16_ok = True
__props__ = ('algo', 'inplace') __props__ = ('algo', 'inplace', 'num_groups')
check_input = False check_input = False
params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionBwdFilterAlgo_t, params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionBwdFilterAlgo_t,
...@@ -769,7 +769,7 @@ class GpuDnnConvGradI(DnnBase): ...@@ -769,7 +769,7 @@ class GpuDnnConvGradI(DnnBase):
""" """
_f16_ok = True _f16_ok = True
__props__ = ('algo', 'inplace',) __props__ = ('algo', 'inplace', 'num_groups')
check_input = False check_input = False
params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionBwdDataAlgo_t, params_type = ParamsType(conv_algo=cudnn.cudnnConvolutionBwdDataAlgo_t,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论