MLIR
18.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/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/Tensor/Utils/Utils.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/ImplicitLocOpBuilder.h"
#include "mlir/IR/Matchers.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/DialectConversion.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/Sequence.h"
#include <numeric>
Go to the source code of this file.
Typedefs | |
using | IndexPool = DenseMap< int64_t, Value > |
Definition at line 567 of file TosaToLinalg.cpp.
|
static |
Definition at line 673 of file TosaToLinalg.cpp.
References mlir::Type::cast(), mlir::OpBuilder::create(), createIndex(), mlir::OpBuilder::createOrFold(), mlir::AffineMap::get(), mlir::Builder::getAffineConstantExpr(), mlir::Builder::getAffineDimExpr(), mlir::Builder::getContext(), mlir::Builder::getMultiDimIdentityMap(), mlir::tosa::getNParallelLoopsAttrs(), getOrFoldTensorDim(), mlir::Operation::getResult(), getTensorDim(), and mlir::Value::getType().
Referenced by broadcastDynamicDimensions().
|
static |
Definition at line 752 of file TosaToLinalg.cpp.
References broadcastDynamicDimension(), mlir::Type::cast(), and mlir::Value::getType().
Referenced by broadcastDynamicDimensions(), and elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 767 of file TosaToLinalg.cpp.
References broadcastDynamicDimensions().
|
static |
Definition at line 658 of file TosaToLinalg.cpp.
References computeTargetSize(), and mlir::ValueRange::getType().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 617 of file TosaToLinalg.cpp.
References mlir::OpBuilder::create(), mlir::Builder::getIndexAttr(), getTensorDim(), and mlir::Value::getType().
Referenced by computeTargetShape().
|
static |
Definition at line 41 of file TosaToLinalg.cpp.
|
static |
Definition at line 572 of file TosaToLinalg.cpp.
References mlir::OpBuilder::create(), and mlir::Builder::getIndexAttr().
Referenced by broadcastDynamicDimension(), and getTensorDim().
|
static |
Definition at line 860 of file TosaToLinalg.cpp.
|
static |
Definition at line 50 of file TosaToLinalg.cpp.
Referenced by emitElementwiseComputation().
|
static |
Definition at line 912 of file TosaToLinalg.cpp.
|
static |
Definition at line 835 of file TosaToLinalg.cpp.
References broadcastDynamicDimensions(), computeTargetShape(), emitElementwiseComputation(), expandInputRanks(), mlir::Operation::getLoc(), mlir::Operation::getNumOperands(), mlir::Operation::getNumResults(), mlir::RewriterBase::notifyMatchFailure(), and operandsAndResultsRanked().
|
static |
Definition at line 783 of file TosaToLinalg.cpp.
References mlir::Type::cast(), mlir::OpBuilder::create(), 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(), mlir::RewriterBase::replaceOp(), and mlir::success().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 558 of file TosaToLinalg.cpp.
References mlir::Type::cast(), mlir::ValueTypeRange< ValueRangeT >::front(), mlir::Operation::getOperands(), and mlir::Operation::getResultTypes().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 525 of file TosaToLinalg.cpp.
References mlir::OpBuilder::create(), mlir::get(), and mlir::Value::getType().
|
static |
Definition at line 587 of file TosaToLinalg.cpp.
References mlir::Builder::getIndexAttr(), getTensorDim(), and mlir::Value::getType().
Referenced by broadcastDynamicDimension().
|
static |
Definition at line 581 of file TosaToLinalg.cpp.
References mlir::OpBuilder::create(), and createIndex().
Referenced by broadcastDynamicDimension(), computeTargetSize(), and getOrFoldTensorDim().
|
static |
Definition at line 598 of file TosaToLinalg.cpp.
References mlir::Operation::getOperands(), and mlir::Operation::getResults().
Referenced by elementwiseMatchAndRewriteHelper().
|
static |
Definition at line 965 of file TosaToLinalg.cpp.