diff --git a/paddle/phi/core/dense_tensor.h b/paddle/phi/core/dense_tensor.h index 125e930ee33824..aac012afdb018a 100644 --- a/paddle/phi/core/dense_tensor.h +++ b/paddle/phi/core/dense_tensor.h @@ -196,6 +196,8 @@ class PADDLE_API DenseTensor : public TensorBase, DenseTensor& Resize(const std::vector& dims); + DenseTensor& Resize(const std::vector& dims); + /// \brief Change the lod information in the metadata. /// \param legacy_lod The new lod of the dense tensor. void ResetLoD(const LegacyLoD& legacy_lod); diff --git a/paddle/phi/core/dense_tensor_impl.cc b/paddle/phi/core/dense_tensor_impl.cc index 72a202ff5ff163..97139c83f9fd26 100644 --- a/paddle/phi/core/dense_tensor_impl.cc +++ b/paddle/phi/core/dense_tensor_impl.cc @@ -307,6 +307,10 @@ DenseTensor& DenseTensor::Resize(const std::vector& dims) { return Resize(make_ddim(dims)); } +DenseTensor& DenseTensor::Resize(const std::vector& dims) { + return Resize(make_ddim(dims)); +} + DenseTensor DenseTensor::Slice(int64_t begin_idx, int64_t end_idx) const { check_memory_size(); PADDLE_ENFORCE_GE( diff --git a/paddle/phi/kernels/impl/lu_kernel_impl.h b/paddle/phi/kernels/impl/lu_kernel_impl.h index 59cb13caa48608..19d0f574837af1 100644 --- a/paddle/phi/kernels/impl/lu_kernel_impl.h +++ b/paddle/phi/kernels/impl/lu_kernel_impl.h @@ -523,7 +523,7 @@ DenseTensor Transpose2DTo6D(const Context& dev_ctx, const DenseTensor& x) { axis[i] = i; } std::swap(axis[rank - 1], axis[rank - 2]); - ret.Resize(make_ddim(x_vec)); + ret.Resize(x_vec); dev_ctx.template Alloc(&ret); switch (rank) { case 2: {