aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/DetectionPostProcessLayer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/DetectionPostProcessLayer.hpp')
-rw-r--r--src/armnn/layers/DetectionPostProcessLayer.hpp8
1 files changed, 7 insertions, 1 deletions
diff --git a/src/armnn/layers/DetectionPostProcessLayer.hpp b/src/armnn/layers/DetectionPostProcessLayer.hpp
index 07eb270f1f..e203032db0 100644
--- a/src/armnn/layers/DetectionPostProcessLayer.hpp
+++ b/src/armnn/layers/DetectionPostProcessLayer.hpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2017-2023 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
@@ -34,6 +34,12 @@ public:
/// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated.
void ValidateTensorShapesFromInputs() override;
+ /// The model does not specify the output shapes. The output shapes are calculated from the max_detection and
+ /// max_classes_per_detection parameters in the DetectionPostProcessDescriptor.
+ /// @param [in] inputShapes The input shapes layer has. These are ignored for DetectionPostProcessLayer.
+ /// @return A vector to the inferred output shape.
+ std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
+
void ExecuteStrategy(IStrategy& strategy) const override;
protected: