Searched refs:linearIndex (Results 1 – 10 of 10) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/ |
| H A D | VectorPattern.cpp | 49 unsigned linearIndex) { in getCoordinates() argument
53 res.push_back(linearIndex % basisElement); in getCoordinates()
54 linearIndex = linearIndex / basisElement; in getCoordinates()
56 if (linearIndex > 0) in getCoordinates()
70 for (unsigned linearIndex = 0; linearIndex < ub; ++linearIndex) { in nDVectorIterate() local
71 auto coords = getCoordinates(info.arraySizes, linearIndex); in nDVectorIterate()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | ConstantFold.cpp | 183 auto computeRemappedLinearIndex = [&](int linearIndex) { in matchAndRewrite() argument
184 int totalCount = linearIndex; in matchAndRewrite()
219 for (int linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
220 computeRemappedLinearIndex(linearIndex); in matchAndRewrite()
240 for (int linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
241 computeRemappedLinearIndex(linearIndex); in matchAndRewrite()
|
| H A D | Vectorization.cpp | 1453 auto linearIndex = [&](int64_t kw, int64_t w) { in conv() local
1461 builder, loc, lhsVals[linearIndex(kw, w)], rhsVals[kw], resVals[w]); in conv()
1577 auto linearIndex = [&](int64_t kw, int64_t w) { in depthwiseConv() local
1585 builder, loc, lhsVals[linearIndex(kw, w)], rhsVals[kw], resVals[w]); in depthwiseConv()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | IndexingUtils.cpp | 15 int64_t linearIndex = 0; in linearize() local
17 linearIndex += offsets[idx] * basis[idx]; in linearize()
18 return linearIndex; in linearize()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MathToLibm/ |
| H A D | MathToLibm.cpp | 78 for (auto linearIndex = 0; linearIndex < numElements; ++linearIndex) { in matchAndRewrite() local
79 SmallVector<int64_t> positions = delinearize(strides, linearIndex); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | PybindUtils.h | 234 intptr_t linearIndex = index * step + startIndex;
235 assert(linearIndex >= 0 &&
236 linearIndex < static_cast<Derived *>(this)->getRawNumElements() &&
238 return linearIndex;
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | IndexingUtils.h | 31 int64_t linearIndex);
|
| /llvm-project-15.0.7/mlir/include/mlir/Conversion/LLVMCommon/ |
| H A D | VectorPattern.h | 45 unsigned linearIndex);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 1842 Value linearIndex = processorId.front(); in mapLoopToProcessorIds() local
1845 loc, mulMap, ValueRange{linearIndex, numProcessors[i]}); in mapLoopToProcessorIds()
1846 linearIndex = b.create<AffineApplyOp>( in mapLoopToProcessorIds()
1851 loc, mulMap, ValueRange{linearIndex, forOp.getStep()}); in mapLoopToProcessorIds()
|
| /llvm-project-15.0.7/mlir/test/Conversion/VectorToLLVM/ |
| H A D | vector-to-llvm.mlir | 1485 // CHECK: %[[linearIndex:.*]] = arith.constant dense
1494 // CHECK: %[[mask:.*]] = arith.cmpi slt, %[[linearIndex]], %[[boundVect]]
1580 // CHECK: %[[linearIndex:.*]] = arith.constant dense
1589 // CHECK: %[[mask:.*]] = arith.cmpi slt, %[[linearIndex]], %[[boundVect]]
|