diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2022-06-29 15:35:57 +0100 |
|---|---|---|
| committer | Nikhil Raj <nikhil.raj@arm.com> | 2022-07-08 15:20:34 +0100 |
| commit | fd33a698ee3c588aa4064b70b7781ab25ff76f66 (patch) | |
| tree | 96c5bd3fa17623c49015df13f92ea06e32dfc61f /src/armnnTfLiteParser/TfLiteParser.hpp | |
| parent | 452c58080e9f8f577de87e0c07d0097aac97f3b8 (diff) | |
| download | armnn-fd33a698ee3c588aa4064b70b7781ab25ff76f66.tar.gz | |
IVGCVSW-7040 Add support for LOG_SOFTMAX to the TFLiteParser
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I1fedfdf2cd8871d6b307fce8620f40adadf75f04
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp')
| -rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index 3fc72057c7..43c5466e69 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -146,6 +146,7 @@ private: void ParseLocalResponseNormalization(size_t subgraphIndex, size_t operatorIndex); void ParseLogicalNot(size_t subgraphIndex, size_t operatorIndex); void ParseLogistic(size_t subgraphIndex, size_t operatorIndex); + void ParseLogSoftmax(size_t subgraphIndex, size_t operatorIndex); void ParseL2Normalization(size_t subgraphIndex, size_t operatorIndex); void ParseMaxPool2D(size_t subgraphIndex, size_t operatorIndex); void ParseMaximum(size_t subgraphIndex, size_t operatorIndex); |