// 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
//    the License. You may obtain a copy of the License at
//
//         https://llvm.org/LICENSE.txt
//
//    Unless required by applicable law or agreed to in writing, software
//    distributed under the License is distributed on an "AS IS" BASIS,
//    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//    See the License for the specific language governing permissions and
//    limitations under the License.

/*
  Syntax:
    DEF_OPERATOR(MLIR_OP)

  Description:
    MLIR_OP: the operator class type, must match mlir/include/mlir/Dialect/Tosa/IR/tosa_ops.td in llvm-project repo
*/

/* tensor operators */
DEF_OPERATOR(ArgMax)
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)

/* elementwise - binary */
DEF_OPERATOR(Add)
DEF_OPERATOR(ArithmeticRightShift)
DEF_OPERATOR(BitwiseAnd)
DEF_OPERATOR(BitwiseOr)
DEF_OPERATOR(BitwiseXor)
DEF_OPERATOR(IntDiv)
DEF_OPERATOR(LogicalAnd)
DEF_OPERATOR(LogicalLeftShift)
DEF_OPERATOR(LogicalRightShift)
DEF_OPERATOR(LogicalOr)
DEF_OPERATOR(LogicalXor)
DEF_OPERATOR(Maximum)
DEF_OPERATOR(Minimum)
DEF_OPERATOR(Mul)
DEF_OPERATOR(Pow)
DEF_OPERATOR(Sub)
DEF_OPERATOR(Table)

/* elementwise - unary */
DEF_OPERATOR(Abs)
DEF_OPERATOR(BitwiseNot)
DEF_OPERATOR(Ceil)
DEF_OPERATOR(Clz)
DEF_OPERATOR(Cos)
DEF_OPERATOR(Exp)
DEF_OPERATOR(Floor)
DEF_OPERATOR(Log)
DEF_OPERATOR(LogicalNot)
DEF_OPERATOR(Negate)
DEF_OPERATOR(Reciprocal)
DEF_OPERATOR(Rsqrt)
DEF_OPERATOR(Sin)

/* elementwise - ternary */
DEF_OPERATOR(Select)

/* logical */
DEF_OPERATOR(Equal)
DEF_OPERATOR(Greater)
DEF_OPERATOR(GreaterEqual)

/* reduction */
DEF_OPERATOR(ReduceAny)
DEF_OPERATOR(ReduceAll)
DEF_OPERATOR(ReduceMax)
DEF_OPERATOR(ReduceMin)
DEF_OPERATOR(ReduceProd)
DEF_OPERATOR(ReduceSum)

/* memory operation */
DEF_OPERATOR(Concat)
DEF_OPERATOR(Pad)
DEF_OPERATOR(Dim)
DEF_OPERATOR(Reshape)
DEF_OPERATOR(Reverse)
DEF_OPERATOR(Slice)
DEF_OPERATOR(Tile)
DEF_OPERATOR(Transpose)

/* gather/scatter */
DEF_OPERATOR(Gather)
DEF_OPERATOR(Scatter)

/* image */
DEF_OPERATOR(Resize)

/* quantization */
DEF_OPERATOR(Cast)
DEF_OPERATOR(Rescale)

/* data nodes */
DEF_OPERATOR(Const)
DEF_OPERATOR(Identity)

/* custom operations */
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)