MLIR
22.0.0git
|
#include "mlir/Conversion/TosaToLinalg/TosaToLinalg.h"
#include "mlir/Dialect/Arith/IR/Arith.h"
#include "mlir/Dialect/Arith/Utils/Utils.h"
#include "mlir/Dialect/Index/IR/IndexOps.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/Dialect/Math/IR/Math.h"
#include "mlir/Dialect/SCF/IR/SCF.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/Dialect/Utils/StaticValueUtils.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/DialectConversion.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/Sequence.h"
#include <type_traits>
Go to the source code of this file.
Typedefs | |
using | IndexPool = DenseMap< int64_t, Value > |
Definition at line 713 of file TosaToLinalg.cpp.
|
static |
Definition at line 819 of file TosaToLinalg.cpp.
References createIndex(), mlir::OpBuilder::createOrFold(), mlir::AffineMap::get(), mlir::Builder::getAffineConstantExpr(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getContext(), mlir::Builder::getMultiDimIdentityMap(), mlir::tosa::getNParallelLoopsAttrs(), getOrFoldTensorDim(), getTensorDim(), and mlir::Value::getType().
Referenced by broadcastDynamicDimensions().
|
static |
Definition at line 901 of file TosaToLinalg.cpp.
References broadcastDynamicDimension(), and mlir::Value::getType().
Referenced by broadcastDynamicDimensions(), and elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 916 of file TosaToLinalg.cpp.
References broadcastDynamicDimensions().
|
static |
Definition at line 804 of file TosaToLinalg.cpp.
References computeTargetSize(), and mlir::ValueRange::getType().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 763 of file TosaToLinalg.cpp.
References mlir::Builder::getIndexAttr(), getTensorDim(), and mlir::Value::getType().
Referenced by computeTargetShape().
|
static |
Definition at line 718 of file TosaToLinalg.cpp.
References mlir::Builder::getIndexAttr().
Referenced by broadcastDynamicDimension(), and getTensorDim().
|
static |
Definition at line 1050 of file TosaToLinalg.cpp.
References mlir::Builder::getFloatAttr(), mlir::Builder::getIntegerAttr(), mlir::Type::getIntOrFloatBitWidth(), getZero(), and mlir::Type::isInteger().
Referenced by reduceMatchAndRewriteHelper().
|
static |
Definition at line 82 of file TosaToLinalg.cpp.
References mlir::presburger::abs(), clamp(), mlir::tosa::clampFloatHelper(), mlir::tosa::clampIntHelper(), mlir::OpBuilder::create(), mlir::arith::ConstantIntOp::create(), mlir::remark::failed(), mlir::get(), mlir::Operation::getAttr(), mlir::Builder::getContext(), mlir::getElementTypeOrSelf(), mlir::Builder::getFloatAttr(), mlir::Builder::getI32Type(), mlir::Builder::getIntegerAttr(), mlir::Builder::getIntegerType(), mlir::Type::getIntOrFloatBitWidth(), mlir::Operation::getLoc(), mlir::Operation::getOperand(), mlir::Value::getType(), mlir::DenseElementsAttr::getValues(), mlir::Builder::getZeroAttr(), mlir::Type::isInteger(), mlir::Type::isIntOrFloat(), mlir::Type::isUnsignedInteger(), mlir::m_Constant(), mlir::matchPattern(), materializeBinaryNanCheckIfRequired(), max(), min(), mlir::RewriterBase::notifyMatchFailure(), and mlir::presburger::round().
Referenced by emitElementwiseComputation().
|
static |
Definition at line 1102 of file TosaToLinalg.cpp.
References mlir::Operation::getLoc(), and mlir::Type::isInteger().
Referenced by reduceMatchAndRewriteHelper().
|
static |
Definition at line 1023 of file TosaToLinalg.cpp.
References broadcastDynamicDimensions(), computeTargetShape(), emitElementwiseComputation(), getBroadcastableOperands(), mlir::Operation::getLoc(), mlir::Operation::getNumOperands(), mlir::Operation::getNumResults(), mlir::RewriterBase::notifyMatchFailure(), and operandsAndResultsRanked().
|
static |
Definition at line 944 of file TosaToLinalg.cpp.
References mlir::TypeConverter::convertType(), createLinalgBodyCalculationForElementwiseOp(), mlir::OpBuilder::createOrFold(), mlir::detail::enumerate(), mlir::ValueTypeRange< ValueRangeT >::front(), mlir::AffineMap::get(), mlir::Builder::getAffineConstantExpr(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getContext(), mlir::Builder::getMultiDimIdentityMap(), mlir::tosa::getNParallelLoopsAttrs(), mlir::Operation::getNumOperands(), mlir::Operation::getResultTypes(), mlir::Value::getType(), mlir::RewriterBase::notifyMatchFailure(), and mlir::ConversionPatternRewriter::replaceOp().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 1005 of file TosaToLinalg.cpp.
References mlir::Operation::getOperand(), mlir::m_Constant(), and mlir::matchPattern().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 733 of file TosaToLinalg.cpp.
References mlir::Builder::getIndexAttr(), getTensorDim(), and mlir::Value::getType().
Referenced by broadcastDynamicDimension().
|
static |
Definition at line 727 of file TosaToLinalg.cpp.
References createIndex().
Referenced by broadcastDynamicDimension(), computeTargetSize(), and getOrFoldTensorDim().
|
static |
Definition at line 61 of file TosaToLinalg.cpp.
References mlir::getElementTypeOrSelf().
Referenced by createLinalgBodyCalculationForElementwiseOp().
|
static |
Definition at line 744 of file TosaToLinalg.cpp.
References mlir::Operation::getOperands(), and mlir::Operation::getResults().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 1152 of file TosaToLinalg.cpp.
References createInitialValueForReduceOp(), createLinalgBodyCalculationForReduceOp(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getBoolAttr(), mlir::Builder::getFloatAttr(), mlir::RewriterBase::notifyMatchFailure(), and mlir::RewriterBase::replaceOpWithNewOp().