#include "mlir/Dialect/Quant/IR/Quant.h"
#include "mlir/Dialect/Tensor/IR/Tensor.h"
#include "mlir/Dialect/Tosa/IR/TosaOps.h"
#include "mlir/Dialect/Tosa/Utils/ConversionUtils.h"
#include "mlir/Dialect/Tosa/Utils/QuantUtils.h"
#include "mlir/Dialect/Tosa/Utils/ShapeUtils.h"
#include "mlir/IR/BuiltinTypeInterfaces.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/FoldUtils.h"
#include "mlir/Transforms/InliningUtils.h"
#include "mlir/Transforms/RegionUtils.h"
#include "llvm/ADT/APFloat.h"
#include "llvm/ADT/APInt.h"
#include "llvm/ADT/DenseMap.h"
#include "llvm/ADT/TypeSwitch.h"
#include <functional>

Classes
struct	ConcatOptimization

struct	ConsolidateTransposeOptimization

struct	TransposeIsReshape

struct	MaterializePadValue

struct	MaxPool2dIsNoOp

struct	ClampIsNoOp

struct	ClampClampOptimization

struct	ConcatSliceOptimization

Macros
#define	REDUCE_FOLDER(OP)

Functions
template<typename IntFolder , typename FloatFolder >
DenseElementsAttr	binaryFolder (DenseElementsAttr lhs, DenseElementsAttr rhs, RankedTensorType returnTy)

static bool	isSplatZero (Type elemType, DenseElementsAttr val)

static bool	isSplatOne (Type elemType, DenseElementsAttr val, int64_t shift)

Macro Definition Documentation

◆ REDUCE_FOLDER

#define REDUCE_FOLDER ( OP )

Value:

  OpFoldResult OP::fold(FoldAdaptor adaptor) {                                 \
    ShapedType inputTy = llvm::cast<ShapedType>(getInput().getType());         \
    if (!inputTy.hasRank())                                                    \
      return {};                                                               \
    if (inputTy != getType())                                                  \
      return {};                                                               \
    if (inputTy.getRank() == 0 || inputTy.getDimSize(getAxis()) == 1)          \
      return getInput();                                                       \
    return {};                                                                 \
  }

Definition at line 811 of file TosaCanonicalizations.cpp.

Function Documentation

◆ binaryFolder()

template<typename IntFolder , typename FloatFolder >

DenseElementsAttr binaryFolder	(	DenseElementsAttr	lhs,
		DenseElementsAttr	rhs,
		RankedTensorType	returnTy
	)

Definition at line 441 of file TosaCanonicalizations.cpp.

References mlir::DenseElementsAttr::getElementType(), mlir::DenseElementsAttr::getSplatValue(), mlir::DenseElementsAttr::getType(), and mlir::DenseElementsAttr::isSplat().

◆ isSplatOne()

static bool isSplatOne	(	Type	elemType,
		DenseElementsAttr	val,
		int64_t	shift
	)

static

Definition at line 475 of file TosaCanonicalizations.cpp.

References mlir::DenseElementsAttr::getSplatValue(), and mlir::DenseElementsAttr::isSplat().

◆ isSplatZero()

static bool isSplatZero	(	Type	elemType,
		DenseElementsAttr	val
	)

static

Definition at line 467 of file TosaCanonicalizations.cpp.