aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ReduceLayer.hpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2023-01-10 15:55:28 +0000
committermike.kelly <mike.kelly@arm.com>2023-01-12 14:46:46 +0000
commit377fb21e956ea68ffd234be47481002a0e46ee46 (patch)
tree3d82f8d43c36cd4054fd6e29d082599e6ef0be86 /src/armnn/layers/ReduceLayer.hpp
parent01f72693d39ed966ad06adadc8aac141bc395659 (diff)
downloadarmnn-377fb21e956ea68ffd234be47481002a0e46ee46.tar.gz
IVGCVSW-7244 ConcatLayer overlapping views on TFLiteParser
* Added ability to calculate dynamic tensors and propagate them through the model so that when those tensors are later used as inputs they have the right shapes. * Added InferOutputShapes to DetectionPostProcessLayer. * Added InferOutputShapes to MeanLayer. * Added InferOutputShapes to RankLayer. * Added InferOutputShapes to ReduceLayer. * Fixed typos in TfLiteParser. Signed-off-by: Mike Kelly <mike.kelly@arm.com> Change-Id: I880c0716938ef278f5dbf01a8a73a5cc99ce5ded
Diffstat (limited to 'src/armnn/layers/ReduceLayer.hpp')
-rw-r--r--src/armnn/layers/ReduceLayer.hpp9
1 files changed, 7 insertions, 2 deletions
diff --git a/src/armnn/layers/ReduceLayer.hpp b/src/armnn/layers/ReduceLayer.hpp
index e9ea5d8e3f..24f19f812f 100644
--- a/src/armnn/layers/ReduceLayer.hpp
+++ b/src/armnn/layers/ReduceLayer.hpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2020 Samsung Electronics Co Ltd and Contributors. All rights reserved.
+// Copyright © 2020-2023 Samsung Electronics Co Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
@@ -23,13 +23,18 @@ public:
/// @param [in] graph The graph into which this layer is being cloned.
ReduceLayer* Clone(Graph& graph) const override;
+ /// By default returns inputShapes if the number of inputs are equal to number of outputs,
+ /// otherwise infers the output shapes from given input shapes and layer properties.
+ /// @param [in] inputShapes The input shapes layer has.
+ /// @return A vector to the inferred output shape.
+ std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
+
/// Check if the input tensor shape(s)
/// will lead to a valid configuration of @ref ReduceLayer.
void ValidateTensorShapesFromInputs() override;
void ExecuteStrategy(IStrategy& strategy) const override;
-
protected:
/// Constructor to create a ReduceLayer.
/// @param [in] param ReduceDescriptor to configure the reduction operation.