ArmNN
 24.02
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 
22 /// Creates a tensor info after reducing the dimensions mentioned in axisData.
23 void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo,
24  const std::set<unsigned int>& axisSet,
25  bool keepDims,
26  armnn::TensorInfo& outputTensorInfo);
27 
28 /// Create output tensor info for a StridedSlice operator
31  armnn::TensorInfo& outputTensorInfo);
32 
33 } // namespace armnnUtils
armnn::TensorInfo
Definition: Tensor.hpp:152
armnnUtils::CalculateStridedSliceOutputTensorInfo
void CalculateStridedSliceOutputTensorInfo(const armnn::TensorInfo &inputTensorInfo, const armnn::StridedSliceDescriptor &desc, armnn::TensorInfo &outputTensorInfo)
Create output tensor info for a StridedSlice operator.
Definition: ParserHelper.cpp:103
armnnUtils::ProcessConcatInputTensorInfo
void ProcessConcatInputTensorInfo(armnn::TensorInfo &inputTensorInfo, armnn::OriginsDescriptor &concatDescriptor, const unsigned int &concatAxis, unsigned int inputIndex, unsigned int &mergeDimOrigin)
Definition: ParserHelper.cpp:19
TensorFwd.hpp
armnnUtils
Definition: CompatibleTypes.hpp:10
DescriptorsFwd.hpp
armnn::StridedSliceDescriptor
A StridedSliceDescriptor for the StridedSliceLayer.
Definition: Descriptors.hpp:1303
armnn::OriginsDescriptor
An OriginsDescriptor for the ConcatLayer.
Definition: Descriptors.hpp:201
armnnUtils::CalculateReducedOutputTensoInfo
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.
Definition: ParserHelper.cpp:52