Home
last modified time | relevance | path

Searched defs:indexingMaps (Results 1 – 22 of 22) sorted by relevance

/llvm-project/mlir/lib/Dialect/Utils/
H A DStructuredOpsUtils.cpp20 bool mlir::isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul()
46 bool mlir::isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul()
72 bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { in isRowMajorBatchMatmul()
99 bool mlir::isVecmat(ArrayAttr indexingMaps) { in isVecmat()
123 bool mlir::isBatchVecmat(ArrayAttr indexingMaps) { in isBatchVecmat()
148 bool mlir::isMatvec(ArrayAttr indexingMaps) { in isMatvec()
172 bool mlir::isBatchMatvec(ArrayAttr indexingMaps) { in isBatchMatvec()
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DGeneralization.cpp60 SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray(); in generalizeNamedOp() local
H A DElementwiseToLinalg.cpp86 SmallVector<AffineMap, 3> indexingMaps( in matchAndRewrite() local
H A DBufferizableOpInterfaceImpl.cpp129 SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray(); in bufferizesToElementwiseAccess() local
H A DSpecialize.cpp196 auto indexingMaps = genericOp.getIndexingMapsArray(); in specializeLinalgContractions() local
H A DConvertToDestinationStyle.cpp140 SmallVector<AffineMap> indexingMaps( in movePaddingToFillOrGenericOp() local
399 SmallVector<AffineMap> indexingMaps( rewriteInDestinationPassingStyle() local
H A DDecomposeLinalgOps.cpp228 auto indexingMaps = llvm::to_vector( in createResidualGenericOp() local
H A DDataLayoutPropagation.cpp70 SmallVector<AffineMap> indexingMaps = genericOp.getIndexingMapsArray(); in getPackingInfoFromOperand() local
307 SmallVector<AffineMap> indexingMaps; in packGenericOp() local
H A DDropUnitDims.cpp106 SmallVector<AffineMap> indexingMaps = genericOp.getIndexingMapsArray(); in matchAndRewrite() local
391 SmallVector<AffineMap> indexingMaps = genericOp.getIndexingMapsArray(); dropUnitDims() local
H A DSplitReduction.cpp391 SmallVector<AffineMap> indexingMaps = { in splitReductionByScaling() local
H A DMeshShardingInterfaceImpl.cpp303 SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray(); spmdize() local
H A DConvertConv2DToImg2Col.cpp254 SmallVector<AffineMap> indexingMaps = { in rewriteInIm2Col() local
307 SmallVector<AffineMap> indexingMaps = { in rewriteInIm2Col() local
H A DTransforms.cpp149 packLinalgMetadataOnce(SmallVectorImpl<AffineMap> &indexingMaps, in packLinalgMetadataOnce() argument
489 SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray(); pack() local
656 SmallVector<AffineMap> indexingMaps = linalgOp.getIndexingMapsArray(); transposeOneLinalgOperandAndReplace() local
[all...]
H A DElementwiseOpFusion.cpp1601 SmallVector<AffineMap> indexingMaps( cloneToCollapsedOp() local
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp38 SmallVector<AffineMap> indexingMaps; canOpOperandsBeDroppedImpl() local
329 inferContractionDimsImpl(ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iterators) inferContractionDimsImpl() argument
380 inferContractionDims(ArrayRef<AffineMap> indexingMaps) inferContractionDims() argument
744 auto indexingMaps = linalgOp.getIndexingMapsArray(); inferConvolutionDims() local
778 auto indexingMaps = linalgOp.getIndexingMapsArray(); isConvolutionInterfaceImpl() local
[all...]
H A DLinalgOps.cpp928 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
942 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
961 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
972 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
983 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
2606 SmallVector<AffineMap> indexingMaps{identityMap, reductionMap}; computeIteratorTypesAndIndexingMaps() local
[all...]
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp73 ArrayRef<AffineMap> indexingMaps) { in linalgIntBroadcastExtSIAdd() argument
119 SmallVector<AffineMap, 2> indexingMaps = { linalgBroadcastAndMaybeExtSI() local
492 SmallVector<AffineMap, 4> indexingMaps; matchAndRewrite() local
[all...]
H A DTosaToLinalg.cpp1173 SmallVector<AffineMap> indexingMaps = { matchAndRewrite() local
2369 auto indexingMaps = AffineMap::inferFromExprList( matchAndRewrite() local
2488 auto indexingMaps = AffineMap::inferFromExprList( matchAndRewrite() local
[all...]
/llvm-project/mlir/lib/Dialect/Mesh/Interfaces/
H A DShardingInterface.cpp624 getMeshAxisAssignmentForLoopIterators(ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<AffineMap> indexingMaps) getMeshAxisAssignmentForLoopIterators() argument
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp733 build(OpBuilder & builder,OperationState & result,Value lhs,Value rhs,Value acc,ArrayAttr indexingMaps,ArrayAttr iteratorTypes) build() argument
741 build(OpBuilder & builder,OperationState & result,Value lhs,Value rhs,Value acc,ArrayAttr indexingMaps,ArrayAttr iteratorTypes,CombiningKind kind) build() argument
1028 auto indexingMaps = this->getIndexingMapsArray(); getExpectedMaskType() local
1072 getDimMap(ArrayRef<AffineMap> indexingMaps,ArrayAttr iteratorTypes,IteratorType targetIteratorType,MLIRContext * context) getDimMap() argument
1093 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); getIterationBounds() local
1129 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); getContractingDimMap() local
1135 SmallVector<AffineMap, 4> indexingMaps(getIndexingMapsArray()); getBatchDimMap() local
[all...]
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp124 LinalgIndexingMapsConfig indexingMaps; global() member
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp265 SmallVector<AffineMap> indexingMaps = { makeTransposeOp() local