| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | AffineMap.cpp | 22 return wrap(unwrap(affineMap).getContext()); in mlirAffineMapGetContext()
32 unwrap(affineMap).print(stream); in mlirAffineMapPrint()
35 void mlirAffineMapDump(MlirAffineMap affineMap) { unwrap(affineMap).dump(); } in mlirAffineMapDump() argument
75 return unwrap(affineMap).isIdentity(); in mlirAffineMapIsIdentity()
79 return unwrap(affineMap).isMinorIdentity(); in mlirAffineMapIsMinorIdentity()
83 return unwrap(affineMap).isEmpty(); in mlirAffineMapIsEmpty()
95 return unwrap(affineMap).getNumDims(); in mlirAffineMapGetNumDims()
99 return unwrap(affineMap).getNumSymbols(); in mlirAffineMapGetNumSymbols()
103 return unwrap(affineMap).getNumResults(); in mlirAffineMapGetNumResults()
111 return unwrap(affineMap).getNumInputs(); in mlirAffineMapGetNumInputs()
[all …]
|
| H A D | AffineExpr.cpp | 60 MlirAffineMap affineMap) { in mlirAffineExprCompose() argument
61 return wrap(unwrap(affineExpr).compose(unwrap(affineMap))); in mlirAffineExprCompose()
|
| /llvm-project-15.0.7/mlir/include/mlir-c/ |
| H A D | AffineMap.h | 47 static inline bool mlirAffineMapIsNull(MlirAffineMap affineMap) { in mlirAffineMapIsNull() argument
48 return !affineMap.ptr; in mlirAffineMapIsNull()
57 MLIR_CAPI_EXPORTED void mlirAffineMapPrint(MlirAffineMap affineMap,
62 MLIR_CAPI_EXPORTED void mlirAffineMapDump(MlirAffineMap affineMap);
110 MLIR_CAPI_EXPORTED bool mlirAffineMapIsIdentity(MlirAffineMap affineMap);
116 MLIR_CAPI_EXPORTED bool mlirAffineMapIsEmpty(MlirAffineMap affineMap);
125 mlirAffineMapGetSingleConstantResult(MlirAffineMap affineMap);
138 mlirAffineMapGetResult(MlirAffineMap affineMap, intptr_t pos);
147 mlirAffineMapIsProjectedPermutation(MlirAffineMap affineMap);
163 mlirAffineMapGetMajorSubMap(MlirAffineMap affineMap, intptr_t numResults);
[all …]
|
| H A D | AffineExpr.h | 93 MlirAffineExpr affineExpr, struct MlirAffineMap affineMap);
|
| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | IRAffine.cpp | 386 affineMap(map) {} in PyAffineMapExprList()
395 return PyAffineExpr(affineMap.getContext(), in getRawElement()
404 PyAffineMap affineMap; member in __anonf6590d6b0611::PyAffineMapExprList
409 return mlirAffineMapEqual(affineMap, other.affineMap); in operator ==()
717 MlirAffineMap affineMap = in populateIRAffine() local
719 return PyAffineMap(context->getRef(), affineMap); in populateIRAffine()
733 MlirAffineMap affineMap = in populateIRAffine() local
743 MlirAffineMap affineMap = in populateIRAffine() local
782 MlirAffineMap affineMap = in populateIRAffine() local
792 MlirAffineMap affineMap = in populateIRAffine() local
[all …]
|
| H A D | IRModule.h | 969 PyAffineMap(PyMlirContextRef contextRef, MlirAffineMap affineMap) in PyAffineMap() argument
970 : BaseContextObject(std::move(contextRef)), affineMap(affineMap) {} in PyAffineMap()
972 operator MlirAffineMap() const { return affineMap; } in MlirAffineMap()
973 MlirAffineMap get() const { return affineMap; } in get()
985 MlirAffineMap affineMap;
|
| H A D | IRAttributes.cpp | 87 [](PyAffineMap &affineMap) { in bindDerived() argument
88 MlirAttribute attr = mlirAffineMapAttrGet(affineMap.get()); in bindDerived()
89 return PyAffineMapAttribute(affineMap.getContext(), attr); in bindDerived()
|
| /llvm-project-15.0.7/mlir/include/mlir-c/Bindings/Python/ |
| H A D | Interop.h | 294 static inline PyObject *mlirPythonAffineMapToCapsule(MlirAffineMap affineMap) { in mlirPythonAffineMapToCapsule() argument
295 return PyCapsule_New(MLIR_PYTHON_GET_WRAPPED_POINTER(affineMap), in mlirPythonAffineMapToCapsule()
305 MlirAffineMap affineMap = {ptr}; in mlirPythonCapsuleToAffineMap() local
306 return affineMap; in mlirPythonCapsuleToAffineMap()
|
| /llvm-project-15.0.7/mlir/test/CAPI/ |
| H A D | ir.c | 1194 MlirAffineMap affineMap = mlirAffineMapZeroResultGet(ctx, 3, 2); in printAffineMap() local
1206 mlirAffineMapDump(affineMap); in printAffineMap()
1220 mlirAffineMapIsIdentity(affineMap) || in printAffineMap()
1228 mlirAffineMapIsMinorIdentity(affineMap) || in printAffineMap()
1242 mlirAffineMapIsSingleConstant(affineMap) || in printAffineMap()
1253 mlirAffineMapGetNumDims(affineMap) != 3 || in printAffineMap()
1261 mlirAffineMapGetNumSymbols(affineMap) != 2 || in printAffineMap()
1269 mlirAffineMapGetNumResults(affineMap) != 0 || in printAffineMap()
1277 mlirAffineMapGetNumInputs(affineMap) != 5 || in printAffineMap()
1286 mlirAffineMapIsProjectedPermutation(affineMap) || in printAffineMap()
[all …]
|
| /llvm-project-15.0.7/mlir/tools/mlir-linalg-ods-gen/ |
| H A D | mlir-linalg-ods-yaml-gen.cpp | 63 AffineMap affineMap() { return affineMapAttr.getValue(); } in affineMap() function
736 size_t size = arg.indexAttrMap->affineMap().getNumResults(); in generateNamedGenericOpOds()
829 AffineMap firstMap = staticMaps->front().affineMap(); in generateNamedGenericOpDefns()
867 llvm::enumerate(arg.indexAttrMap->affineMap().getResults())) { in generateNamedGenericOpDefns()
975 attrFmt, arg.name, arg.indexAttrMap->affineMap().getNumResults())); in generateNamedGenericOpDefns()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/ |
| H A D | Utils.h | 304 AffineMap affineMap,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 582 AffineMap affineMap = contract.getIndexingMapsArray()[accIndex]; in matchAndRewrite() local
587 for (unsigned i : llvm::seq(unsigned(0), affineMap.getNumResults())) in matchAndRewrite()
588 map[affineMap.getDimPosition(i)] = extract.getResultType().getDimSize(i); in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Transforms/ |
| H A D | AffinePromotion.cpp | 399 auto affineMap = in createAffineOps() local
407 affineMap, indexArgs); in createAffineOps()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 820 auto affineMap = rewriter.getMultiDimIdentityMap(resultTy.getRank()); in matchAndRewrite() local
828 ArrayRef<AffineMap>({affineMap, affineMap}), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 225 AffineMap affineMap, in expandAffineMap() argument
227 auto numDims = affineMap.getNumDims(); in expandAffineMap()
229 llvm::map_range(affineMap.getResults(), in expandAffineMap()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.cpp | 1793 AffineMap affineMap = map.getAffineMap(); in getRelationFromMap() local
1794 if (failed(getRelationFromMap(affineMap, rel))) in getRelationFromMap()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 524 AffineMap affineMap = getMap(); in verify() local
527 if (getNumOperands() != affineMap.getNumDims() + affineMap.getNumSymbols()) in verify()
532 if (affineMap.getNumResults() != 1) in verify()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.td | 551 OpBuilder<(ins "AffineMap":$affineMap, "ValueRange":$mapOperands),
553 build($_builder, $_state, $_builder.getIndexType(), affineMap, mapOperands);
|