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/graph_optimiser.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/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 85c2b602..4a857507 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -32,10 +32,10 @@ from .numeric_util import clamp_sigmoid from .numeric_util import full_shape from .numeric_util import round_away_zero from .operation import create_activation_function -from .operation import create_avgpool_nop from .operation import NpuBlockType from .operation import Op from .operation import Operation +from .operation_util import create_avgpool_nop from .softmax import SoftMax from .tensor import check_quantized_tens_scaling_equal from .tensor import create_const_tensor |