diff options
Diffstat (limited to 'ethosu/vela/shared_buffer_allocation.py')
-rw-r--r-- | ethosu/vela/shared_buffer_allocation.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py index b5408d19..29be6d8d 100644 --- a/ethosu/vela/shared_buffer_allocation.py +++ b/ethosu/vela/shared_buffer_allocation.py @@ -19,10 +19,9 @@ # Shared buffer allocation works out how to allocate the Ethos-U55 shared buffer for a given pass. import numpy as np -from .nn_graph import NpuBlockType -from .numeric_util import round_up_divide, round_up + +from .operation import NpuBlockType from .architecture_features import Block, Kernel, SHRAMElements, SharedBufferArea, ArchitectureFeatures -from . import pass_packing class SharedBufferAllocation: |