aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfLiteParser/TfLiteParser.hpp
diff options
context:
space:
mode:
authorNina Drozd <nina.drozd@arm.com>2019-04-15 09:47:39 +0100
committerNina Drozd <nina.drozd@arm.com>2019-04-15 14:37:58 +0100
commit200e38039cf2cef21ae9ba6f86fab6fd524e5077 (patch)
treeb1bf4335036c1e8047130b2e82a1d2067be5e8fe /src/armnnTfLiteParser/TfLiteParser.hpp
parentb1390fcea347e8e0476a961c128f2334c4d73217 (diff)
downloadarmnn-200e38039cf2cef21ae9ba6f86fab6fd524e5077.tar.gz
IVGCVSW-2848 - Add TfLite Parser support for Unpack layer
* Added ParseUnpack in TfLiteParser * New Unpack test file with test reproducing unpack in DeepSpeechV1 model * Added documentation for supported Unpack to TensorflorLiteSupport.md Signed-off-by: Nina Drozd <nina.drozd@arm.com> Change-Id: Ie920d46254ff4b4ab544407ace4c1d489af83157
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
-rw-r--r--src/armnnTfLiteParser/TfLiteParser.hpp13
1 files changed, 7 insertions, 6 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp
index a3ef22fa7e..929af1f0db 100644
--- a/src/armnnTfLiteParser/TfLiteParser.hpp
+++ b/src/armnnTfLiteParser/TfLiteParser.hpp
@@ -91,6 +91,7 @@ private:
void ParseUnsupportedOperator(size_t subgraphIndex, size_t operatorIndex);
void ParseActivation(size_t subgraphIndex, size_t operatorIndex, armnn::ActivationFunction activationType);
+ void ParseAdd(size_t subgraphIndex, size_t operatorIndex);
void ParseAveragePool2D(size_t subgraphIndex, size_t operatorIndex);
void ParseBatchToSpaceND(size_t subgraphIndex, size_t operatorIndex);
void ParseConcatenation(size_t subgraphIndex, size_t operatorIndex);
@@ -101,23 +102,23 @@ private:
void ParseLogistic(size_t subgraphIndex, size_t operatorIndex);
void ParseMaxPool2D(size_t subgraphIndex, size_t operatorIndex);
void ParseMaximum(size_t subgraphIndex, size_t operatorIndex);
+ void ParseMean(size_t subgraphIndex, size_t operatorIndex);
void ParseMinimum(size_t subgraphIndex, size_t operatorIndex);
+ void ParseMul(size_t subgraphIndex, size_t operatorIndex);
+ void ParsePad(size_t subgraphIndex, size_t operatorIndex);
+ void ParsePool(size_t subgraphIndex, size_t operatorIndex, armnn::PoolingAlgorithm algorithm);
void ParseRelu(size_t subgraphIndex, size_t operatorIndex);
void ParseRelu6(size_t subgraphIndex, size_t operatorIndex);
void ParseReshape(size_t subgraphIndex, size_t operatorIndex);
void ParseResizeBilinear(size_t subgraphIndex, size_t operatorIndex);
void ParseSoftmax(size_t subgraphIndex, size_t operatorIndex);
void ParseSpaceToBatchND(size_t subgraphIndex, size_t operatorIndex);
+ void ParseSplit(size_t subgraphIndex, size_t operatorIndex);
void ParseSqueeze(size_t subgraphIndex, size_t operatorIndex);
void ParseStridedSlice(size_t subgraphIndex, size_t operatorIndex);
void ParseSub(size_t subgraphIndex, size_t operatorIndex);
- void ParseAdd(size_t subgraphIndex, size_t operatorIndex);
- void ParseMul(size_t subgraphIndex, size_t operatorIndex);
- void ParseMean(size_t subgraphIndex, size_t operatorIndex);
- void ParsePad(size_t subgraphIndex, size_t operatorIndex);
- void ParsePool(size_t subgraphIndex, size_t operatorIndex, armnn::PoolingAlgorithm algorithm);
- void ParseSplit(size_t subgraphIndex, size_t operatorIndex);
void ParseTanH(size_t subgraphIndex, size_t operatorIndex);
+ void ParseUnpack(size_t subgraphIndex, size_t operatorIndex);
void RegisterProducerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IOutputSlot* slot);
void RegisterConsumerOfTensor(size_t subgraphIndex, size_t tensorIndex, armnn::IInputSlot* slot);