From 79a4199be5b87eb6220b6688b02f5f6cbebb6131 Mon Sep 17 00:00:00 2001
From: Kevin Cheng <kevin.cheng@arm.com>
Date: Tue, 31 Aug 2021 16:04:40 -0700
Subject: 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
---
 python/tosa/Attribute.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'python/tosa/Attribute.py')

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
-- 
cgit v1.2.1