Deconvolution ocl fix

Signed-off-by: Li Peng <peng.li@intel.com>
pull/10925/head
Li Peng 7 years ago
parent 5caf6244a3
commit 608968aa83
  1. 4
      modules/dnn/src/layers/convolution_layer.cpp
  2. 4
      modules/dnn/test/test_tf_importer.cpp

@ -1390,8 +1390,8 @@ public:
{
int total = outGroupCn * decnBlob.cols;
int index = 0;
int height_col = (outH + 2 * pad.height - kernel.height) / stride.height + 1;
int width_col = (outW + 2 * pad.width - kernel.width) / stride.width + 1;
int height_col = inpH;
int width_col = inpW;
int coeff_h = (1 - stride.height * kernel.width * height_col) * width_col;
int coeff_w = (1 - stride.width * height_col * width_col);

@ -181,6 +181,10 @@ TEST(Test_TensorFlow, deconvolution)
OCL_TEST(Test_TensorFlow, deconvolution)
{
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)

Loading…
Cancel
Save