| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 31 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, context)); in isRowMajorMatmul()
32 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, context)); in isRowMajorMatmul()
33 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {m, n}, context)); in isRowMajorMatmul()
42 auto mapA = AffineMapAttr::get(AffineMap::get(3, 0, {k, n}, context)); in isColumnMajorMatmul()
43 auto mapB = AffineMapAttr::get(AffineMap::get(3, 0, {m, k}, context)); in isColumnMajorMatmul()
44 auto mapC = AffineMapAttr::get(AffineMap::get(3, 0, {n, m}, context)); in isColumnMajorMatmul()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.h | 109 AffineMapAttr getSrcMapAttr() { in getSrcMapAttr()
110 return getAttr(getSrcMapAttrName()).cast<AffineMapAttr>(); in getSrcMapAttr()
148 AffineMapAttr getDstMapAttr() { in getDstMapAttr()
149 return getAttr(getDstMapAttrName()).cast<AffineMapAttr>(); in getDstMapAttr()
177 AffineMapAttr getTagMapAttr() { in getTagMapAttr()
178 return getAttr(getTagMapAttrName()).cast<AffineMapAttr>(); in getTagMapAttr()
291 AffineMapAttr getTagMapAttr() { in getTagMapAttr()
292 return getAttr(getTagMapAttrName()).cast<AffineMapAttr>(); in getTagMapAttr()
|
| D | AffineOps.td | 68 let arguments = (ins AffineMapAttr:$map, Variadic<Index>:$mapOperands);
266 AffineMapAttr getLowerBoundMapAttr() {
267 return getAttr(getLowerBoundAttrName()).cast<AffineMapAttr>();
271 AffineMapAttr getUpperBoundMapAttr() {
272 return getAttr(getUpperBoundAttrName()).cast<AffineMapAttr>();
473 AffineMapAttr getAffineMapAttr() {
474 return getAttr(getMapAttrName()).cast<AffineMapAttr>();
524 let arguments = (ins AffineMapAttr:$map, Variadic<Index>:$operands);
645 AffineMapAttr:$lowerBoundsMap,
646 AffineMapAttr:$upperBoundsMap,
[all …]
|
| D | AffineMemoryOpInterfaces.td | 72 /*desc=*/"Returns the AffineMapAttr associated with 'memref'.",
151 /*desc=*/"Returns the AffineMapAttr associated with 'memref'.",
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 361 AffineMapAttr mapAttr; in parseAffineApplyOp()
970 result.addAttribute(getSrcMapAttrName(), AffineMapAttr::get(srcMap)); in build()
973 result.addAttribute(getDstMapAttrName(), AffineMapAttr::get(dstMap)); in build()
976 result.addAttribute(getTagMapAttrName(), AffineMapAttr::get(tagMap)); in build()
1009 AffineMapAttr srcMapAttr; in parse()
1012 AffineMapAttr dstMapAttr; in parse()
1015 AffineMapAttr tagMapAttr; in parse()
1136 result.addAttribute(getTagMapAttrName(), AffineMapAttr::get(tagMap)); in build()
1158 AffineMapAttr tagMapAttr; in parse()
1230 result.addAttribute(getLowerBoundAttrName(), AffineMapAttr::get(lbMap)); in build()
[all …]
|
| /external/llvm-project/flang/lib/Optimizer/Dialect/ |
| D | FIRType.cpp | 61 mlir::AffineMapAttr map; in parseBox()
172 mlir::AffineMapAttr map; in parseSequence()
545 using KeyTy = std::tuple<mlir::Type, mlir::AffineMapAttr>;
550 hashVal, llvm::hash_combine(std::get<mlir::AffineMapAttr>(key))); in hashKey()
555 std::get<mlir::AffineMapAttr>(key) == getLayoutMap(); in operator ==()
562 std::get<mlir::AffineMapAttr>(key)}; in construct()
566 mlir::AffineMapAttr getLayoutMap() const { return map; } in getLayoutMap()
570 mlir::AffineMapAttr map;
574 explicit BoxTypeStorage(mlir::Type eleTy, mlir::AffineMapAttr map) in BoxTypeStorage()
710 std::tuple<SequenceType::Shape, mlir::Type, mlir::AffineMapAttr>;
[all …]
|
| /external/llvm-project/flang/include/flang/Optimizer/Dialect/ |
| D | FIRType.h | 161 static BoxType get(mlir::Type eleTy, mlir::AffineMapAttr map = {});
163 mlir::AffineMapAttr getLayoutMap() const;
167 mlir::AffineMapAttr map);
288 mlir::AffineMapAttr map = {});
297 mlir::AffineMapAttr getLayoutMap() const;
322 mlir::Type eleTy, mlir::AffineMapAttr map);
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Interchange.cpp | 68 AffineMap m = indexingMaps[i].cast<AffineMapAttr>().getValue(); in interchange()
71 newIndexingMaps.push_back(AffineMapAttr::get(m)); in interchange()
|
| D | Loops.cpp | 99 auto mapsRange = op.indexing_maps().template getAsRange<AffineMapAttr>(); in getInputAndOutputIndices()
101 llvm::map_range(mapsRange, [](AffineMapAttr a) { return a.getValue(); })); in getInputAndOutputIndices()
341 auto mapsRange = convOp.indexing_maps().getAsRange<AffineMapAttr>(); in emitScalarImplementation()
343 llvm::map_range(mapsRange, [](AffineMapAttr a) { return a.getValue(); })); in emitScalarImplementation()
|
| D | FusionOnTensors.cpp | 84 fusedOpIndexingMapAttrs.push_back(AffineMapAttr::get(indexingMap)); in getIndexingMapOfProducerOperandsInFusedOp()
422 return attr.cast<AffineMapAttr>() in isFusableWithReshapeByDimExpansion()
711 op.indexing_maps().template getAsValueRange<AffineMapAttr>()); in matchAndRewrite()
800 producer.indexing_maps().getAsValueRange<AffineMapAttr>()); in matchAndRewrite()
888 linalgOp.indexing_maps().getAsValueRange<AffineMapAttr>()); in matchAndRewrite()
|
| D | DropUnitDims.cpp | 143 [](AffineMap map) -> Attribute { return AffineMapAttr::get(map); })), in replaceUnitDims()
276 reassociationMaps.push_back(AffineMapAttr::get(AffineMap::get( in replaceUnitExtents()
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | ParallelLoopMapperAttr.td | 42 StructFieldAttr<"map", AffineMapAttr>,
43 StructFieldAttr<"bound", AffineMapAttr>]>;
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | ParallelLoopMapper.cpp | 40 AffineMapAttr::get(map), AffineMapAttr::get(bound), context); in getParallelLoopDimMappingAttr()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
| D | LinalgToSPIRV.cpp | 96 auto inputMap = genericOp.indexing_maps().getValue()[0].cast<AffineMapAttr>(); in matchAsPerformingReduction()
98 genericOp.indexing_maps().getValue()[1].cast<AffineMapAttr>(); in matchAsPerformingReduction()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 147 AffineMapAttr::get(*inputPerm)); in matchAndRewrite()
151 AffineMapAttr::get(*outputPerm)); in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 128 OptionalAttr<AffineMapAttr>:$inputPermutation,
129 OptionalAttr<AffineMapAttr>:$outputPermutation);
136 $_builder, $_state, input, output, AffineMapAttr(), AffineMapAttr());
548 - indexing_maps: a list of AffineMapAttr, one AffineMapAttr per each input
549 and output view. Such AffineMapAttr specifies the mapping between the
683 - indexing_maps: a list of AffineMapAttr, one AffineMapAttr per each input
684 and output view. Such AffineMapAttr specifies the mapping between the
|
| D | LinalgStructuredOpsInterface.td | 670 $_op.indexing_maps().template getAsValueRange<AffineMapAttr>());
790 auto r = $_op.indexing_maps().template getAsRange<AffineMapAttr>();
792 llvm::map_range(r, [](AffineMapAttr a) { return a.getValue(); }));
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinDialect.cpp | 34 if (attr.isa<AffineMapAttr>()) { in getAlias()
56 addAttributes<AffineMapAttr, ArrayAttr, DenseIntOrFPElementsAttr, in initialize()
|
| D | AsmPrinter.cpp | 431 void printAffineMapOfSSAIds(AffineMapAttr, ValueRange) override {} in printAffineMapOfSSAIds() argument
594 visit(AffineMapAttr::get(map)); in visit()
1538 } else if (auto affineMapAttr = attr.dyn_cast<AffineMapAttr>()) { in printAttribute()
1840 printAttribute(AffineMapAttr::get(map)); in printType()
2275 void printAffineMapOfSSAIds(AffineMapAttr mapAttr,
2487 void OperationPrinter::printAffineMapOfSSAIds(AffineMapAttr mapAttr, in printAffineMapOfSSAIds()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| D | SimplifyAffineStructures.cpp | 89 if (auto mapAttr = attr.second.dyn_cast<AffineMapAttr>()) in runOnFunction()
|
| /external/llvm-project/mlir/lib/CAPI/IR/ |
| D | BuiltinAttributes.cpp | 23 return unwrap(attr).isa<AffineMapAttr>(); in mlirAttributeIsAAffineMap()
27 return wrap(AffineMapAttr::get(unwrap(map))); in mlirAffineMapAttrGet()
31 return wrap(unwrap(attr).cast<AffineMapAttr>().getValue()); in mlirAffineMapAttrGetValue()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | Utils.cpp | 102 AffineMap oldMap = oldMapAttrPair.second.cast<AffineMapAttr>().getValue(); in replaceAllMemRefUsesWith()
204 auto newMapAttr = AffineMapAttr::get(newMap); in replaceAllMemRefUsesWith()
|
| /external/llvm-project/mlir/test/lib/Dialect/Affine/ |
| D | TestVectorizationUtils.cpp | 215 .cast<AffineMapAttr>() in testComposeMaps()
|
| /external/llvm-project/mlir/lib/Conversion/VectorToSCF/ |
| D | VectorToSCF.cpp | 284 AffineMapAttr::get(map), xferOp.padding(), in doReplace()
389 AffineMapAttr::get(map), masked); in doReplace()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | BuiltinAttributes.h | 48 class AffineMapAttr
49 : public Attribute::AttrBase<AffineMapAttr, Attribute,
55 static AffineMapAttr get(AffineMap value);
|