diff options
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): |