Home
last modified time | relevance | path

Searched defs:iteratorTypes (Results 1 – 24 of 24) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/Utils/
H A DStructuredOpsUtils.h77 inline void findPositionsOfType(ArrayRef<utils::IteratorType> iteratorTypes, in findPositionsOfType()
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseToLinalg.cpp89 SmallVector<utils::IteratorType, 6> iteratorTypes( in matchAndRewrite() local
H A DConvertToDestinationStyle.cpp138 SmallVector<utils::IteratorType> iteratorTypes(resultType.getRank(), in movePaddingToFillOrGenericOp() local
397 SmallVector<utils::IteratorType> iteratorTypes(tensorType.getRank(), rewriteInDestinationPassingStyle() local
H A DElementwiseOpFusion.cpp524 SmallVector<utils::IteratorType> iteratorTypes = isFusableWithReshapeByDimExpansion() local
877 SmallVector<utils::IteratorType> iteratorTypes( fuseWithReshapeByExpansion() local
1233 auto iteratorTypes = genericOp.getIteratorTypesArray(); getCollapsableIterationSpaceDims() local
1407 getCollapsedOpIteratorTypes(ArrayRef<utils::IteratorType> iteratorTypes,const CollapsingInfo & collapsingInfo) getCollapsedOpIteratorTypes() argument
1606 SmallVector<utils::IteratorType> iteratorTypes(getCollapsedOpIteratorTypes( cloneToCollapsedOp() local
[all...]
H A DLoops.cpp224 auto iteratorTypes = linalgOp.getIteratorTypesArray(); in linalgOpToLoopsImpl() local
H A DSplitReduction.cpp364 auto iteratorTypes = op.getIteratorTypesArray(); in splitReductionByScaling() local
H A DMeshShardingInterfaceImpl.cpp284 SmallVector<utils::IteratorType> iteratorTypes = getReductionLoopIteratorKinds() local
H A DTransforms.cpp150 SmallVectorImpl<utils::IteratorType> &iteratorTypes, in packLinalgMetadataOnce() argument
490 SmallVector<utils::IteratorType> iteratorTypes = pack() local
[all...]
H A DTiling.cpp650 SmallVector<utils::IteratorType, 4> iteratorTypes; tileLinalgOpImpl() local
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp271 SmallVector<utils::IteratorType> iteratorTypes(transposeVector.size(), makeTransposeOp() local
298 SmallVector<utils::IteratorType> iteratorTypes(memrefTypeTo.getRank(), makeMemRefCopyOp() local
315 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
358 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo>) doit() argument
410 generateParallelLoopNest(OpBuilder & b,Location loc,ValueRange lbs,ValueRange ubs,ValueRange steps,ArrayRef<utils::IteratorType> iteratorTypes,ArrayRef<linalg::ProcInfo> procInfo,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuilderFn,SmallVectorImpl<Value> & ivStorage) generateParallelLoopNest() argument
525 doit(OpBuilder & b,Location loc,ArrayRef<Range> loopRanges,LinalgOp linalgOp,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<scf::ValueVector (OpBuilder &,Location,ValueRange,ValueRange)> bodyBuilderFn,ArrayRef<linalg::ProcInfo> procInfo) doit() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp929 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayAttr indexingMaps,ArrayAttr iteratorTypes,StringAttr doc,StringAttr libraryCall,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
943 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,StringRef doc,StringRef libraryCall,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
962 build(OpBuilder & builder,OperationState & result,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,StringRef doc,StringRef libraryCall,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
973 build(OpBuilder & builder,OperationState & result,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
984 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
1004 auto iteratorTypes = print() local
1072 auto iteratorTypes = dyn_cast_or_null<ArrayAttr>( parse() local
1547 SmallVector<utils::IteratorType> iteratorTypes(inputRank, getIteratorTypesArray() local
2479 SmallVector<utils::IteratorType> iteratorTypes(getInputOperandRank(), getLoopIteratorTypes() local
2593 SmallVector<utils::IteratorType> iteratorTypes(inputRank, computeIteratorTypesAndIndexingMaps() local
[all...]
H A DLinalgInterfaces.cpp794 auto iteratorTypes = linalgOp.getIteratorTypesArray(); isConvolutionInterfaceImpl() local
/llvm-project/mlir/lib/Dialect/ArmNeon/Transforms/
H A DLowerContractionToSMMLAPattern.cpp75 auto iteratorTypes = op.getIteratorTypesArray(); in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DCodegenEnv.cpp169 const auto iteratorTypes = linalgOp.getIteratorTypesArray(); in isAdmissibleTensorExp() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorTilingInterfaceImpl.cpp31 SmallVector<utils::IteratorType> iteratorTypes( getLoopIteratorTypes() local
110 SmallVector<utils::IteratorType> iteratorTypes( getLoopIteratorTypes() local
360 SmallVector<utils::IteratorType> iteratorTypes( getLoopIteratorTypes() local
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DLowerVectorContract.cpp54 static SmallVector<Attribute> adjustIter(ArrayAttr iteratorTypes, in adjustIter() argument
174 ArrayAttr iteratorTypes) { in getReductionIndex() argument
679 auto iteratorTypes = op.getIteratorTypes().getValue(); in matchAndRewriteMaskableOp() local
1299 auto iteratorTypes = op.getIteratorTypes().getValue(); matchAndRewriteMaskableOp() local
[all...]
H A DVectorTransforms.cpp153 SmallVector<vector::IteratorType> iteratorTypes; in matchAndRewrite() local
1470 auto iteratorTypes = op.getIteratorTypes().getValue(); matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp85 auto iteratorTypes = contract.getIteratorTypes().getValue(); in contractSupportsMMAMatrixType() local
401 auto iteratorTypes = op.getIteratorTypes().getValue(); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseReinterpretMap.cpp501 const auto iteratorTypes = linalgOp.getIteratorTypesArray(); in isAdmissibleOrder() local
H A DSparsification.cpp819 auto iteratorTypes = op.getIteratorTypesArray(); shouldTryParallize() local
H A DSparseGPUCodegen.cpp1262 const auto iteratorTypes = op.getIteratorTypesArray(); in matchAndRewrite() local
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp2012 SmallVector<utils::IteratorType, 4> iteratorTypes; matchAndRewrite() local
2357 llvm::SmallVector<utils::IteratorType, 5> iteratorTypes = { matchAndRewrite() local
2474 SmallVector<utils::IteratorType, 5> iteratorTypes = { matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp742 iteratorTypes, [&](IteratorType t) -> mlir::Attribute { in build() argument
716 build(OpBuilder & builder,OperationState & result,Value lhs,Value rhs,Value acc,ArrayRef<ArrayRef<AffineExpr>> indexingExprs,ArrayRef<IteratorType> iteratorTypes) build() argument
734 build(OpBuilder & builder,OperationState & result,Value lhs,Value rhs,Value acc,ArrayAttr indexingMaps,ArrayAttr iteratorTypes) build() argument
780 ArrayAttr iteratorTypes = llvm::cast<ArrayAttr>( parse() local
826 auto iteratorTypes = print() local
1072 getDimMap(ArrayRef<AffineMap> indexingMaps,ArrayAttr iteratorTypes,IteratorType targetIteratorType,MLIRContext * context) getDimMap() argument
[all...]
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp125 SmallVector<LinalgIteratorTypeDef> iteratorTypes; global() member