Searched refs:indexingMap (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/tools/mlir-linalg-ods-gen/ |
| D | mlir-linalg-ods-gen.cpp | 866 : Expression(Kind::TensorUse), tensorId(name), indexingMap(map) {} in TensorUse()
874 return tensorId == other.tensorId && indexingMap == other.indexingMap; in operator ==()
883 AffineMap indexingMap; member
1009 AffineMap indexingMap; member
1293 use.indexingMap = in parseOneComprehension()
1295 use.indexingMap.getResults(), parser.context); in parseOneComprehension()
1308 tensor.indexingMap = def.indexingMap; in parseOneComprehension()
1320 << "\nuse: " << use.tensorId << " map: " << use.indexingMap); in parseOneComprehension()
1324 if (tensor.indexingMap && state.orderedTensorArgs.count(use) == 0) { in parseOneComprehension()
1325 LLVM_DEBUG(llvm::dbgs() << "\nexisting: " << tensor.indexingMap); in parseOneComprehension()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 83 AffineMap indexingMap = t1.compose(fusedConsumerArgIndexMap); in getIndexingMapOfProducerOperandsInFusedOp() local
84 fusedOpIndexingMapAttrs.push_back(AffineMapAttr::get(indexingMap)); in getIndexingMapOfProducerOperandsInFusedOp()
512 for (AffineMap indexingMap : linalgOp.getIndexingMaps()) { in fuseWithReshapeByExpansion() local
514 for (AffineExpr expr : indexingMap.getResults()) { in fuseWithReshapeByExpansion()
522 AffineMap::get(remapping.back(), indexingMap.getNumSymbols(), newExprs, in fuseWithReshapeByExpansion()
551 AffineMap indexingMap = linalgOp.getIndexingMap(operand.index()); in fuseWithReshapeByExpansion() local
554 getReshapeInfo(indexingMap, reassociation, expandedOperandShape); in fuseWithReshapeByExpansion()
569 AffineMap indexingMap = in fuseWithReshapeByExpansion() local
573 getReshapeInfo(indexingMap, reassociation, expandedResultShape); in fuseWithReshapeByExpansion()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 1751 AffineMap indexingMap = linalgOp.getInputIndexingMap(i); in matchAndRewrite() local
1755 auto pair = canonicalInput.insert({{input, indexingMap}, i}); in matchAndRewrite()
|