From 5b542a755f103c337c8d12a63c49329a5e895a53 Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Fri, 14 Dec 2018 21:08:20 -0800 Subject: [PATCH] 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 --- caffe2/operators/elementwise_ops.h | 2 +- caffe2/operators/gather_op.h | 2 +- caffe2/operators/layer_norm_op.cc | 2 +- caffe2/operators/layer_norm_op.h | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/caffe2/operators/elementwise_ops.h b/caffe2/operators/elementwise_ops.h index 2cc32df..5659c64 100644 --- a/caffe2/operators/elementwise_ops.h +++ b/caffe2/operators/elementwise_ops.h @@ -160,7 +160,7 @@ class BinaryElementwiseWithArgsOp final : public Operator { !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(A.numel())}; B_dims = {1}; diff --git a/caffe2/operators/gather_op.h b/caffe2/operators/gather_op.h index 4afc5c5..d97a5df 100644 --- 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 shape = - calc_output_shape_vector(data.dims(), indices.dims(), axis); + calc_output_shape_vector(data.sizes(), indices.sizes(), axis); Tensor* output = op->Output(outputIdx, shape, at::dtype(dataType)); auto out = static_cast(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 --- 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 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 --- a/caffe2/operators/layer_norm_op.h +++ b/caffe2/operators/layer_norm_op.h @@ -33,7 +33,7 @@ class LayerNormOp final : public Operator { auto* Y = Output(0); const int canonical_axis = X.canonical_axis_index(axis_); std::vector 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()); auto* sig = Output(2, moments_dims, at::dtype()); -- 2.7.4