1 files changed, 17 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index de97710a..257cb5ff 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -19,6 +19,7 @@ import copy
 import enum
 import uuid
 from collections import defaultdict
+from enum import auto
 from functools import lru_cache
 from typing import Dict
 from typing import List
@@ -62,6 +63,22 @@ class MemType(enum.IntFlag):
         return self.name
 
 
+class BandwidthDirection(enum.IntEnum):
+    Read = 0
+    Write = auto()
+    Size = auto()
+
+    def display_name(self):
+        return self.name
+
+    def identifier_name(self):
+        return self.name.lower()
+
+    @staticmethod
+    def all():
+        return (BandwidthDirection.Read, BandwidthDirection.Write)
+
+
 class MemArea(enum.IntFlag):
     Unknown = 0
     Sram = 1