diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2023-09-05 11:38:19 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2024-01-30 14:14:52 +0000 |
commit | fdbb072dacae339dd3f8efd3fb70fa84b9296033 (patch) | |
tree | 36d5704fc2e99f429bd3662f15916335e97e27f8 /ethosu/vela/tflite_mapping.py | |
parent | 014bc28324014dc82e3b352d7a81df9ed0338c56 (diff) | |
download | ethos-u-vela-fdbb072dacae339dd3f8efd3fb70fa84b9296033.tar.gz |
MLBEDSW-8491: Add support for Mirror pad
Change-Id: I3c13118e14195a5fb8e522a38b205b75fb07b74b
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 0faa079b..c599019b 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -867,7 +867,11 @@ builtin_operator_map = { OptionsSerializer("SquaredDifferenceOptions"), TFLITE_IFM_IFM2_INDICES, ), - BuiltinOperator.MIRROR_PAD: (Op.MirrorPad, OptionsSerializer("MirrorPadOptions", ("mode",)), TFLITE_NO_INDICES), + BuiltinOperator.MIRROR_PAD: ( + Op.MirrorPad, + OptionsSerializer("MirrorPadOptions", ("mode",)), + TFLITE_IFM_IFM2_INDICES, + ), BuiltinOperator.ABS: (Op.Abs, OptionsSerializer("AbsOptions"), TFLITE_IFM_INDICES), BuiltinOperator.SPLIT_V: (Op.SplitV, OptionsSerializer("SplitVOptions", ("num_splits",)), TFLITE_IFM_INDICES), BuiltinOperator.UNIQUE: ( |