Tensor method rename dims()->sizes() (#15246)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15246

Codemod generated with clangr shard mode, 25 files per diff,

Reviewed By: igorsugak

Differential Revision: D13470369

fbshipit-source-id: ce995beab7c64bebe8b234fb5e6d015940ec2952

diff --git a/caffe2/operators/elementwise_ops.h b/caffe2/operators/elementwise_ops.h
index 2cc32df..5659c64 100644 (file)
--- a/caffe2/operators/elementwise_ops.h
+++ b/caffe2/operators/elementwise_ops.h
@@ -160,7 +160,7 @@ class BinaryElementwiseWithArgsOp final : public Operator<Context> {
           !IsInputOutputAlias(1, 0),
           "In-place is allowed only with the first tensor when "
           "legacy-broadcasting");
-      C_dims = A.dims().vec();
+      C_dims = A.sizes().vec();
       if (B.numel() == 1) {
         A_dims = {static_cast<int>(A.numel())};
         B_dims = {1};

diff --git a/caffe2/operators/gather_op.h b/caffe2/operators/gather_op.h
index 4afc5c5..d97a5df 100644 (file)
--- a/caffe2/operators/gather_op.h
+++ b/caffe2/operators/gather_op.h
@@ -80,7 +80,7 @@ static bool gather_impl(
   // New shape:
   //  [data dims before axis] + [indices dims] + [data dims after axis]
   vector<int64_t> shape =
-      calc_output_shape_vector<int64_t>(data.dims(), indices.dims(), axis);
+      calc_output_shape_vector<int64_t>(data.sizes(), indices.sizes(), axis);
   Tensor* output = op->Output(outputIdx, shape, at::dtype(dataType));
   auto out = static_cast<char*>(output->raw_mutable_data(dataType));
 

diff --git a/caffe2/operators/layer_norm_op.cc b/caffe2/operators/layer_norm_op.cc
index f39dde1..9be131d 100644 (file)
--- a/caffe2/operators/layer_norm_op.cc
+++ b/caffe2/operators/layer_norm_op.cc
@@ -197,7 +197,7 @@ void layer_norm_c10(
     caffe2::BaseContext* context) {
   const int canonical_axis = X.canonical_axis_index(axis);
   std::vector<int64_t> moments_dims(
-      X.dims().cbegin(), X.dims().cbegin() + canonical_axis);
+      X.sizes().cbegin(), X.sizes().cbegin() + canonical_axis);
   moments_dims.push_back(1);
   mean->Resize(moments_dims);
   sig->Resize(moments_dims);

diff --git a/caffe2/operators/layer_norm_op.h b/caffe2/operators/layer_norm_op.h
index d47064e..49f2307 100644 (file)
--- a/caffe2/operators/layer_norm_op.h
+++ b/caffe2/operators/layer_norm_op.h
@@ -33,7 +33,7 @@ class LayerNormOp final : public Operator<Context> {
     auto* Y = Output(0);
     const int canonical_axis = X.canonical_axis_index(axis_);
     std::vector<int64_t> moments_dims(
-        X.dims().cbegin(), X.dims().cbegin() + canonical_axis);
+        X.sizes().cbegin(), X.sizes().cbegin() + canonical_axis);
     moments_dims.push_back(1);
     auto* mean = Output(1, moments_dims, at::dtype<T>());
     auto* sig = Output(2, moments_dims, at::dtype<T>());