Namespaces | |
| namespace | detail |
| Implemented by ops that correspond to the Tosa specification. | |
| namespace | impl |
Classes | |
| class | TargetEnv |
| This class represents the capability enabled in the target implementation such as profile, extension, and level. More... | |
| struct | TosaAttachTargetOptions |
| struct | TosaInferShapesPassOptions |
| struct | TosaInputShapeOptions |
| struct | TosaLayerwiseConstantFoldPassOptions |
| struct | TosaLevel |
| struct | TosaNarrowF64ToF32PassOptions |
| struct | TosaNarrowI64ToI32PassOptions |
| class | TosaOp |
| class | TosaSpecificationVersion |
| A thin wrapper around the SpecificationVersion enum to represent and provide utilities around the TOSA specification version. More... | |
| struct | TosaValidationOptions |
| struct | ValueKnowledge |
| Statically known information for a particular Value. More... | |
Variables | |
| static constexpr TosaLevel | TOSA_LEVEL_EIGHTK |
| static constexpr TosaLevel | TOSA_LEVEL_NONE |
| constexpr int64_t | kInferableDimSize = -1 |
| Represents a dimension in the shape of a tensor that can be inferred based on the other provided dimensions. | |
Function Documentation
◆ addTosaToLinalgPasses()
| void mlir::tosa::addTosaToLinalgPasses | ( | OpPassManager & | pm, |
| const TosaToLinalgOptions & | options, | ||
| const TosaToLinalgNamedOptions & | tosaToLinalgNamedOptions = TosaToLinalgNamedOptions(), | ||
| std::optional< tosa::TosaValidationOptions > | validationOptions = tosa::TosaValidationOptions{false, false}, | ||
| std::optional< TosaAttachTargetOptions > | attachTargetOptions = std::nullopt ) |
Populates passes to convert from TOSA to Linalg.
At the end of the pass, the function will only contain linalg ops or standard ops if the pipeline succeeds. The option to disable decompositions is available for benchmarking performance improvements from the canonicalizations.
Definition at line 81 of file TosaToLinalgPass.cpp.
References mlir::OpPassManager::addNestedPass(), mlir::OpPassManager::addPass(), mlir::createCanonicalizerPass(), createTosaAttachTarget(), createTosaInferShapesPass(), createTosaLayerwiseConstantFoldPass(), createTosaMakeBroadcastablePass(), createTosaOptionalDecompositionsPass(), createTosaToLinalg(), createTosaToLinalgNamed(), createTosaValidation(), and options.
Referenced by registerTosaToLinalgPipelines().
◆ addTosaToSCFPasses()
| void mlir::tosa::addTosaToSCFPasses | ( | OpPassManager & | pm | ) |
Populates passes to convert from TOSA to SCF.
Definition at line 49 of file TosaToSCFPass.cpp.
References mlir::OpPassManager::addNestedPass(), and mlir::createTosaToSCFPass().
◆ applyTOSAPermutation()
| SmallVector< T > mlir::tosa::applyTOSAPermutation | ( | ArrayRef< T > | input, |
| ArrayRef< int32_t > | perms ) |
Definition at line 223 of file ConversionUtils.h.
◆ buildConvOpQuantizationAttr()
| ConvOpQuantizationAttr mlir::tosa::buildConvOpQuantizationAttr | ( | OpBuilder & | builder, |
| Value | input, | ||
| Value | weight ) |
Method to build ConvOpQuantizationAttr, called from ConvOpQuantInfoBuilder/TransConvOpQuantInfoBuilder: input_zp: input zeropoint weight_zp: weight zeropoint.
Definition at line 198 of file QuantUtils.cpp.
References mlir::Builder::getAttr(), and getConvZeroPoints().
Referenced by buildConvOpWithQuantInfo(), and buildTransConvOpWithQuantInfo().
◆ buildConvOpResultTypeInfo()
| Type mlir::tosa::buildConvOpResultTypeInfo | ( | OpBuilder & | builder, |
| Type | outputType, | ||
| Value | input, | ||
| Value | weight ) |
construct ConvOp output type with correct bitwidth based on input/weight width.
Builds output type for a quantized ConvOp with the right bitwidth.
This is called by the builder when dealing with quantized content.
Definition at line 289 of file QuantUtils.cpp.
References GET_QTYPE, mlir::Builder::getI32Type(), mlir::Builder::getIntegerType(), and mlir::Value::getType().
Referenced by buildConvOpWithQuantInfo(), and buildTransConvOpWithQuantInfo().
◆ buildMatMulOpQuantizationAttr()
| MatMulOpQuantizationAttr mlir::tosa::buildMatMulOpQuantizationAttr | ( | OpBuilder & | builder, |
| Value | a, | ||
| Value | b ) |
Builds MatMulOpQuantizationAttr, called from MatMulOpQuantInfoBuilder: aZp: input a zeropoint bZp: input b zeropoint.
Definition at line 214 of file QuantUtils.cpp.
References b, GET_UQTYPE, mlir::Builder::getAttr(), and mlir::Value::getType().
Referenced by buildMatMulOpWithQuantInfo().
◆ buildPadOpQuantizationAttr()
Builds PadOpQuantizationAttr, called from PadOpQuantInfoBuilder: inputZp: input zeropoint.
Definition at line 269 of file QuantUtils.cpp.
References GET_UQTYPE, mlir::Builder::getAttr(), and mlir::Value::getType().
Referenced by buildPadOpWithQuantInfo().
◆ buildQTypeAttrFromMinMax()
| TypeAttr mlir::tosa::buildQTypeAttrFromMinMax | ( | OpBuilder | builder, |
| Type | inputDType, | ||
| Attribute | minAttr, | ||
| Attribute | maxAttr, | ||
| IntegerAttr | quantBits, | ||
| int | filterQuantDim, | ||
| bool | isSigned, | ||
| BoolAttr | narrowRange ) |
Builds Tosa quantization attributes from min/max values.
Definition at line 389 of file QuantUtils.cpp.
References buildQTypeFromMinMax().
◆ buildQTypeFromMinMax()
| Type mlir::tosa::buildQTypeFromMinMax | ( | OpBuilder | builder, |
| Type | inputDType, | ||
| Attribute | minAttr, | ||
| Attribute | maxAttr, | ||
| IntegerAttr | quantBits, | ||
| int | filterQuantDim, | ||
| bool | isSigned, | ||
| BoolAttr | narrowRange ) |
Builds Tosa quantization attributes from min/max values.
Definition at line 321 of file QuantUtils.cpp.
References mlir::quant::fakeQuantAttrsToType(), mlir::quant::ExpressedToQuantizedConverter::forInputType(), mlir::Builder::getUnknownLoc(), mlir::BoolAttr::getValue(), max(), and min().
Referenced by buildQTypeAttrFromMinMax().
◆ buildUnaryOpQuantizationAttr()
| UnaryOpQuantizationAttr mlir::tosa::buildUnaryOpQuantizationAttr | ( | OpBuilder & | builder, |
| Value | input, | ||
| Type | outputRawType ) |
Builds UnaryOpQuantizationAttr UnaryOpQuantInfoBuilder: inputZp: input zeropoint outputZp: output zeropoint.
Definition at line 243 of file QuantUtils.cpp.
References GET_UQTYPE, mlir::Builder::getAttr(), and mlir::Value::getType().
Referenced by buildAvgPool2dOpWithQuantInfo(), and buildNegateOpWithQuantInfo().
◆ checkHasDynamicBatchDims()
| std::optional< SmallVector< Value > > mlir::tosa::checkHasDynamicBatchDims | ( | PatternRewriter & | rewriter, |
| Op | op, | ||
| ArrayRef< Value > | params ) |
Definition at line 52 of file ConversionUtils.h.
References mlir::OpState::getLoc(), and mlir::RewriterBase::notifyMatchFailure().
◆ clampFloatHelper()
| Value mlir::tosa::clampFloatHelper | ( | Location | loc, |
| Value | arg, | ||
| Value | min, | ||
| Value | max, | ||
| OpBuilder & | rewriter ) |
Definition at line 35 of file ConversionUtils.cpp.
Referenced by createLinalgBodyCalculationForElementwiseOp().
◆ clampIntHelper()
| Value mlir::tosa::clampIntHelper | ( | Location | loc, |
| Value | arg, | ||
| Value | min, | ||
| Value | max, | ||
| OpBuilder & | rewriter, | ||
| bool | isUnsigned ) |
Definition at line 41 of file ConversionUtils.cpp.
Referenced by createLinalgBodyCalculationForElementwiseOp().
◆ computeMultiplierAndShift()
| bool mlir::tosa::computeMultiplierAndShift | ( | double | scale, |
| int32_t & | multiplier, | ||
| int32_t & | shift, | ||
| int32_t | scaleWidth ) |
From a scale value, computes multiplier and shift values for 16 or 32-bit scale widths.
Generates a quantized multiplier/shift from double.
Definition at line 95 of file QuantUtils.cpp.
References computeMultiplierAndShiftTosaScale16(), and computeMultiplierAndShiftTosaScale32().
◆ condenseValues()
| SmallVector< Value > mlir::tosa::condenseValues | ( | const SmallVector< Value > & | values | ) |
Definition at line 27 of file ConversionUtils.cpp.
Referenced by inferDynamicDimsForConv().
◆ convertFromIntAttr()
| SmallVector< int64_t > mlir::tosa::convertFromIntAttr | ( | const DenseElementsAttr & | attr, |
| const int | rank ) |
Definition at line 208 of file ConversionUtils.cpp.
References mlir::DenseElementsAttr::getSplatValue(), and mlir::DenseElementsAttr::isSplat().
◆ convertFromMlirShape()
| SmallVector< int64_t > mlir::tosa::convertFromMlirShape | ( | ArrayRef< int64_t > | shape | ) |
Definition at line 183 of file ConversionUtils.cpp.
Referenced by buildVariableOp(), and getTosaConstShape().
◆ CreateOpAndInferShape() [1/2]
| TosaOp mlir::tosa::CreateOpAndInferShape | ( | ImplicitLocOpBuilder & | builder, |
| Type | resultTy, | ||
| Args &&... | args ) |
Definition at line 147 of file ConversionUtils.h.
References EqualizeRanks().
Referenced by CreateOpAndInferShape().
◆ CreateOpAndInferShape() [2/2]
| TosaOp mlir::tosa::CreateOpAndInferShape | ( | PatternRewriter & | rewriter, |
| Location | loc, | ||
| Type | resultTy, | ||
| Args &&... | args ) |
Definition at line 214 of file ConversionUtils.h.
References CreateOpAndInferShape().
◆ createPadConstTensor()
| Value mlir::tosa::createPadConstTensor | ( | OpBuilder & | builder, |
| Location | loc, | ||
| Value | src, | ||
| int32_t | val = 0 ) |
Definition at line 605 of file TosaOps.cpp.
References mlir::DenseElementsAttr::get(), mlir::getElementTypeOrSelf(), mlir::Builder::getFloatAttr(), mlir::Builder::getIntegerAttr(), and getStorageElementTypeOrSelf().
Referenced by buildPadOpWithQuantInfo().
◆ createTosaArithConstantToTosaConstPass()
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaArithConstantToTosaConstPass | ( | ) |
We declare an explicit private instantiation because Pass classes should only be visible by the current library.
Definition at line 90 of file TosaArithConstantToConst.cpp.
◆ createTosaAttachTarget() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaAttachTarget | ( | ) |
Definition at line 201 of file TosaAttachTarget.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaAttachTarget() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaAttachTarget | ( | TosaAttachTargetOptions | options | ) |
Definition at line 205 of file TosaAttachTarget.cpp.
◆ createTosaConvertIntegerTypeToSignless()
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaConvertIntegerTypeToSignless | ( | ) |
We declare an explicit private instantiation because Pass classes should only be visible by the current library.
Definition at line 280 of file TosaConvertIntegerTypeToSignless.cpp.
◆ createTosaInferShapesPass() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaInferShapesPass | ( | ) |
Definition at line 376 of file TosaInferShapes.cpp.
References result.
Referenced by addTosaToLinalgPasses().
◆ createTosaInferShapesPass() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaInferShapesPass | ( | TosaInferShapesPassOptions | options | ) |
Definition at line 380 of file TosaInferShapes.cpp.
◆ createTosaInputShapePass()
| std::unique_ptr< Pass > mlir::tosa::createTosaInputShapePass | ( | std::vector< std::string > | args = {} | ) |
Definition at line 180 of file TosaInputShape.cpp.
◆ createTosaLayerwiseConstantFoldPass() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaLayerwiseConstantFoldPass | ( | ) |
Definition at line 540 of file TosaLayerwiseConstantFoldPass.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaLayerwiseConstantFoldPass() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaLayerwiseConstantFoldPass | ( | TosaLayerwiseConstantFoldPassOptions | options | ) |
Definition at line 544 of file TosaLayerwiseConstantFoldPass.cpp.
◆ createTosaMakeBroadcastablePass()
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaMakeBroadcastablePass | ( | ) |
We declare an explicit private instantiation because Pass classes should only be visible by the current library.
Definition at line 619 of file TosaMakeBroadcastable.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaNarrowF64ToF32Pass() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaNarrowF64ToF32Pass | ( | ) |
Definition at line 714 of file TosaNarrowTypes.cpp.
◆ createTosaNarrowF64ToF32Pass() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaNarrowF64ToF32Pass | ( | TosaNarrowF64ToF32PassOptions | options | ) |
Definition at line 718 of file TosaNarrowTypes.cpp.
◆ createTosaNarrowI64ToI32Pass() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaNarrowI64ToI32Pass | ( | ) |
Definition at line 813 of file TosaNarrowTypes.cpp.
◆ createTosaNarrowI64ToI32Pass() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaNarrowI64ToI32Pass | ( | TosaNarrowI64ToI32PassOptions | options | ) |
Definition at line 817 of file TosaNarrowTypes.cpp.
◆ createTosaOptionalDecompositionsPass()
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaOptionalDecompositionsPass | ( | ) |
We declare an explicit private instantiation because Pass classes should only be visible by the current library.
Definition at line 892 of file TosaOptionalDecompositions.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaReduceTransposes()
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaReduceTransposes | ( | ) |
We declare an explicit private instantiation because Pass classes should only be visible by the current library.
Definition at line 967 of file TosaReduceTransposes.cpp.
◆ createTosaTestQuantUtilAPIPass()
| std::unique_ptr< Pass > mlir::tosa::createTosaTestQuantUtilAPIPass | ( | ) |
◆ createTosaToLinalg()
| std::unique_ptr< Pass > mlir::tosa::createTosaToLinalg | ( | ) |
Definition at line 77 of file TosaToLinalgPass.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaToLinalgNamed()
| std::unique_ptr< Pass > mlir::tosa::createTosaToLinalgNamed | ( | const TosaToLinalgNamedOptions & | options = TosaToLinalgNamedOptions() | ) |
Definition at line 79 of file TosaToLinalgNamedPass.cpp.
References options.
Referenced by addTosaToLinalgPasses().
◆ createTosaValidation() [1/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaValidation | ( | ) |
Definition at line 1061 of file TosaValidation.cpp.
Referenced by addTosaToLinalgPasses().
◆ createTosaValidation() [2/2]
| std::unique_ptr<::mlir::Pass > mlir::tosa::createTosaValidation | ( | TosaValidationOptions | options | ) |
Definition at line 1065 of file TosaValidation.cpp.
References mlir::Operation::emitOpError(), and idivCheck().
◆ createZeroPointTensor()
| std::optional< Value > mlir::tosa::createZeroPointTensor | ( | OpBuilder & | builder, |
| Location | loc, | ||
| Type | srcElemType, | ||
| int64_t | zp = 0 ) |
Definition at line 4920 of file TosaOps.cpp.
References mlir::DenseElementsAttr::get(), mlir::Builder::getFloatAttr(), mlir::Builder::getIntegerAttr(), and getStorageElementTypeOrSelf().
Referenced by buildAvgPool2dOpWithQuantInfo(), buildNegateOpWithQuantInfo(), and createZPsAsConst().
◆ createZPsAsConst()
| std::pair< Value, Value > mlir::tosa::createZPsAsConst | ( | OpBuilder & | builder, |
| Value | input, | ||
| Value | weight ) |
Definition at line 162 of file QuantUtils.cpp.
References createZeroPointTensor(), getConvZeroPoints(), mlir::getElementTypeOrSelf(), mlir::Value::getLoc(), and mlir::Value::getType().
Referenced by buildConvOpWithQuantInfo(), buildMatMulOpWithQuantInfo(), and buildTransConvOpWithQuantInfo().
◆ EqualizeRanks() [1/2]
| LogicalResult mlir::tosa::EqualizeRanks | ( | ImplicitLocOpBuilder & | builder, |
| Value & | input1, | ||
| Value & | input2 ) |
Definition at line 114 of file ConversionUtils.cpp.
References getTosaConstShape(), mlir::Value::getType(), and success().
◆ EqualizeRanks() [2/2]
| LogicalResult mlir::tosa::EqualizeRanks | ( | PatternRewriter & | rewriter, |
| Location | loc, | ||
| Value & | input1, | ||
| Value & | input2 ) |
Common code to create the reshape op where necessary to make the rank of two values equal.
input1 and input2 will be updated when the rank has changed. The caller is expected to use these to rewrite the original operator with the RESHAPE now in the graph.
Definition at line 108 of file ConversionUtils.cpp.
References EqualizeRanks().
Referenced by CreateOpAndInferShape(), and EqualizeRanks().
◆ getBitWidth()
Definition at line 620 of file TosaOps.cpp.
References mlir::Type::getIntOrFloatBitWidth().
◆ getConstShapeValues()
| bool mlir::tosa::getConstShapeValues | ( | Operation * | op, |
| llvm::SmallVector< int64_t > & | result_shape ) |
Definition at line 188 of file ConversionUtils.cpp.
References mlir::DenseElementsAttr::getValues(), and mlir::DenseElementsAttr::size().
Referenced by ConvInferShapeAdaptor< Conv2DBlockScaledOp::Adaptor >::getSpatialParameters().
◆ getConstTensorInt()
| Value mlir::tosa::getConstTensorInt | ( | OpBuilder & | builder, |
| Location | loc, | ||
| ArrayRef< IntType > | vec ) |
Definition at line 36 of file QuantUtils.h.
References mlir::DenseElementsAttr::get(), and mlir::Builder::getIntegerType().
◆ getCooperativeProfiles()
| SmallVector< Profile, 2 > mlir::tosa::getCooperativeProfiles | ( | Extension | ext | ) |
Definition at line 55 of file TargetEnv.cpp.
Referenced by mlir::tosa::TargetEnv::verifyTargetInformation().
◆ getDefaultTargetEnv()
| TargetEnvAttr mlir::tosa::getDefaultTargetEnv | ( | MLIRContext * | context | ) |
Definition at line 169 of file TargetEnv.cpp.
Referenced by lookupTargetEnvOrDefault().
◆ getMinVersion() [1/3]
| TosaSpecificationVersion mlir::tosa::getMinVersion | ( | const Extension & | extension | ) |
Definition at line 30 of file TargetEnv.cpp.
◆ getMinVersion() [2/3]
| TosaSpecificationVersion mlir::tosa::getMinVersion | ( | const Level & | level | ) |
Definition at line 81 of file TargetEnv.cpp.
◆ getMinVersion() [3/3]
| TosaSpecificationVersion mlir::tosa::getMinVersion | ( | const Profile & | profile | ) |
Definition at line 19 of file TargetEnv.cpp.
Referenced by mlir::tosa::TargetEnv::verifyTargetInformation().
◆ getNParallelLoopsAttrs()
| SmallVector< utils::IteratorType > mlir::tosa::getNParallelLoopsAttrs | ( | unsigned | nParallelLoops | ) |
Definition at line 21 of file ConversionUtils.cpp.
Referenced by broadcastDynamicDimension(), emitElementwiseComputation(), linalgBroadcastAndMaybeExt(), linalgIntBroadcastExtSIAdd(), and reduceMatchAndRewriteHelper().
◆ getStorageElementTypeFromQuantized()
| Type mlir::tosa::getStorageElementTypeFromQuantized | ( | quant::QuantizedType | quantizedType | ) |
Definition at line 399 of file QuantUtils.cpp.
References mlir::quant::QuantizedType::getStorageType(), and mlir::quant::QuantizedType::isSigned().
Referenced by getStorageElementTypeOrSelf(), ClampClampOptimization::matchAndRewrite(), verifyConvOp(), and verifyConvOpModes().
◆ getTosaConstShape() [1/2]
| Value mlir::tosa::getTosaConstShape | ( | ImplicitLocOpBuilder & | builder, |
| llvm::ArrayRef< int64_t > | shape ) |
Definition at line 169 of file ConversionUtils.cpp.
References convertFromMlirShape(), mlir::Builder::getContext(), mlir::Builder::getIndexTensorAttr(), and mlir::Operation::getResult().
Referenced by EqualizeRanks(), getTosaConstShape(), ConcatSliceOptimization::matchAndRewrite(), PadSliceOptimization::matchAndRewrite(), SliceDynamicSizeCanonicalization::matchAndRewrite(), and TransposeIsReshape::matchAndRewrite().
◆ getTosaConstShape() [2/2]
| Value mlir::tosa::getTosaConstShape | ( | PatternRewriter & | rewriter, |
| Location | loc, | ||
| llvm::ArrayRef< int64_t > | shape ) |
Definition at line 177 of file ConversionUtils.cpp.
References getTosaConstShape().
◆ getVariableType()
| RankedTensorType mlir::tosa::getVariableType | ( | VariableOp | variableOp | ) |
Referenced by verifyVariableOpErrorIf().
◆ hasUniqueConstantScatterIndices()
| bool mlir::tosa::hasUniqueConstantScatterIndices | ( | ShapedType | indicesType, |
| DenseIntElementsAttr | indicesAttr ) |
Definition at line 224 of file ConversionUtils.cpp.
◆ isa_tosa_shape_type()
| bool mlir::tosa::isa_tosa_shape_type | ( | mlir::Type | t | ) |
Definition at line 4944 of file TosaOps.cpp.
◆ lookupTargetEnv()
| TargetEnvAttr mlir::tosa::lookupTargetEnv | ( | Operation * | op | ) |
Definition at line 154 of file TargetEnv.cpp.
References mlir::Operation::getAttrOfType(), mlir::SymbolTable::getNearestSymbolTable(), and mlir::Operation::getParentOp().
Referenced by lookupTargetEnvOrDefault().
◆ lookupTargetEnvOrDefault()
| TargetEnvAttr mlir::tosa::lookupTargetEnvOrDefault | ( | Operation * | op | ) |
Queries the target environment recursively from enclosing symbol table ops containing the given op or returns the default target environment as returned by getDefaultTargetEnv() if not provided.
Definition at line 174 of file TargetEnv.cpp.
References mlir::Operation::getContext(), getDefaultTargetEnv(), and lookupTargetEnv().
◆ parseVariableOpTypeOrInitialValue()
| ParseResult mlir::tosa::parseVariableOpTypeOrInitialValue | ( | OpAsmParser & | parser, |
| DenseElementsAttr & | varShapeAttr, | ||
| TypeAttr & | typeAttr, | ||
| Attribute & | initialValueAttr ) |
Definition at line 227 of file TosaOps.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseAttribute(), mlir::AsmParser::parseColonType(), and mlir::AsmParser::parseOptionalEqual().
◆ populateTosaConstantReduction()
| void mlir::tosa::populateTosaConstantReduction | ( | MLIRContext * | ctx, |
| RewritePatternSet & | patterns, | ||
| bool | aggressiveReduceConstant ) |
Definition at line 427 of file TosaFolders.cpp.
References mlir::RewritePatternSet::add().
◆ populateTosaDecomposeDepthwise()
| void mlir::tosa::populateTosaDecomposeDepthwise | ( | MLIRContext * | ctx, |
| RewritePatternSet & | patterns ) |
Definition at line 198 of file TosaDecomposeDepthwise.cpp.
References mlir::RewritePatternSet::add().
◆ populateTosaDecomposeTransposeConv()
| void mlir::tosa::populateTosaDecomposeTransposeConv | ( | MLIRContext * | ctx, |
| RewritePatternSet & | patterns ) |
Definition at line 345 of file TosaDecomposeTransposeConv.cpp.
References mlir::RewritePatternSet::add().
◆ populateTosaFoldConstantReciprocalPatterns()
| void mlir::tosa::populateTosaFoldConstantReciprocalPatterns | ( | MLIRContext * | ctx, |
| RewritePatternSet & | patterns ) |
Definition at line 449 of file TosaFolders.cpp.
References mlir::RewritePatternSet::add().
◆ populateTosaFoldConstantTransposePatterns()
| void mlir::tosa::populateTosaFoldConstantTransposePatterns | ( | MLIRContext * | ctx, |
| RewritePatternSet & | patterns ) |
Definition at line 444 of file TosaFolders.cpp.
References mlir::RewritePatternSet::add().
◆ populateTosaRescaleToArithConversionPatterns()
| void mlir::tosa::populateTosaRescaleToArithConversionPatterns | ( | RewritePatternSet * | patterns, |
| bool | include32Bit = false ) |
Definition at line 258 of file TosaToArith.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaToArithConversionPatterns()
| void mlir::tosa::populateTosaToArithConversionPatterns | ( | RewritePatternSet * | patterns | ) |
Definition at line 253 of file TosaToArith.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaToLinalgConversionPatterns()
| void mlir::tosa::populateTosaToLinalgConversionPatterns | ( | const TypeConverter & | converter, |
| RewritePatternSet * | patterns ) |
Populates conversion passes from TOSA dialect to Linalg dialect.
Definition at line 3032 of file TosaToLinalg.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaToLinalgNamedConversionPatterns()
| void mlir::tosa::populateTosaToLinalgNamedConversionPatterns | ( | const TypeConverter & | converter, |
| RewritePatternSet * | patterns, | ||
| const TosaToLinalgNamedOptions & | options ) |
Populates conversion passes from TOSA dialect to Linalg named operations.
Definition at line 1118 of file TosaToLinalgNamed.cpp.
References mlir::RewritePatternSet::add(), mlir::RewritePatternSet::getContext(), and options.
◆ populateTosaToMLProgramConversionPatterns()
| void mlir::tosa::populateTosaToMLProgramConversionPatterns | ( | RewritePatternSet * | patterns | ) |
Definition at line 72 of file TosaToMLProgram.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaToSCFConversionPatterns()
| void mlir::tosa::populateTosaToSCFConversionPatterns | ( | RewritePatternSet * | patterns | ) |
Definition at line 181 of file TosaToSCF.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaToTensorConversionPatterns()
| void mlir::tosa::populateTosaToTensorConversionPatterns | ( | const TypeConverter & | converter, |
| RewritePatternSet * | patterns ) |
Definition at line 459 of file TosaToTensor.cpp.
References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().
◆ populateTosaTypeConversion()
| void mlir::tosa::populateTosaTypeConversion | ( | TypeConverter & | converter | ) |
Definition at line 20 of file TosaTypeConverters.cpp.
References mlir::Type::getContext(), mlir::Type::getIntOrFloatBitWidth(), and mlir::Type::isUnsignedInteger().
◆ printVariableOpTypeOrInitialValue()
| void mlir::tosa::printVariableOpTypeOrInitialValue | ( | OpAsmPrinter & | p, |
| Operation * | op, | ||
| DenseElementsAttr | varShapeAttr, | ||
| TypeAttr | typeAttr, | ||
| Attribute | initialValueAttr ) |
Definition at line 252 of file TosaOps.cpp.
References convertToMlirShape(), mlir::DenseElementsAttr::getValues(), and mlir::AsmPrinter::printAttribute().
◆ registerShardingInterfaceExternalModels()
| void mlir::tosa::registerShardingInterfaceExternalModels | ( | DialectRegistry & | registry | ) |
Definition at line 116 of file ShardingInterfaceImpl.cpp.
References mlir::DialectRegistry::addExtension().
Referenced by mlir::registerAllDialects().
◆ registerTosaArithConstantToTosaConstPass()
|
inline |
◆ registerTosaArithConstantToTosaConstPassPass()
|
inline |
◆ registerTosaAttachTarget()
◆ registerTosaAttachTargetPass()
◆ registerTosaConvertIntegerTypeToSignless()
|
inline |
◆ registerTosaConvertIntegerTypeToSignlessPass()
|
inline |
◆ registerTosaInferShapesPass()
◆ registerTosaInferShapesPassPass()
|
inline |
◆ registerTosaInputShape()
◆ registerTosaInputShapePass()
◆ registerTosaLayerwiseConstantFoldPass()
|
inline |
◆ registerTosaLayerwiseConstantFoldPassPass()
|
inline |
◆ registerTosaMakeBroadcastablePass()
|
inline |
◆ registerTosaMakeBroadcastablePassPass()
|
inline |
◆ registerTosaNarrowF64ToF32Pass()
|
inline |
◆ registerTosaNarrowF64ToF32PassPass()
|
inline |
◆ registerTosaNarrowI64ToI32Pass()
|
inline |
◆ registerTosaNarrowI64ToI32PassPass()
|
inline |
◆ registerTosaOptionalDecompositionsPass()
|
inline |
◆ registerTosaOptionalDecompositionsPassPass()
|
inline |
◆ registerTosaPasses()
|
inline |
Definition at line 1343 of file Passes.h.
Referenced by mlir::registerAllPasses().
◆ registerTosaReduceTransposes()
◆ registerTosaReduceTransposesPass()
|
inline |
◆ registerTosaToLinalgPipelines()
| void mlir::tosa::registerTosaToLinalgPipelines | ( | ) |
Populates TOSA to linalg pipelines Currently, this includes only the "tosa-to-linalg-pipeline".
Definition at line 118 of file TosaToLinalgPass.cpp.
References addTosaToLinalgPasses(), mlir::tosa::TosaValidationOptions::allowInvalidOpDatatypeCombinations, and mlir::tosa::TosaValidationOptions::strictOpSpecAlignment.
Referenced by mlir::registerAllPasses().
◆ registerTosaValidation()
◆ registerTosaValidationPass()
◆ stringifyVersion()
| llvm::SmallString< 4 > mlir::tosa::stringifyVersion | ( | TosaSpecificationVersion | version | ) |
Definition at line 15 of file TargetEnv.cpp.
References mlir::tosa::TosaSpecificationVersion::getMajor(), and mlir::tosa::TosaSpecificationVersion::getMinor().
Referenced by TosaProfileCompliance::checkProfileOrExtension(), and mlir::tosa::TargetEnv::verifyTargetInformation().
◆ tryGetDenseResourceValues()
| std::optional< ArrayRef< T > > mlir::tosa::tryGetDenseResourceValues | ( | ElementsAttr | attr | ) |
Definition at line 255 of file ConversionUtils.h.
References mlir::AsmResourceBlob::getData(), and mlir::DenseElementsAttr::isValidRawBuffer().
◆ validIntegerRange()
Definition at line 51 of file ConversionUtils.cpp.
Variable Documentation
◆ kInferableDimSize
|
constexpr |
Represents a dimension in the shape of a tensor that can be inferred based on the other provided dimensions.
For example, in a reshape operation, -1 can be used to indicate a size that is the remainder of the other dimensions.
Definition at line 153 of file TosaOps.h.
Referenced by PadSliceOptimization::matchAndRewrite(), and SliceDynamicSizeCanonicalization::matchAndRewrite().
◆ TOSA_LEVEL_EIGHTK
|
staticconstexpr |
Definition at line 43 of file TargetEnv.h.
Referenced by mlir::tosa::TargetEnv::getLevel().
◆ TOSA_LEVEL_NONE
|
staticconstexpr |
Definition at line 45 of file TargetEnv.h.
Referenced by mlir::tosa::TargetEnv::getLevel().
Generated on for MLIR by
1.14.0