diff options
author | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-08-10 15:26:48 +0200 |
---|---|---|
committer | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-08-10 17:52:50 +0200 |
commit | dd49a7205b76d668308262b2ea4cc38968ff576c (patch) | |
tree | 5f879db220aa77b05c86cb42e393f50d534f997c /ethosu/vela/operation.py | |
parent | 885033b5bf2f6513b438f273b2bc71964f0c6c59 (diff) | |
download | ethos-u-vela-dd49a7205b76d668308262b2ea4cc38968ff576c.tar.gz |
Revert reversion of TensorFlow 2.9 update3.5.0.rc1
With the errors caused by the previous TensorFlow 2.9 update
being fixed, we can proceed with the upgrade.
Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: Ie1f025e8d984efaebc68b8d051126d49bee6b2b8
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 1a34d0e1..47f4fe0f 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -241,6 +241,7 @@ class Op(Enum): Rank = OperatorInfo() ReduceSum = OperatorInfo(block_type=NpuBlockType.ReduceSum, indices=NNG_IFM_INDICES) Relu = OperatorInfo(indices=NNG_IFM_INDICES) + Relu0To1 = OperatorInfo(indices=NNG_IFM_INDICES) Relu6 = OperatorInfo(indices=NNG_IFM_INDICES) ReluN1To1 = OperatorInfo(indices=NNG_IFM_INDICES) ReluN = OperatorInfo(indices=NNG_IFM_INDICES) # TOSA specific @@ -320,6 +321,7 @@ class Op(Enum): Multinomial = OperatorInfo() Gelu = OperatorInfo() DynamicUpdateSlice = OperatorInfo() + UnsortedSegmentProd = OperatorInfo() @property def info(self): |