diff options
| author | Rickard Bolin <rickard.bolin@arm.com> | 2022-12-19 12:33:40 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-03-22 10:11:21 +0000 |
| commit | 6986a079020ab6344c9191aa67af13beeb475593 (patch) | |
| tree | c5cfef45fd915dc44b313f5fd8510f375dafa806 /ethosu/vela/tflite_mapping.py | |
| parent | 3340a88d51cdce4d00ea82a2211d9e7fd40cf5c5 (diff) | |
| download | ethos-u-vela-6986a079020ab6344c9191aa67af13beeb475593.tar.gz | |
MLBEDSW-6435: Implement support for ArgMax along depth dimension
- Add support for ArgMax along depth dimension with a depth limit of 127.
- Only supports 8-bit input and 32-bit output
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I5f6f0503135bebabbb1ca637f9729587b7c60740
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -rw-r--r-- | ethosu/vela/tflite_mapping.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 8ec01737..98fe287d 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2020-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2020-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> # # SPDX-License-Identifier: Apache-2.0 # @@ -749,7 +749,7 @@ builtin_operator_map = { BuiltinOperator.ARG_MAX: ( Op.ArgMax, OptionsSerializer("ArgMaxOptions", (("output_type", datatype_deserialize, datatype_serialize),)), - TFLITE_NO_INDICES, + TFLITE_IFM_INDICES, ), BuiltinOperator.MINIMUM: (Op.Minimum, OptionsSerializer("MaximumMinimumOptions"), TFLITE_IFM_IFM2_INDICES), BuiltinOperator.LESS: (Op.Less, OptionsSerializer("LessOptions"), TFLITE_NO_INDICES), |