ArmNN
 21.02
Public Member Functions | List of all members
StaticRangeStrategy Class Reference

#include <StaticRangeStrategy.hpp>

Inheritance diagram for StaticRangeStrategy:
IStrategy

Public Member Functions

 StaticRangeStrategy (RangeTracker &rangeTracker)
 
 ~StaticRangeStrategy ()=default
 
void ExecuteStrategy (const armnn::IConnectableLayer *layer, const BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id) override
 
- Public Member Functions inherited from IStrategy
virtual void FinishStrategy ()
 

Additional Inherited Members

- Protected Member Functions inherited from IStrategy
 IStrategy ()
 
virtual ~IStrategy ()
 

Detailed Description

Definition at line 18 of file StaticRangeStrategy.hpp.

Constructor & Destructor Documentation

◆ StaticRangeStrategy()

StaticRangeStrategy ( RangeTracker &  rangeTracker)

Definition at line 17 of file StaticRangeStrategy.cpp.

References IInputSlot::GetConnection(), IConnectableLayer::GetInputSlot(), IOutputSlot::GetOwningLayerGuid(), RangeTracker::GetRange(), and RangeTracker::SetRange().

18  : m_RangeTracker(rangeTracker)
19 {}

◆ ~StaticRangeStrategy()

~StaticRangeStrategy ( )
default

Member Function Documentation

◆ ExecuteStrategy()

void ExecuteStrategy ( const armnn::IConnectableLayer *  layer,
const BaseDescriptor &  descriptor,
const std::vector< armnn::ConstTensor > &  constants,
const char *  name,
const armnn::LayerBindingId  id 
)
overridevirtual

Implements IStrategy.

Definition at line 33 of file StaticRangeStrategy.cpp.

References armnn::Abs, armnn::Activation, armnn::Addition, armnn::ArgMinMax, armnn::BatchNormalization, armnn::BatchToSpaceNd, armnn::BoundedReLu, IOutputSlot::CalculateIndexOnOwner(), armnn::Concat, armnn::Constant, armnn::Convolution2d, armnn::DepthwiseConvolution2d, armnn::Float32, armnn::FullyConnected, IInputSlot::GetConnection(), IConnectableLayer::GetInputSlot(), IConnectableLayer::GetNumInputSlots(), IOutputSlot::GetOwningLayerGuid(), RangeTracker::GetRange(), IConnectableLayer::GetType(), armnn::IgnoreUnused(), armnn::LeakyReLu, armnn::Linear, ActivationDescriptor::m_A, ActivationDescriptor::m_Function, armnn::Permute, armnn::Pooling2d, armnn::ReLu, armnn::Reshape, armnn::Resize, armnn::Softmax, armnn::SoftReLu, armnn::SpaceToBatchNd, armnn::Splitter, armnn::StridedSlice, and armnn::TanH.

38 {
39 IgnoreUnused(id, name);
40 
41 switch (layer->GetType())
42 {
43  case armnn::LayerType::Activation :
44  {
45  const ActivationDescriptor& activationDescriptor = static_cast<const ActivationDescriptor&>(descriptor);
46 
47  switch (activationDescriptor.m_Function)
48  {
49  // Range is 0, 15 for Abs, Linear, ReLu and Soft ReLu
50  case ActivationFunction::Abs:
51  case ActivationFunction::Linear:
52  case ActivationFunction::ReLu:
53  case ActivationFunction::SoftReLu:
54  SetRange(layer, 0, 0.f, 15.f);
55  break;
56  case ActivationFunction::BoundedReLu:
57  SetRange(layer, 0, 0.f, activationDescriptor.m_A);
58  break;
59  case ActivationFunction::TanH:
60  SetRange(layer, 0, -1.f, 1.f);
61  break;
62  case ActivationFunction::LeakyReLu:
63  SetRange(layer, 0, -5.f, 15.f);
64  break;
65  default:
66  SetRange(layer, 0, -15.f, 15.f);
67  break;
68  }
69  break;
70  }
71  case armnn::LayerType::Addition :
72  {
73  SetRange(layer, 0, -20.f, 20.f);
74  break;
75  }
76  case armnn::LayerType::ArgMinMax :
77  {
78  ForwardParentParameters(layer);
79  break;
80  }
81  case armnn::LayerType::BatchToSpaceNd :
82  {
83  ForwardParentParameters(layer);
84  break;
85  }
86  case armnn::LayerType::BatchNormalization :
87  {
88  SetRange(layer, 0, -15.0f, 15.0f);
89  break;
90  }
91  case armnn::LayerType::Concat :
92  {
93  float min = std::numeric_limits<float>::max();
94  float max = std::numeric_limits<float>::lowest();
95  for (unsigned int i = 0; i < layer->GetNumInputSlots(); ++i)
96  {
97  const IOutputSlot* outputSlot = layer->GetInputSlot(i).GetConnection();
98  LayerGuid layerId = outputSlot->GetOwningLayerGuid();
99  unsigned int slotIndex = outputSlot->CalculateIndexOnOwner();
100  RangeTracker::MinMaxRange range = m_RangeTracker.GetRange(layerId, slotIndex);
101  min = std::min(min, range.first);
102  max = std::max(max, range.second);
103  }
104  SetRange(layer, 0, min, max);
105  break;
106  }
107  case armnn::LayerType::Constant :
108  {
109 
110  if (constants[0].GetDataType() != DataType::Float32)
111  {
112  throw InvalidArgumentException("Quantization is supported only for FP32 tensors");
113  }
114 
115  // Work out the range based on the input constants
116  unsigned int inputNumElements = constants[0].GetNumElements();
117  const float* inputData = reinterpret_cast<const float*>(constants[0].GetMemoryArea());
118 
119  float min = std::numeric_limits<float>::max();
120  float max = std::numeric_limits<float>::lowest();
121 
122  for (unsigned int i = 0; i < inputNumElements; i++)
123  {
124  const float inputValue = inputData[i];
125 
126  min = std::min(min, inputValue);
127  max = std::max(max, inputValue);
128  }
129  SetRange(layer, 0, min, max);
130  break;
131  }
132  case armnn::LayerType::Convolution2d :
133  {
134  SetRange(layer, 0, -15.0f, 15.0f);
135  break;
136  }
137  case armnn::LayerType::DepthwiseConvolution2d :
138  {
139  SetRange(layer, 0, -15.0f, 15.0f);
140  break;
141  }
142  case armnn::LayerType::FullyConnected :
143  {
144  SetRange(layer, 0, -15.0f, 15.0f);
145  break;
146  }
147  case armnn::LayerType::Permute :
148  {
149  ForwardParentParameters(layer);
150  break;
151  }
152  case armnn::LayerType::Pooling2d :
153  {
154  ForwardParentParameters(layer);
155  break;
156  }
157  case armnn::LayerType::Reshape :
158  {
159  ForwardParentParameters(layer);
160  break;
161  }
162  case armnn::LayerType::Resize :
163  {
164  ForwardParentParameters(layer);
165  break;
166  }
167  case armnn::LayerType::Splitter :
168  {
169  ForwardParentParameters(layer);
170  break;
171  }
172  case armnn::LayerType::SpaceToBatchNd :
173  {
174  ForwardParentParameters(layer);
175  break;
176  }
177  case armnn::LayerType::Softmax :
178  {
179  SetRange(layer, 0, 0.f, 1.f);
180  break;
181  }
182  case armnn::LayerType::StridedSlice :
183  {
184  ForwardParentParameters(layer);
185  break;
186  }
187  default:
188  {
189  }
190 }
191 }
armnn::IConnectableLayer::GetNumInputSlots
virtual unsigned int GetNumInputSlots() const =0
Returns the number of connectable input slots.
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
armnn::RangeTracker::MinMaxRange
std::pair< float, float > MinMaxRange
Definition: RangeTracker.hpp:20
armnn::RangeTracker::GetRange
MinMaxRange GetRange(LayerGuid guid, unsigned int idx) const
Retrieve the Range for a particular output slot on a particular layer.
Definition: RangeTracker.cpp:29
armnn::ActivationFunction::BoundedReLu
min(a, max(b, input)) ReLu1 & ReLu6.
armnn::IConnectableLayer::GetType
virtual LayerType GetType() const =0
Returns the armnn::LayerType of this layer.
armnn::LayerGuid
profiling::ProfilingGuid LayerGuid
Define LayerGuid type.
Definition: Types.hpp:275
armnn::IConnectableLayer::GetInputSlot
virtual const IInputSlot & GetInputSlot(unsigned int index) const =0
Get a const input slot handle by slot index.
armnn::IInputSlot::GetConnection
virtual const IOutputSlot * GetConnection() const =0
The documentation for this class was generated from the following files: