提交 c729fbd6 编写于 作者: A Alexander Alekhin

Merge pull request #10925 from pengli:dnn

...@@ -1390,8 +1390,8 @@ public: ...@@ -1390,8 +1390,8 @@ public:
{ {
int total = outGroupCn * decnBlob.cols; int total = outGroupCn * decnBlob.cols;
int index = 0; int index = 0;
int height_col = (outH + 2 * pad.height - kernel.height) / stride.height + 1; int height_col = inpH;
int width_col = (outW + 2 * pad.width - kernel.width) / stride.width + 1; int width_col = inpW;
int coeff_h = (1 - stride.height * kernel.width * height_col) * width_col; int coeff_h = (1 - stride.height * kernel.width * height_col) * width_col;
int coeff_w = (1 - stride.width * height_col * width_col); int coeff_w = (1 - stride.width * height_col * width_col);
......
...@@ -181,6 +181,10 @@ TEST(Test_TensorFlow, deconvolution) ...@@ -181,6 +181,10 @@ TEST(Test_TensorFlow, deconvolution)
OCL_TEST(Test_TensorFlow, deconvolution) OCL_TEST(Test_TensorFlow, deconvolution)
{ {
runTensorFlowNet("deconvolution", DNN_TARGET_OPENCL); runTensorFlowNet("deconvolution", DNN_TARGET_OPENCL);
runTensorFlowNet("deconvolution_same", DNN_TARGET_OPENCL);
runTensorFlowNet("deconvolution_stride_2_same", DNN_TARGET_OPENCL);
runTensorFlowNet("deconvolution_adj_pad_valid", DNN_TARGET_OPENCL);
runTensorFlowNet("deconvolution_adj_pad_same", DNN_TARGET_OPENCL);
} }
TEST(Test_TensorFlow, matmul) TEST(Test_TensorFlow, matmul)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册