| /llvm-project/mlir/lib/Dialect/Utils/ |
| H A D | IndexingUtils.cpp | 48 ExprType linearIndex = zero; in linearizeImpl() local
55 SmallVector<ExprType> delinearizeImpl(ExprType linearIndex, in delinearizeImpl() argument
107 SmallVector<int64_t> mlir::delinearize(int64_t linearIndex, in delinearize() argument
187 delinearize(AffineExpr linearIndex,ArrayRef<AffineExpr> strides) delinearize() argument
194 delinearize(AffineExpr linearIndex,ArrayRef<int64_t> strides) delinearize() argument
[all...] |
| /llvm-project/mlir/lib/Conversion/LLVMCommon/ |
| H A D | VectorPattern.cpp | 49 unsigned linearIndex) { in getCoordinates()
70 for (unsigned linearIndex = 0; linearIndex < ub; ++linearIndex) { in nDVectorIterate() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConstantFold.cpp | 183 auto computeRemappedLinearIndex = [&](int linearIndex) { in matchAndRewrite()
219 for (int linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
240 for (int linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
|
| H A D | Vectorization.cpp | 3048 auto linearIndex = [&](int64_t kw, int64_t w) { conv() local
3302 auto linearIndex = [&](int64_t kw, int64_t w) { depthwiseConv() local
|
| /llvm-project/mlir/include/mlir/Dialect/Utils/ |
| H A D | IndexingUtils.h | 283 getTileOffsets(T linearIndex) getTileOffsets() argument
[all...] |
| /llvm-project/mlir/lib/Conversion/MathToLibm/ |
| H A D | MathToLibm.cpp | 91 for (auto linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Transforms.cpp | 52 Value linearIndex = builder.create<ProcessLinearIndexOp>(mesh); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Bindings/Python/ |
| H A D | PybindUtils.h | |
| /llvm-project/mlir/lib/Conversion/MathToFuncs/ |
| H A D | MathToFuncs.cpp | 127 for (int64_t linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | SPIRVConversion.cpp | 1056 Value linearIndex; getOpenCLElementPtr() local
|
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | Utils.cpp | 1853 delinearizeIndex(OpBuilder & b,Location loc,Value linearIndex,ArrayRef<Value> basis) delinearizeIndex() argument
|
| H A D | LoopUtils.cpp | 1736 Value linearIndex = processorId.front(); mapLoopToProcessorIds() local
|
| /llvm-project/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 4516 build(OpBuilder & builder,OperationState & result,Value linearIndex,ArrayRef<OpFoldResult> basis) build() argument
|