#include "TosaToLinalgUtils.h"
#include "mlir/Conversion/TosaToLinalg/TosaToLinalg.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Math/IR/Math.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/Utils/ReshapeOpsUtils.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/DialectConversion.h"
#include "llvm/ADT/SmallVectorExtras.h"
#include "llvm/ADT/TypeSwitch.h"
#include <optional>
#include <type_traits>

Functions
static std::optional< linalg::ElementwiseKind >	getElementwiseKind (Operation *op, Type elementTy)
static mlir::Value	applyPad (Location loc, Value input, ArrayRef< int64_t > pad, TypedAttr padAttr, OpBuilder &rewriter)
static mlir::Value	linalgIntBroadcastExtSIAdd (PatternRewriter &rewriter, Location loc, Value bias, Value conv, Value result, ArrayRef< AffineMap > indexingMaps)
static AffineMap	getBroadcastingMap (PatternRewriter &rewriter, Value source, Value result)
static mlir::Value	linalgBroadcastAndMaybeExt (PatternRewriter &rewriter, Location loc, Value source, Value result)
static mlir::Value	reifyConstantDim (int64_t attr, ImplicitLocOpBuilder &builder)
static mlir::Value	getConvOrPoolOutputDim (Location loc, Value inputDim, int64_t padBeforeAttr, int64_t padAfterAttr, Value kernelDim, int64_t strideAttr, int64_t dilationAttr, OpBuilder &rewriter)
static SmallVector< Value >	inferDynamicDimsForConv (Location loc, Value input, Value weight, ShapedType resultTy, ArrayRef< int64_t > padAttr, ArrayRef< int64_t > strideAttr, ArrayRef< int64_t > dilationAttr, ArrayRef< int64_t > inputSizeDims, ArrayRef< int64_t > kernelSizeDims, OpBuilder &rewriter)
static void	createDepthwiseConvCollapseMap (int64_t outputRank, SmallVector< ReassociationExprs, 4 > &reassociationMap, OpBuilder &rewriter)

Function Documentation

◆ applyPad()

mlir::Value applyPad	(	Location	loc,
		Value	input,
		ArrayRef< int64_t >	pad,
		TypedAttr	padAttr,
		OpBuilder &	rewriter )

static

Definition at line 160 of file TosaToLinalgNamed.cpp.

References mlir::Builder::getIndexAttr(), and mlir::Value::getType().

◆ createDepthwiseConvCollapseMap()

void createDepthwiseConvCollapseMap	(	int64_t	outputRank,
		SmallVector< ReassociationExprs, 4 > &	reassociationMap,
		OpBuilder &	rewriter )

static

Definition at line 357 of file TosaToLinalgNamed.cpp.

References mlir::Builder::getAffineDimExpr().

◆ getBroadcastingMap()

AffineMap getBroadcastingMap	(	PatternRewriter &	rewriter,
		Value	source,
		Value	result )

static

Definition at line 217 of file TosaToLinalgNamed.cpp.

References mlir::AffineMap::get(), mlir::Builder::getAffineConstantExpr(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getContext(), mlir::Value::getType(), and result.

Referenced by linalgBroadcastAndMaybeExt().

◆ getConvOrPoolOutputDim()

mlir::Value getConvOrPoolOutputDim	(	Location	loc,
		Value	inputDim,
		int64_t	padBeforeAttr,
		int64_t	padAfterAttr,
		Value	kernelDim,
		int64_t	strideAttr,
		int64_t	dilationAttr,
		OpBuilder &	rewriter )

static

Definition at line 290 of file TosaToLinalgNamed.cpp.

References mlir::Value::getType(), and reifyConstantDim().

Referenced by inferDynamicDimsForConv().

◆ getElementwiseKind()

std::optional< linalg::ElementwiseKind > getElementwiseKind	(	Operation *	op,
		Type	elementTy )

static

Definition at line 42 of file TosaToLinalgNamed.cpp.

References mlir::DenseElementsAttr::getValues(), mlir::Type::isInteger(), mlir::Type::isSignlessInteger(), mlir::m_Constant(), and mlir::matchPattern().

Referenced by mlir::tosa::isConvertibleToLinalgElementwise().

◆ inferDynamicDimsForConv()

SmallVector< Value > inferDynamicDimsForConv	(	Location	loc,
		Value	input,
		Value	weight,
		ShapedType	resultTy,
		ArrayRef< int64_t >	padAttr,
		ArrayRef< int64_t >	strideAttr,
		ArrayRef< int64_t >	dilationAttr,
		ArrayRef< int64_t >	inputSizeDims,
		ArrayRef< int64_t >	kernelSizeDims,
		OpBuilder &	rewriter )

static

Definition at line 316 of file TosaToLinalgNamed.cpp.

References mlir::tosa::condenseValues(), getConvOrPoolOutputDim(), and mlir::Value::getType().

◆ linalgBroadcastAndMaybeExt()

mlir::Value linalgBroadcastAndMaybeExt	(	PatternRewriter &	rewriter,
		Location	loc,
		Value	source,
		Value	result )

static

Definition at line 248 of file TosaToLinalgNamed.cpp.

References getBroadcastingMap(), mlir::Builder::getMultiDimIdentityMap(), mlir::tosa::getNParallelLoopsAttrs(), mlir::Value::getType(), result, and ValueRange.

◆ linalgIntBroadcastExtSIAdd()

mlir::Value linalgIntBroadcastExtSIAdd	(	PatternRewriter &	rewriter,
		Location	loc,
		Value	bias,
		Value	conv,
		Value	result,
		ArrayRef< AffineMap >	indexingMaps )