diff options
Diffstat (limited to 'ethosu/vela/extract_npu_subgraphs.py')
| -rw-r--r-- | ethosu/vela/extract_npu_subgraphs.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/extract_npu_subgraphs.py b/ethosu/vela/extract_npu_subgraphs.py index 5b9ba8b0..ab3db21f 100644 --- a/ethosu/vela/extract_npu_subgraphs.py +++ b/ethosu/vela/extract_npu_subgraphs.py @@ -23,10 +23,11 @@ # by NpuOp operations. Later, Vela generates command streams and compressed weight streams for the NPU subgraphs and # attaches them to the NpuOp. This encapsulates everything the NPU subgraph is supposed to do. -from .nn_graph import Pass, PassPlacement, NpuBlockType, Subgraph -from .operation import Operation import numpy as np +from .nn_graph import Pass, PassPlacement, Subgraph +from .operation import Operation, NpuBlockType + def make_npu_call_op_pass(npu_subgraph): op = Operation("NpuOp", "call_" + npu_subgraph.name) |