diff options
Diffstat (limited to 'include/operator.def')
-rw-r--r-- | include/operator.def | 22 |
1 files changed, 18 insertions, 4 deletions
diff --git a/include/operator.def b/include/operator.def index 85bb5c9..0bd0d08 100644 --- a/include/operator.def +++ b/include/operator.def @@ -1,8 +1,8 @@ -// Copyright (c) 2020-2021, ARM Limited. +// Copyright (c) 2020-2024, ARM Limited. // -// Licensed under the Apache License, Version 2.0 with LLVM Exceptions -// (the "License"); you may not use this file except in compliance with +// Licensed under the Apache License, Version 2.0 with LLVM Exceptions +// (the "License"); you may not use this file except in compliance with // the License. You may obtain a copy of the License at // // https://llvm.org/LICENSE.txt @@ -27,12 +27,15 @@ DEF_OPERATOR(AvgPool2d) DEF_OPERATOR(Conv2D) DEF_OPERATOR(Conv3D) DEF_OPERATOR(DepthwiseConv2D) +DEF_OPERATOR(FFT2d) DEF_OPERATOR(FullyConnected) DEF_OPERATOR(MatMul) DEF_OPERATOR(MaxPool2d) +DEF_OPERATOR(RFFT2d) DEF_OPERATOR(TransposeConv2D) /* activation */ +DEF_OPERATOR(Erf) DEF_OPERATOR(Clamp) DEF_OPERATOR(Sigmoid) DEF_OPERATOR(Tanh) @@ -43,7 +46,7 @@ DEF_OPERATOR(ArithmeticRightShift) DEF_OPERATOR(BitwiseAnd) DEF_OPERATOR(BitwiseOr) DEF_OPERATOR(BitwiseXor) -DEF_OPERATOR(Div) +DEF_OPERATOR(IntDiv) DEF_OPERATOR(LogicalAnd) DEF_OPERATOR(LogicalLeftShift) DEF_OPERATOR(LogicalRightShift) @@ -61,6 +64,7 @@ DEF_OPERATOR(Abs) DEF_OPERATOR(BitwiseNot) DEF_OPERATOR(Ceil) DEF_OPERATOR(Clz) +DEF_OPERATOR(Cos) DEF_OPERATOR(Exp) DEF_OPERATOR(Floor) DEF_OPERATOR(Log) @@ -68,6 +72,7 @@ DEF_OPERATOR(LogicalNot) DEF_OPERATOR(Negate) DEF_OPERATOR(Reciprocal) DEF_OPERATOR(Rsqrt) +DEF_OPERATOR(Sin) /* elementwise - ternary */ DEF_OPERATOR(Select) @@ -88,6 +93,7 @@ DEF_OPERATOR(ReduceSum) /* memory operation */ DEF_OPERATOR(Concat) DEF_OPERATOR(Pad) +DEF_OPERATOR(Dim) DEF_OPERATOR(Reshape) DEF_OPERATOR(Reverse) DEF_OPERATOR(Slice) @@ -115,3 +121,11 @@ DEF_OPERATOR(Custom) /* control flow operators */ DEF_OPERATOR(If) DEF_OPERATOR(While) + +/* shape operators */ +DEF_OPERATOR(ConstShape) +DEF_OPERATOR(ConcatShape) +DEF_OPERATOR(AddShape) +DEF_OPERATOR(SubShape) +DEF_OPERATOR(MulShape) +DEF_OPERATOR(DivShape) |