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

add num_groups to __props__ in gpucorr3dmm

上级 8411f061
...@@ -1070,7 +1070,7 @@ class BaseGpuCorr3dMM(CGpuKernelBase): ...@@ -1070,7 +1070,7 @@ class BaseGpuCorr3dMM(CGpuKernelBase):
""" """
check_broadcast = False check_broadcast = False
__props__ = ('border_mode', 'subsample', 'filter_dilation') __props__ = ('border_mode', 'subsample', 'filter_dilation', 'num_groups')
_f16_ok = True _f16_ok = True
def __init__(self, border_mode="valid", subsample=(1, 1, 1), def __init__(self, border_mode="valid", subsample=(1, 1, 1),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论