Add CumSum&LRN operator to trace module (#639)

2023-08-22 16:53:59 +08:00 · 2023-08-22 16:53:59 +08:00 · 54af5c2216
parent bab571b569
commit 54af5c2216
1 changed files with 2 additions and 0 deletions
--- a/include/tim/experimental/trace/tvx/ops.h
+++ b/include/tim/experimental/trace/tvx/ops.h
@ -153,6 +153,8 @@
  (UnidirectionalSequenceRnnExt)                                               \
  (Unstack)                                                                    \
  (Conv3d)                                                                     \
+  (CumSum)                                                                     \
+  (LocalResponseNormalization)                                                 \
  (CustomOpBase)                                                               \
  (Topk)                                                                       \
  (BidirectionalSequenceLstm)                                                  \