diff options
author | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-06-29 09:46:30 +0200 |
---|---|---|
committer | erik.andersson@arm.com <erik.andersson@arm.com> | 2022-07-14 14:09:44 +0200 |
commit | 93f492bae9c4dd16a1f64b851b237263695ee03e (patch) | |
tree | b3c3c7b8d091ebb737dc53df23b690eadea87335 /ethosu/vela/operation.py | |
parent | d03dc504452dbb32db383121f3dad81f4280bb3c (diff) | |
download | ethos-u-vela-93f492bae9c4dd16a1f64b851b237263695ee03e.tar.gz |
MLBEDSW-6635: Update to TensorFlow 2.9
Update the flatbuffers generated code to comply with TensorFlow 2.9
Signed-off-by: erik.andersson@arm.com <erik.andersson@arm.com>
Change-Id: I6bf506ffb85da2d4a57a32198b471513deeaca73
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 9488fb7f..5ed18621 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 @@ -319,6 +320,7 @@ class Op(Enum): Multinomial = OperatorInfo() Gelu = OperatorInfo() DynamicUpdateSlice = OperatorInfo() + UnsortedSegmentProd = OperatorInfo() @property def info(self): |