diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-01 12:43:02 +0200 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-09-07 15:35:49 +0200 |
commit | f1580f0167d7e9539a17ac8e33b0b595300f8090 (patch) | |
tree | 2fe1d9d4715ac38be9cfdc5fbe049c07ab5b9563 /ethosu/vela/operation_util.py | |
parent | 94292fe32c34357a8935a42c77b759a499eb0db9 (diff) | |
download | ethos-u-vela-f1580f0167d7e9539a17ac8e33b0b595300f8090.tar.gz |
TOSA: Added RESHAPE, SLICE and CONCAT
Added support for Data layout ops
RESHAPE, SLICE and CONCAT.
-No support for bool_t
-Support limited to Rank <= 4 and N = 1
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I487ac494b6506a2a6ba947ee758aa193194dd796
Diffstat (limited to 'ethosu/vela/operation_util.py')
-rw-r--r-- | ethosu/vela/operation_util.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/ethosu/vela/operation_util.py b/ethosu/vela/operation_util.py index 4a4fd335..0fbed46b 100644 --- a/ethosu/vela/operation_util.py +++ b/ethosu/vela/operation_util.py @@ -44,6 +44,12 @@ def create_avgpool_nop(name: str) -> Operation: return op +def create_add_nop(name: str) -> Operation: + op = Operation(Op.Add, name) + op.run_on_npu = True + return op + + def create_depthwise_maxpool( name: str, ifm: Tensor, |