From d9c2c4258c50f04ef3a6c3849508d317249e8ebf Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Tue, 1 Dec 2020 16:33:45 +0100 Subject: [MLBEDSW-3690] Refactor Softmax Move operator generation code to common functions. Signed-off-by: Fredrik Svedberg Change-Id: I02e185fd793a96ae435fa7d235c9d1e97f388a03 --- ethosu/vela/graph_optimiser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/graph_optimiser.py') 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 -- cgit v1.2.1