diff options
Diffstat (limited to 'ethosu/vela/nn_graph.py')
| -rw-r--r-- | ethosu/vela/nn_graph.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 7dc2d722..97afde31 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -44,6 +44,11 @@ class TensorAllocator(enum.Enum): return self.name +class NetworkType(enum.Enum): + TFLite = 1 + TOSA = 2 + + class Pass: def __init__(self, name, placement, is_element_wise, npu_block_type): self.inputs = [] |