ArmNN  NotReleased
ParserHelper.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 #include <armnn/TensorFwd.hpp>
10 
11 #include <set>
12 
13 namespace armnnUtils
14 {
15 
17  armnn::OriginsDescriptor& concatDescriptor,
18  const unsigned int& concatAxis,
19  unsigned int inputIndex,
20  unsigned int& mergeDimOrigin);
21 
23 void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo,
24  const std::set<unsigned int>& axisSet,
25  bool keepDims,
26  armnn::TensorInfo& outputTensorInfo);
27 
31  armnn::TensorInfo& outputTensorInfo);
32 
33 } // namespace armnnUtils
void ProcessConcatInputTensorInfo(armnn::TensorInfo &inputTensorInfo, armnn::OriginsDescriptor &concatDescriptor, const unsigned int &concatAxis, unsigned int inputIndex, unsigned int &mergeDimOrigin)
void CalculateStridedSliceOutputTensorInfo(const armnn::TensorInfo &inputTensorInfo, const armnn::StridedSliceDescriptor &desc, armnn::TensorInfo &outputTensorInfo)
Create output tensor info for a StridedSlice operator.
void CalculateReducedOutputTensoInfo(const armnn::TensorInfo &inputTensorInfo, const std::set< unsigned int > &axisSet, bool keepDims, armnn::TensorInfo &outputTensorInfo)
Creates a tensor info after reducing the dimensions mentioned in axisData.
An OriginsDescriptor for the ConcatLayer. Descriptor to configure the concatenation process...
A StridedSliceDescriptor for the StridedSliceLayer.