aboutsummaryrefslogtreecommitdiff
path: root/src/armnnUtils/ParserHelper.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnUtils/ParserHelper.hpp')
-rw-r--r--src/armnnUtils/ParserHelper.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnnUtils/ParserHelper.hpp b/src/armnnUtils/ParserHelper.hpp
index 93dfbf9360..24369dc521 100644
--- a/src/armnnUtils/ParserHelper.hpp
+++ b/src/armnnUtils/ParserHelper.hpp
@@ -14,4 +14,9 @@ void ProcessConcatInputTensorInfo(armnn::TensorInfo& inputTensorInfo, armnn::Ori
const unsigned int& concatAxis, unsigned int inputIndex,
std::vector<unsigned int>& mergeDimSizes, unsigned int& mergeDim);
+/// Creates a tensor info after reducing the dimensions mentioned in axisData.
+void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo, const armnn::TensorInfo& axisTensorInfo,
+ const std::set<unsigned int>& axisSet, bool keepDims,
+ armnn::TensorInfo& outputTensorInfo);
+
} // namespace armnnUtils