aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/operation.py
diff options
context:
space:
mode:
authorerik.andersson@arm.com <erik.andersson@arm.com>2022-08-10 15:26:48 +0200
committererik.andersson@arm.com <erik.andersson@arm.com>2022-08-10 17:52:50 +0200
commitdd49a7205b76d668308262b2ea4cc38968ff576c (patch)
tree5f879db220aa77b05c86cb42e393f50d534f997c /ethosu/vela/operation.py
parent885033b5bf2f6513b438f273b2bc71964f0c6c59 (diff)
downloadethos-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.py2
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):