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

add num_groups to __props__

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