diff options
Diffstat (limited to 'ethosu/vela/tosa/Attribute.py')
-rw-r--r-- | ethosu/vela/tosa/Attribute.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/tosa/Attribute.py b/ethosu/vela/tosa/Attribute.py index 8a2ccaea..b26ef483 100644 --- a/ethosu/vela/tosa/Attribute.py +++ b/ethosu/vela/tosa/Attribute.py @@ -4,9 +4,9 @@ class Attribute(object): NONE = 0 - Pool2dAttribute = 1 - Conv2dAttribute = 2 - TransposeConv2dAttribute = 3 + PoolAttribute = 1 + ConvAttribute = 2 + TransposeConvAttribute = 3 ReluNAttribute = 4 AxisAttribute = 5 ReshapeAttribute = 6 |