12 #include <schema_generated.h>
14 #include <unordered_map>
17 #include <tensorflow/lite/version.h>
19 #if TF_MAJOR_VERSION > 2 || (TF_MAJOR_VERSION == 2 && TF_MINOR_VERSION > 3)
20 #define ARMNN_POST_TFLITE_2_3
30 using ModelPtr = std::unique_ptr<tflite::ModelT>;
53 const std::string& name)
const;
58 const std::string& name)
const;
91 const std::vector<int32_t>& targetDimsIn);
106 using OperatorParsingFunction = void(
TfLiteParserImpl::*)(
size_t subgraphIndex,
size_t operatorIndex);
108 void ParseCustomOperator(
size_t subgraphIndex,
size_t operatorIndex);
109 void ParseUnsupportedOperator(
size_t subgraphIndex,
size_t operatorIndex);
111 void ParseAbs(
size_t subgraphIndex,
size_t operatorIndex);
113 void ParseAdd(
size_t subgraphIndex,
size_t operatorIndex);
115 void ParseArgMin(
size_t subgraphIndex,
size_t operatorIndex);
116 void ParseArgMax(
size_t subgraphIndex,
size_t operatorIndex);
117 void ParseAveragePool2D(
size_t subgraphIndex,
size_t operatorIndex);
118 void ParseBatchMatMul(
size_t subgraphIndex,
size_t operatorIndex);
119 void ParseBatchToSpaceND(
size_t subgraphIndex,
size_t operatorIndex);
120 void ParseCast(
size_t subgraphIndex,
size_t operatorIndex);
122 void ParseConcatenation(
size_t subgraphIndex,
size_t operatorIndex);
123 void ParseConv2D(
size_t subgraphIndex,
size_t operatorIndex);
125 #if defined(ARMNN_POST_TFLITE_2_4)
126 void ParseConv3D(
size_t subgraphIndex,
size_t operatorIndex);
128 void ParseDepthToSpace(
size_t subgraphIndex,
size_t operatorIndex);
129 void ParseDepthwiseConv2D(
size_t subgraphIndex,
size_t operatorIndex);
130 void ParseDequantize(
size_t subgraphIndex,
size_t operatorIndex);
131 void ParseDetectionPostProcess(
size_t subgraphIndex,
size_t operatorIndex);
132 void ParseDiv(
size_t subgraphIndex,
size_t operatorIndex);
133 void ParseElementwiseUnary(
size_t subgraphIndex,
size_t operatorIndex,
armnn::UnaryOperation unaryOperation);
134 void ParseElu(
size_t subgraphIndex,
size_t operatorIndex);
135 void ParseEqual(
size_t subgraphIndex,
size_t operatorIndex);
136 void ParseExp(
size_t subgraphIndex,
size_t operatorIndex);
137 void ParseExpandDims(
size_t subgraphIndex,
size_t operatorIndex);
138 void ParseFloorDiv(
size_t subgraphIndex,
size_t operatorIndex);
139 void ParseFullyConnected(
size_t subgraphIndex,
size_t operatorIndex);
140 void ParseGather(
size_t subgraphIndex,
size_t operatorIndex);
141 void ParseGatherNd(
size_t subgraphIndex,
size_t operatorIndex);
142 void ParseGreater(
size_t subgraphIndex,
size_t operatorIndex);
143 void ParseGreaterOrEqual(
size_t subgraphIndex,
size_t operatorIndex);
144 void ParseHardSwish(
size_t subgraphIndex,
size_t operatorIndex);
145 void ParseLeakyRelu(
size_t subgraphIndex,
size_t operatorIndex);
146 void ParseLess(
size_t subgraphIndex,
size_t operatorIndex);
147 void ParseLessOrEqual(
size_t subgraphIndex,
size_t operatorIndex);
148 void ParseLog(
size_t subgraphIndex,
size_t operatorIndex);
149 void ParseLocalResponseNormalization(
size_t subgraphIndex,
size_t operatorIndex);
150 void ParseLogicalNot(
size_t subgraphIndex,
size_t operatorIndex);
151 void ParseLogistic(
size_t subgraphIndex,
size_t operatorIndex);
152 void ParseLogSoftmax(
size_t subgraphIndex,
size_t operatorIndex);
153 void ParseL2Normalization(
size_t subgraphIndex,
size_t operatorIndex);
154 void ParseMaxPool2D(
size_t subgraphIndex,
size_t operatorIndex);
155 void ParseMaximum(
size_t subgraphIndex,
size_t operatorIndex);
156 void ParseMean(
size_t subgraphIndex,
size_t operatorIndex);
157 void ParseMinimum(
size_t subgraphIndex,
size_t operatorIndex);
158 void ParseMirrorPad(
size_t subgraphIndex,
size_t operatorIndex);
159 void ParseMul(
size_t subgraphIndex,
size_t operatorIndex);
160 void ParseNeg(
size_t subgraphIndex,
size_t operatorIndex);
161 void ParseNotEqual(
size_t subgraphIndex,
size_t operatorIndex);
162 void ParsePack(
size_t subgraphIndex,
size_t operatorIndex);
163 void ParsePad(
size_t subgraphIndex,
size_t operatorIndex);
165 void ParsePrelu(
size_t subgraphIndex,
size_t operatorIndex);
166 void ParseQuantize(
size_t subgraphIndex,
size_t operatorIndex);
168 void ParseReduceMax(
size_t subgraphIndex,
size_t operatorIndex);
169 void ParseReduceMin(
size_t subgraphIndex,
size_t operatorIndex);
170 void ParseReduceProd(
size_t subgraphIndex,
size_t operatorIndex);
171 void ParseRelu(
size_t subgraphIndex,
size_t operatorIndex);
172 void ParseRelu6(
size_t subgraphIndex,
size_t operatorIndex);
173 void ParseReshape(
size_t subgraphIndex,
size_t operatorIndex);
174 void ParseResize(
size_t subgraphIndex,
size_t operatorIndex,
armnn::ResizeMethod resizeMethod);
175 void ParseResizeBilinear(
size_t subgraphIndex,
size_t operatorIndex);
176 void ParseResizeNearestNeighbor(
size_t subgraphIndex,
size_t operatorIndex);
177 void ParseRsqrt(
size_t subgraphIndex,
size_t operatorIndex);
178 void ParseShape(
size_t subgraphIndex,
size_t operatorIndex);
179 void ParseSin(
size_t subgraphIndex,
size_t operatorIndex);
180 void ParseSlice(
size_t subgraphIndex,
size_t operatorIndex);
181 void ParseSoftmax(
size_t subgraphIndex,
size_t operatorIndex);
182 void ParseSqrt(
size_t subgraphIndex,
size_t operatorIndex);
183 void ParseSpaceToBatchND(
size_t subgraphIndex,
size_t operatorIndex);
184 void ParseSplit(
size_t subgraphIndex,
size_t operatorIndex);
185 void ParseSplitV(
size_t subgraphIndex,
size_t operatorIndex);
186 void ParseSqueeze(
size_t subgraphIndex,
size_t operatorIndex);
187 void ParseStridedSlice(
size_t subgraphIndex,
size_t operatorIndex);
188 void ParseSub(
size_t subgraphIndex,
size_t operatorIndex);
189 void ParseSum(
size_t subgraphIndex,
size_t operatorIndex);
190 void ParseTanH(
size_t subgraphIndex,
size_t operatorIndex);
191 void ParseTranspose(
size_t subgraphIndex,
size_t operatorIndex);
192 void ParseTransposeConv(
size_t subgraphIndex,
size_t operatorIndex);
193 void ParseUnidirectionalSequenceLSTM(
size_t subgraphIndex,
size_t operatorIndex);
194 void ParseUnpack(
size_t subgraphIndex,
size_t operatorIndex);
196 void RegisterProducerOfTensor(
size_t subgraphIndex,
size_t tensorIndex,
armnn::IOutputSlot* slot);
197 void RegisterConsumerOfTensor(
size_t subgraphIndex,
size_t tensorIndex,
armnn::IInputSlot* slot);
198 void RegisterInputSlots(
size_t subgraphIndex,
199 size_t operatorIndex,
201 const std::vector<unsigned int>& tensorIndexes,
202 unsigned int startingSlotIndex = 0);
203 void RegisterOutputSlots(
size_t subgraphIndex,
204 size_t operatorIndex,
206 const std::vector<unsigned int>& tensorIndexes);
208 void SetupInputLayerTensorInfos(
size_t subgraphIndex);
209 void SetupConstantLayerTensorInfos(
size_t subgraphIndex);
211 void SetupInputLayers(
size_t subgraphIndex);
212 void SetupOutputLayers(
size_t subgraphIndex);
213 void SetupConstantLayers(
size_t subgraphIndex);
217 void AddBroadcastReshapeLayer(
size_t subgraphIndex,
218 size_t operatorIndex,
223 unsigned int outputSlot,
224 std::string reshapeLayerName,
229 unsigned int outputSlot,
230 tflite::ActivationFunctionType activationType);
237 struct SupportedDataStorage
241 SupportedDataStorage(std::unique_ptr<
float[]>&& data);
242 SupportedDataStorage(std::unique_ptr<uint8_t[]>&& data);
243 SupportedDataStorage(std::unique_ptr<int8_t[]>&& data);
244 SupportedDataStorage(std::unique_ptr<int32_t[]>&& data);
248 std::unique_ptr<float[]> m_FloatData;
249 std::unique_ptr<uint8_t[]> m_Uint8Data;
250 std::unique_ptr<int8_t[]> m_Int8Data;
251 std::unique_ptr<int32_t[]> m_Int32Data;
254 bool ShouldConstantTensorBeCreated(
unsigned int tensorIndex);
265 std::pair<armnn::ConstTensor, SupportedDataStorage>
270 std::pair<armnn::ConstTensor, std::unique_ptr<float[]>>
276 std::pair<armnn::ConstTensor, TfLiteParserImpl::SupportedDataStorage>
282 std::pair<armnn::ConstTensor*, std::unique_ptr<float[]>>
287 size_t operatorIndex,
291 size_t operatorIndex,
294 std::vector<int> inputs);
297 size_t operatorIndex,
300 std::vector<armnn::TensorShape> inputShapes = {});
309 std::vector<OperatorParsingFunction> m_ParserFunctions;
310 std::unordered_map<std::string, OperatorParsingFunction> m_CustomParserFunctions;
318 std::vector<armnn::IInputSlot*> inputSlots;
320 TensorSlots() : outputSlot(nullptr) { }
322 typedef std::vector<TensorSlots> TensorConnections;
325 std::vector<TensorConnections> m_SubgraphConnections;
329 std::vector<std::vector<unsigned int>> m_OverriddenOutputShapes;
331 std::vector<unsigned int> m_ConstantsToDequantize;
332 std::vector<unsigned int> m_ConstantsToBeCreated;
333 std::map<size_t, armnn::TensorInfo> m_TensorInfos;