From 168954814fb6a1cc5e7b2d44784b24402ef30199 Mon Sep 17 00:00:00 2001 From: Michael McGeagh Date: Mon, 14 Dec 2020 15:51:20 +0000 Subject: MLBEDSW-3694 Replace padding with enum Use an Enum instead of a bytestring to specify VALID or SAME padding Signed-off-by: Michael McGeagh Change-Id: I4e87f8c32b3bfac176d822a68de061e85a558fce --- ethosu/vela/tflite_mapping.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/tflite_mapping.py') diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index fe582614..cc6053c0 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -24,6 +24,7 @@ import numpy as np from .data_type import DataType from .operation import CustomType from .operation import Op +from .operation import Padding as opPad from .tflite import AbsOptions from .tflite import AddNOptions from .tflite import AddOptions @@ -425,8 +426,8 @@ class CustomOptionsSerializer: padding_map = { - Padding.SAME: b"SAME", - Padding.VALID: b"VALID", + Padding.SAME: opPad.SAME, + Padding.VALID: opPad.VALID, } padding_inv_map = inverse_map(padding_map) -- cgit v1.2.1