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/operation.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/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 14 |
1 files changed, 0 insertions, 14 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 5cb4b6ae..45fae217 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -355,20 +355,6 @@ def create_activation_function(op_type: Op) -> ActivationFunction: return act -def create_avgpool_nop(name): - op = Operation(Op.AvgPool, name) - op.attrs["padding"] = b"VALID" - op.attrs["stride_w"] = 1 - op.attrs["stride_h"] = 1 - op.attrs["filter_width"] = 1 - op.attrs["filter_height"] = 1 - op.attrs["strides"] = [1, 1, 1, 1] - op.attrs["ksize"] = [1, 1, 1, 1] - op.attrs["skirt"] = [0, 0, 0, 0] - op.attrs["explicit_padding"] = [0, 0, 0, 0] - return op - - def get_slice_offsets(input_shape, offset_tens, offset_mask, is_begin=True): # For strided slice operator: get start or end offsets offsets = len(input_shape) * [0] if is_begin else input_shape[:] |