diff options
| -rw-r--r-- | src/armnn/layers/PadLayer.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/armnn/layers/PadLayer.cpp b/src/armnn/layers/PadLayer.cpp index 324dd4a2e7..f620a6e9a4 100644 --- a/src/armnn/layers/PadLayer.cpp +++ b/src/armnn/layers/PadLayer.cpp @@ -46,8 +46,7 @@ std::vector<TensorShape> PadLayer::InferOutputShapes(const std::vector<TensorSha ARMNN_ASSERT(m_Param.m_PadList.size() == rank); ARMNN_ASSERT(rank != 0); - std::vector<unsigned int> outputDimensionSizes; - outputDimensionSizes.reserve(rank); + std::vector<unsigned int> outputDimensionSizes(rank); for (unsigned int i = 0; i < rank; ++i) { outputDimensionSizes[i] = inputShape[i] + m_Param.m_PadList[i].first + m_Param.m_PadList[i].second; @@ -77,4 +76,4 @@ void PadLayer::Accept(ILayerVisitor& visitor) const visitor.VisitPadLayer(this, GetParameters(), GetName()); } -} // namespace armnn
\ No newline at end of file +} // namespace armnn |