提交 191de4eb authored 作者: --global's avatar --global

Change default implementation in Conv3d and Conv3dGrads to one supported in both V2 and V3

上级 557c3fd3
...@@ -605,7 +605,7 @@ class GpuDnnConv3d(GpuDnnConv): ...@@ -605,7 +605,7 @@ class GpuDnnConv3d(GpuDnnConv):
assert algo == None assert algo == None
algo = workmem algo = workmem
super(GpuDnnConv3d, self).__init__(inplace=inplace, algo='guess_once') super(GpuDnnConv3d, self).__init__(inplace=inplace, algo='none')
assert self.algo in ['none', 'guess_once', 'guess_on_shape_change', assert self.algo in ['none', 'guess_once', 'guess_on_shape_change',
'time_once', 'time_on_shape_change'] 'time_once', 'time_on_shape_change']
...@@ -827,7 +827,7 @@ class GpuDnnConv3dGradW(GpuDnnConvGradW): ...@@ -827,7 +827,7 @@ class GpuDnnConv3dGradW(GpuDnnConvGradW):
algo = workmem algo = workmem
super(GpuDnnConv3dGradW, self).__init__(inplace=inplace, super(GpuDnnConv3dGradW, self).__init__(inplace=inplace,
algo='guess_once') algo='none')
assert self.algo in ['none', 'guess_once', 'guess_on_shape_change'] assert self.algo in ['none', 'guess_once', 'guess_on_shape_change']
def grad(self, inp, grads): def grad(self, inp, grads):
...@@ -1015,7 +1015,7 @@ class GpuDnnConv3dGradI(GpuDnnConvGradI): ...@@ -1015,7 +1015,7 @@ class GpuDnnConv3dGradI(GpuDnnConvGradI):
algo = workmem algo = workmem
super(GpuDnnConv3dGradI, self).__init__(inplace=inplace, super(GpuDnnConv3dGradI, self).__init__(inplace=inplace,
algo="guess_once") algo="none")
assert self.algo in ['none', 'guess_once', 'guess_on_shape_change'] assert self.algo in ['none', 'guess_once', 'guess_on_shape_change']
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论