提交 809a1ab5 authored 作者: Dustin Webb's avatar Dustin Webb

Fixed bugs due to an interface change and that only appeared after performing a rebase.

上级 7801d91e
...@@ -291,7 +291,13 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -291,7 +291,13 @@ class TestDnnInferShapes(utt.InferShapeTester):
subsample=params[1], subsample=params[1],
conv_mode=params[2] conv_mode=params[2]
)(temp_img.shape, shape) )(temp_img.shape, shape)
conv_grad_w = dnn.GpuDnnConvGradW()(temp_img, temp_kerns, desc) conv_grad_w = dnn.GpuDnnConvGradW()(
temp_img,
temp_kerns,
desc,
shape[2],
shape[3]
)
self._compile_and_check( self._compile_and_check(
[temp_img, temp_kerns], [temp_img, temp_kerns],
[conv_grad_w], [conv_grad_w],
...@@ -328,7 +334,13 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -328,7 +334,13 @@ class TestDnnInferShapes(utt.InferShapeTester):
subsample=params[1], subsample=params[1],
conv_mode=params[2] conv_mode=params[2]
)(shape, temp_kerns.shape) )(shape, temp_kerns.shape)
conv_grad_i = dnn.GpuDnnConvGradI()(temp_kerns, img, desc) conv_grad_i = dnn.GpuDnnConvGradI()(
temp_kerns,
img,
desc,
shape[2],
shape[3]
)
self._compile_and_check( self._compile_and_check(
[temp_kerns, img], [temp_kerns, img],
[conv_grad_i], [conv_grad_i],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论