MLBEDSW-3694 Replace padding with enum
Use an Enum instead of a bytestring to specify VALID or SAME padding
Signed-off-by: Michael McGeagh <michael.mcgeagh@arm.com>
Change-Id: I4e87f8c32b3bfac176d822a68de061e85a558fce
diff --git a/ethosu/vela/operation_util.py b/ethosu/vela/operation_util.py
index 2fc7622..a267b2a 100644
--- a/ethosu/vela/operation_util.py
+++ b/ethosu/vela/operation_util.py
@@ -22,6 +22,7 @@
from .operation import ActivationFunction
from .operation import Op
from .operation import Operation
+from .operation import Padding
from .tensor import create_reshape_tensor
from .tensor import QuantizationParameters
from .tensor import Tensor
@@ -29,7 +30,7 @@
def create_avgpool_nop(name: str) -> Operation:
op = Operation(Op.AvgPool, name)
- op.attrs["padding"] = b"VALID"
+ op.attrs["padding"] = Padding.VALID
op.attrs["stride_w"] = 1
op.attrs["stride_h"] = 1
op.attrs["filter_width"] = 1
@@ -48,7 +49,7 @@
height = ifm.shape[1] * ifm.shape[2]
width = ifm.shape[3]
ifm_shape = [1, height, width, 1]
- op.attrs["padding"] = b"VALID"
+ op.attrs["padding"] = Padding.VALID
op.attrs["stride_w"] = 1
op.attrs["stride_h"] = 1
op.attrs["filter_width"] = width
@@ -67,7 +68,7 @@
name: str, ifm: Tensor, quantization: QuantizationParameters, activation: Optional[ActivationFunction] = None
) -> Operation:
op = Operation(Op.ReduceSum, name)
- op.attrs["padding"] = b"VALID"
+ op.attrs["padding"] = Padding.VALID
op.attrs["stride_w"] = 1
op.attrs["stride_h"] = 1
op.attrs["filter_width"] = 1