[Numpy] fix test_operator_gpu.test_upsampling_bilinear_with_type (apa…

…che#14557) * Fix test_operator_gpu.test_upsampling_bilinear_with_type * Address comments
reminisce · Apr 12, 2019 · 5cb093b · 5cb093b
1 parent f6877df
commit 5cb093b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/operator/nn/deconvolution-inl.h b/src/operator/nn/deconvolution-inl.h
@@ -143,9 +143,9 @@ struct DeconvolutionParam : public dmlc::Parameter<DeconvolutionParam> {
         }
       }
     } else {
-      for (size_t i = 0; i < ndim; i++) {
-        o_pad[i] = pad[i];
-        o_adj[i] = adj[i];
+      for (int i = 0; i < (int) ndim; i++) {
+        o_pad[i] = i < pad.ndim() ? pad[i] : 0;
+        o_adj[i] = i < adj.ndim() ? adj[i] : 0;
       }
     }
   }