diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 16:04:40 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-08-31 17:23:15 -0700 |
| commit | 79a4199be5b87eb6220b6688b02f5f6cbebb6131 (patch) | |
| tree | 01ac992667773b5b3506a4a3fed1a3fad85088ab /python/tosa/Attribute.py | |
| parent | ab905ec865a9f4889e6818ce1c4bc934cff2070d (diff) | |
| download | serialization_lib-79a4199be5b87eb6220b6688b02f5f6cbebb6131.tar.gz | |
Adding Conv attribute to Conv3d.
- Rename attribute: Pool2d, Conv2d, TransposeConv2d -> Pool, Conv, TransposeConv
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: If47dfd1ffc57843018d81d674e4386212fd8d4b4
Diffstat (limited to 'python/tosa/Attribute.py')
| -rw-r--r-- | python/tosa/Attribute.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/tosa/Attribute.py b/python/tosa/Attribute.py index e70c0ac..334b4ae 100644 --- a/python/tosa/Attribute.py +++ b/python/tosa/Attribute.py @@ -19,9 +19,9 @@ class Attribute(object): NONE = 0 - Pool2dAttribute = 1 - Conv2dAttribute = 2 - TransposeConv2dAttribute = 3 + PoolAttribute = 1 + ConvAttribute = 2 + TransposeConvAttribute = 3 ReluNAttribute = 4 AxisAttribute = 5 ReshapeAttribute = 6 |