diff options
| author | Johan Alfven <johan.alfven@arm.com> | 2023-05-07 13:12:37 +0200 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-06-14 12:53:30 +0000 |
| commit | 8e525ca8aad4a52b80c1986c5067b9b74fb3e321 (patch) | |
| tree | 85f0a781d011aaf1c7631d34acea4398502a7a9f /ethosu/vela/operation.py | |
| parent | c2c3063d05494f3968a7fead1b3118602fe100b9 (diff) | |
| download | ethos-u-vela-8e525ca8aad4a52b80c1986c5067b9b74fb3e321.tar.gz | |
MLBEDSW-7748: Add RSQRT support
- Added RSQRT int8 support, implemented as LUT.
- Added test to supported operators
- Updated SUPPORTED_OPS.md
Change-Id: I34904772e044be8d22a6dfe426edf85358a205b7
Signed-off-by: Johan Alfven <johan.alfven@arm.com>
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 52f06cf0..da923119 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -263,7 +263,7 @@ class Op(Enum): ReverseV2 = OperatorInfo() Rnn = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) Round = OperatorInfo() - Rsqrt = OperatorInfo() + Rsqrt = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_INDICES, is_unary=True) SHL = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) # NPU specific operation SHR = OperatorInfo(block_type=NpuBlockType.ElementWise, indices=NNG_IFM_IFM2_INDICES) # NPU specific operation ScatterNd = OperatorInfo() |