diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-12-01 16:33:45 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-12-08 14:00:17 +0000 |
commit | d9c2c4258c50f04ef3a6c3849508d317249e8ebf (patch) | |
tree | 6604f3fef997fb946884bf5bd2f54236b523393f /ethosu/vela/pass_packing.py | |
parent | d5cf7650b6bcfa5d81321661fa300763660d31ab (diff) | |
download | ethos-u-vela-d9c2c4258c50f04ef3a6c3849508d317249e8ebf.tar.gz |
[MLBEDSW-3690] Refactor Softmax
Move operator generation code to common functions.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I02e185fd793a96ae435fa7d235c9d1e97f388a03
Diffstat (limited to 'ethosu/vela/pass_packing.py')
-rw-r--r-- | ethosu/vela/pass_packing.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 59376a85..ea2eaa4f 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -21,9 +21,9 @@ import enum from .debug_database import DebugDatabase from .nn_graph import Pass from .nn_graph import PassPlacement -from .operation import create_avgpool_nop from .operation import NpuBlockType from .operation import Op +from .operation_util import create_avgpool_nop from .tensor import TensorPurpose |