diff options
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 0faa079..c599019 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: ( |