diff --git a/xdl/xdl/core/backend/mxnet/convert_utils.cc b/xdl/xdl/core/backend/mxnet/convert_utils.cc index 8aabe708..ed079117 100644 --- a/xdl/xdl/core/backend/mxnet/convert_utils.cc +++ b/xdl/xdl/core/backend/mxnet/convert_utils.cc @@ -163,6 +163,7 @@ Status MX2XDL::ConvertType(const int s, xdl::DataType* d) { Status MX2XDL::ConvertShape(const std::vector& s, xdl::TensorShape* d) { std::vector dims; + dims.reserve(s.size()); for (auto dim: s) { dims.push_back(static_cast(dim)); }