[Numpy] fix test_operator_gpu.test_upsampling_bilinear_with_type (#14557

) * Fix test_operator_gpu.test_upsampling_bilinear_with_type * Address comments
apache · Apr 4, 2019 · 29281cd · 29281cd
1 parent cc078c2
commit 29281cd
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;
       }
     }
   }