11 #include <arm_compute/core/Types.h> 16 inline arm_compute::NormalizationLayerInfo
21 const unsigned int depth = tensorInfo.
GetShape()[depthDimension];
34 const uint32_t normSize = depth * 2u + 1u;
38 const float alpha = 1.0f;
41 const float kappa = 0.0f;
44 const float beta = 0.5f;
46 return arm_compute::NormalizationLayerInfo(arm_compute::NormType::CROSS_MAP, normSize, alpha, beta, kappa,
false);
54 switch (armnnFunction)
73 inline arm_compute::ActivationLayerInfo
96 using arm_compute::PoolingType;
98 switch (poolingAlgorithm)
110 using arm_compute::DimensionRoundingType;
120 inline arm_compute::NormType
123 using arm_compute::NormType;
132 inline arm_compute::FullyConnectedLayerInfo
135 arm_compute::FullyConnectedLayerInfo fc_info;
142 switch (resizeMethod)
145 return arm_compute::InterpolationPolicy::BILINEAR;
147 return arm_compute::InterpolationPolicy::NEAREST_NEIGHBOR;
157 if (softmaxDesc.
m_Axis == -1)
167 return static_cast<T
>(dim) - 1;
174 std::set<unsigned int> splitAxis;
176 for (
unsigned int i = 0; i < numSplit; ++i)
178 for (
unsigned int dimIdx = 0; dimIdx < numDimensions; ++dimIdx)
182 splitAxis.insert(dimIdx);
A ViewsDescriptor for the SplitterLayer.
arm_compute::InterpolationPolicy ConvertResizeMethodToAclInterpolationPolicy(ResizeMethod resizeMethod)
int m_Axis
Scalar, defaulted to the last index (-1), specifying the dimension the activation will be performed o...
const TensorShape & GetShape() const
arm_compute::PoolingType ConvertPoolingAlgorithmToAclPoolingType(PoolingAlgorithm poolingAlgorithm)
uint32_t GetNumDimensions() const
Get the number of dimensions.
A ComparisonDescriptor for the ComparisonLayer.
arm_compute::NormalizationLayerInfo CreateAclNormalizationLayerInfoForL2Normalization(const armnn::TensorInfo &tensorInfo, armnn::DataLayout dataLayout)
bool m_TransposeWeightMatrix
Enable/disable transpose weight matrix.
NormalizationAlgorithmChannel
arm_compute::NormType ConvertNormalizationAlgorithmChannelToAclNormType(NormalizationAlgorithmChannel channelType)
uint32_t GetNumViews() const
Get the number of views.
Copyright (c) 2020 ARM Limited.
std::set< unsigned int > ComputeSplitAxis(const armnn::SplitterDescriptor &desc, const TensorShape &input)
arm_compute::ActivationLayerInfo::ActivationFunction ConvertActivationFunctionToAclActivationFunction(ActivationFunction armnnFunction)
arm_compute::FullyConnectedLayerInfo ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo(const FullyConnectedDescriptor &fullyConnectedDesc)
arm_compute::DimensionRoundingType ConvertOutputShapeRoundingToAclDimensionRoundingType(OutputShapeRounding rounding)
A FullyConnectedDescriptor for the FullyConnectedLayer.
#define ARMNN_ASSERT(COND)
An ActivationDescriptor for the ActivationLayer.
min(a, max(b, input)) ReLu1 & ReLu6.
ComparisonOperation m_Operation
Specifies the comparison operation to execute.
const uint32_t * GetViewSizes(uint32_t idx) const
Get the view sizes at the int value idx.
float m_A
Alpha upper bound value used by the activation functions. (BoundedReLu, Linear, TanH, Elu).
arm_compute::ComparisonOperation ConvertComparisonOperationToAcl(const ComparisonDescriptor &descriptor)
unsigned int GetNumDimensions() const
T ComputeSoftmaxAclAxis(const SoftmaxDescriptor &softmaxDesc, const armnn::TensorInfo &tensor)
float m_B
Beta lower bound value used by the activation functions. (BoundedReLu, Linear, TanH).
A SoftmaxDescriptor for the SoftmaxLayer.
ActivationFunction m_Function
The activation function to use (Sigmoid, TanH, Linear, ReLu, BoundedReLu, SoftReLu, LeakyReLu, Abs, Sqrt, Square, Elu).
arm_compute::ActivationLayerInfo ConvertActivationDescriptorToAclActivationLayerInfo(const ActivationDescriptor &actDesc)