Home
last modified time | relevance | path

Searched refs:getIndexingMaps (Results 1 – 8 of 8) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DGeneralization.cpp41 SmallVector<AffineMap, 4> indexingMaps = namedOp.getIndexingMaps(); in createGenericOpFromNamedOp()
151 SmallVector<AffineMap, 4> indexingMaps = convOp.getIndexingMaps(); in createGenericOp()
DDropUnitDims.cpp181 SmallVector<AffineMap, 4> indexingMaps = op.getIndexingMaps(); in matchAndRewrite()
309 llvm::zip(op.getIndexingMaps(), op.getInputOutputShapedTypes())) { in matchAndRewrite()
DFusionOnTensors.cpp512 for (AffineMap indexingMap : linalgOp.getIndexingMaps()) { in fuseWithReshapeByExpansion()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc377 Derived::getIndexingMaps(op, &rewriter); in matchAndRewrite()
412 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy broadcast_op, in getIndexingMaps() function in mlir::__anon57cf929c0111::BroadcastConverter
449 static SmallVector<AffineMap, 2> getIndexingMaps( in getIndexingMaps() function in mlir::__anon57cf929c0111::HloBroadcastInDimConverter
572 auto indexing_maps = getIndexingMaps(op, broadcast_dims, result_shape, in matchAndRewrite()
656 SmallVector<AffineMap, 2> getIndexingMaps(lmhlo::BroadcastInDimOp op, in getIndexingMaps() function in mlir::__anon57cf929c0111::LhloBroadcastInDimConverter
697 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy op, Builder* b) { in getIndexingMaps() function in mlir::__anon57cf929c0111::TransposeConverter
1018 static SmallVector<AffineMap, 2> getIndexingMaps(OpTy op, Builder* b) { in getIndexingMaps() function in mlir::__anon57cf929c0111::ReverseConverter
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOpsInterface.td665 /*methodName=*/"getIndexingMaps",
683 return getIndexingMaps()[i];
696 return getIndexingMaps()[i];
709 return getIndexingMaps()[i + $_op.getNumInputs()];
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp331 AffineMap lhsMap = op.getIndexingMaps()[0]; in verifyOutputShape()
332 AffineMap rhsMap = op.getIndexingMaps()[1]; in verifyOutputShape()
347 AffineMap resMap = op.getIndexingMaps()[2]; in verifyOutputShape()
475 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getIterationBounds()
511 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getContractingDimMap()
517 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getBatchDimMap()
522 SmallVector<AffineMap, 4> ContractionOp::getIndexingMaps() { in getIndexingMaps() function in ContractionOp
DVectorTransforms.cpp1710 SmallVector<AffineMap, 4> maps = op.getIndexingMaps(); in matchAndRewrite()
1820 SmallVector<AffineMap, 4> maps = op.getIndexingMaps(); in matchAndRewrite()
2014 SmallVector<AffineMap, 4> iMap = op.getIndexingMaps(); in lowerParallel()
2065 SmallVector<AffineMap, 4> iMap = op.getIndexingMaps(); in lowerReduction()
/external/llvm-project/mlir/include/mlir/Dialect/Vector/
DVectorOps.td176 SmallVector<AffineMap, 4> getIndexingMaps();