From fd33a698ee3c588aa4064b70b7781ab25ff76f66 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 29 Jun 2022 15:35:57 +0100 Subject: IVGCVSW-7040 Add support for LOG_SOFTMAX to the TFLiteParser Signed-off-by: Teresa Charlin Change-Id: I1fedfdf2cd8871d6b307fce8620f40adadf75f04 --- src/armnnTfLiteParser/TfLiteParser.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/armnnTfLiteParser/TfLiteParser.hpp') 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); -- cgit v1.2.1