From 46439c78d05db91aacd9e7d511154bf0246903af Mon Sep 17 00:00:00 2001 From: Junjie Bai Date: Thu, 21 Mar 2019 16:24:45 -0700 Subject: [PATCH] Replace the remaining usages of IntList in caffe2 to IntArrayRef Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18282 Differential Revision: D14569269 Pulled By: bddppq fbshipit-source-id: 5fc33701b83f9efdec4b456d2691764831d10e7f --- caffe2/operators/conv_pool_op_base.h | 2 +- caffe2/quantization/server/conv_pool_dnnlowp_op_base.h | 2 +- caffe2/quantization/server/dnnlowp_op.h | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/caffe2/operators/conv_pool_op_base.h b/caffe2/operators/conv_pool_op_base.h index 455afe1..5fb07d2 100644 --- a/caffe2/operators/conv_pool_op_base.h +++ b/caffe2/operators/conv_pool_op_base.h @@ -295,7 +295,7 @@ class ConvPoolOpBase : public Operator { } static void InferOutputSize64( - const at::IntList& input_dims, + const at::IntArrayRef& input_dims, const int output_channel, const StorageOrder order, const bool global_pooling, diff --git a/caffe2/quantization/server/conv_pool_dnnlowp_op_base.h b/caffe2/quantization/server/conv_pool_dnnlowp_op_base.h index 4a2c72b..8de6e18 100644 --- a/caffe2/quantization/server/conv_pool_dnnlowp_op_base.h +++ b/caffe2/quantization/server/conv_pool_dnnlowp_op_base.h @@ -69,7 +69,7 @@ class ConvPoolDNNLowPOpBase : public ConvPoolOpBase { return &Outputs()[idx]->template GetMutable()->t; } - Tensor* OutputTensorCPU_(int idx, at::IntList dims, at::TensorOptions options) { + Tensor* OutputTensorCPU_(int idx, at::IntArrayRef dims, at::TensorOptions options) { auto* t = &Outputs()[idx]->template GetMutable()->t; ReinitializeTensor(t, dims, options.device(CPU)); return t; diff --git a/caffe2/quantization/server/dnnlowp_op.h b/caffe2/quantization/server/dnnlowp_op.h index 88a5a1d..9db43c3 100644 --- a/caffe2/quantization/server/dnnlowp_op.h +++ b/caffe2/quantization/server/dnnlowp_op.h @@ -122,7 +122,7 @@ class DNNLowPOp : public Operator { } } - Tensor* OutputTensorCPU_(int idx, at::IntList dims, at::TensorOptions options) { + Tensor* OutputTensorCPU_(int idx, at::IntArrayRef dims, at::TensorOptions options) { if (dequantize_output_) { return Output(idx, dims, options.device(CPU)); } else { -- 2.7.4