diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-04-09 14:38:52 +0100 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-04-09 16:34:18 +0100 |
commit | 28f177c45cb8e23b86c9f3ff435921183340fc33 (patch) | |
tree | f44a2a25e918191cf5f841f03a1a09ab264011bf /docs/01_01_parsers.dox | |
parent | 1f140f7226c4ed7bc5cbaf2ce09654eee452f4bf (diff) | |
download | armnn-28f177c45cb8e23b86c9f3ff435921183340fc33.tar.gz |
IVGCVSW-5804 TfLiteParser fails to correctly parse ArgMinMax
* Fix for GitHub#523.
* Updated ParseArgMinMax function to read correct axis data.
* Improved validation in ParseArgMinMax function.
* Added ARG_MIN support to TfLiteParser.
* Added ArgMinMax unit tests for TfLiteParser.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: Ib4ce1a7c66e210c47859a130c4896aac958f2654
Diffstat (limited to 'docs/01_01_parsers.dox')
-rw-r--r-- | docs/01_01_parsers.dox | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/docs/01_01_parsers.dox b/docs/01_01_parsers.dox index ae49303876..05e1225f93 100644 --- a/docs/01_01_parsers.dox +++ b/docs/01_01_parsers.dox @@ -158,6 +158,8 @@ This reference guide provides a list of TensorFlow Lite operators the Arm NN SDK The Arm NN SDK TensorFlow Lite parser currently supports the following operators: - ADD +- ARG_MAX +_ ARG_MIN - AVERAGE_POOL_2D, Supported Fused Activation: RELU , RELU6 , TANH, NONE - BATCH_TO_SPACE - CONCATENATION, Supported Fused Activation: RELU , RELU6 , TANH, NONE |