Tensor construction codemod - 3/3 (#14835)

Summary: Pull Request resolved: pytorch/pytorch#14835 Codemod generated with clangr shard mode, 25 files per diff, motivation: pytorch/pytorch#12407 Reviewed By: bddppq Differential Revision: D13335184 fbshipit-source-id: 26d8247e16b30bdff045530034af9b72c76d066f
TonyTangYu · Dec 6, 2018 · b5db6ac · b5db6ac
1 parent 20d1bff
commit b5db6ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/detectron/upsample_nearest_op.h b/modules/detectron/upsample_nearest_op.h
@@ -36,11 +36,11 @@ class UpsampleNearestOp final : public Operator<Context> {
 
   bool RunOnDevice() override {
     auto& X = Input(0);
-    auto* Y = Output(0);
+
     auto out_shape = X.sizes().vec();
     out_shape[X.dim() - 1] *= scale_;
     out_shape[X.dim() - 2] *= scale_;
-    Y->Resize(out_shape);
+    auto* Y = Output(0, out_shape, at::dtype<T>());
 
     int d1;
     int d2;