| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | DropUnitDims.cpp | 115 dimReplacements.push_back(getAffineDimExpr(numKeptDims++, context)); in replaceUnitDims()
266 reassociations.push_back(getAffineDimExpr(dim++, context)); in replaceUnitExtents()
268 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
274 reassociations.push_back(getAffineDimExpr(dim, context)); in replaceUnitExtents()
455 rewriter.getAffineDimExpr(expandedDim++)); in matchAndRewrite()
460 rewriter.getAffineDimExpr(expandedDim++)); in matchAndRewrite()
468 rewriter.getAffineDimExpr(expandedDim++)); in matchAndRewrite()
|
| D | Transforms.cpp | 391 AffineExpr lb = getAffineDimExpr(dims.size(), ctx); in substituteLoopInExpr()
393 AffineExpr ub = getAffineDimExpr(dims.size(), ctx); in substituteLoopInExpr()
415 AffineExpr dimExpr = getAffineDimExpr(dimIdx, expr.getContext()); in substitute()
|
| D | Tiling.cpp | 292 {getAffineDimExpr(/*position=*/0, b.getContext()), in makeTiledShapes()
293 getAffineDimExpr(/*position=*/1, b.getContext()) - in makeTiledShapes()
294 getAffineDimExpr(/*position=*/2, b.getContext())}, in makeTiledShapes()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | ParallelLoopTiling.cpp | 69 {getAffineDimExpr(/*position=*/0, b.getContext()), in tileParallelLoop()
70 getAffineDimExpr(/*position=*/1, b.getContext()) - in tileParallelLoop()
71 getAffineDimExpr(/*position=*/2, b.getContext())}, in tileParallelLoop()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_transforms.cc | 124 replacements.push_back(builder.getAffineDimExpr(i) * size + in TileLoop()
125 builder.getAffineDimExpr(new_dim)); in TileLoop()
127 replacements.push_back(builder.getAffineDimExpr(i)); in TileLoop()
|
| D | conv_emitter.cc | 96 affine_exprs.push_back(builder.getAffineDimExpr(dim)); in GetShapeInfo()
296 input_indices.push_back(builder.getAffineDimExpr(0)); in CreateNaiveMlirConv()
297 input_indices.push_back(builder.getAffineDimExpr(1)); in CreateNaiveMlirConv()
306 builder.getAffineDimExpr(i + 2) * window_dim.stride() + in CreateNaiveMlirConv()
307 builder.getAffineDimExpr(2 + num_spatial_dims + i) - in CreateNaiveMlirConv()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 297 AffineExpr Builder::getAffineDimExpr(unsigned position) { in getAffineDimExpr() function in Builder
298 return mlir::getAffineDimExpr(position, context); in getAffineDimExpr()
317 return AffineMap::get(/*dimCount=*/1, /*symbolCount=*/0, getAffineDimExpr(0)); in getDimIdentityMap()
324 dimExprs.push_back(getAffineDimExpr(i)); in getMultiDimIdentityMap()
336 auto expr = getAffineDimExpr(0) + shift; in getSingleDimShiftAffineMap()
|
| D | AffineMap.cpp | 120 affExprs.push_back(getAffineDimExpr(index, context)); in getPermutationMap()
173 dimExprs.push_back(mlir::getAffineDimExpr(i, context)); in getMultiDimIdentityMap()
320 newDims[idx] = getAffineDimExpr(idx, getContext()); in compose()
428 exprs[d.getPosition()] = getAffineDimExpr(en.index(), d.getContext()); in inversePermutation()
466 resultExprs.push_back(getAffineDimExpr(dim.index(), context)); in getProjectedMap()
|
| D | AffineExpr.cpp | 255 return *this == mlir::getAffineDimExpr(position, getContext()); in isFunctionOfDim()
453 AffineExpr mlir::getAffineDimExpr(unsigned position, MLIRContext *context) { in getAffineDimExpr() function in mlir
840 auto id = j < numDims ? getAffineDimExpr(j, context) in getAffineExprFromFlatForm()
|
| D | BuiltinTypes.cpp | 704 auto d = getAffineDimExpr(dim, context); in makeStridedLinearLayoutMap()
792 exprs.push_back(getAffineDimExpr(dim, context)); in makeCanonicalStridedLayoutExpr()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 425 input_dim_exprs.push_back(b->getAffineDimExpr(num_prepended_dims + i)); in getIndexingMaps()
473 : b->getAffineDimExpr(size)); in getIndexingMaps()
682 dim_exprs.push_back(b->getAffineDimExpr(size)); in getIndexingMaps()
705 b->getAffineDimExpr(permutation.index()); in getIndexingMaps()
755 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
760 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
768 rewriter.getAffineDimExpr(curr_src_dim++)); in matchAndRewrite()
786 exprs.push_back(rewriter.getAffineDimExpr(i)); in matchAndRewrite()
953 src_exprs.push_back(mlir::getAffineDimExpr(i, rewriter.getContext())); in matchAndRewrite()
955 dst_exprs.push_back(mlir::getAffineDimExpr(i, rewriter.getContext())); in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| D | AffineLoopNormalize.cpp | 56 auto expr = lbExpr + builder.getAffineDimExpr(nDims) * step; in normalizeAffineParallel()
173 AffineExpr origIVExpr = opBuilder.getAffineDimExpr(lb.getMap().getNumDims()); in normalizeAffineFor()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | AffineExpr.h | 230 AffineExpr getAffineDimExpr(unsigned position, MLIRContext *context);
283 e = getAffineDimExpr(N, ctx); in bindDims()
|
| D | Builders.h | 146 AffineExpr getAffineDimExpr(unsigned position);
|
| /external/llvm-project/mlir/unittests/SDBM/ |
| D | SDBMTest.cpp | 424 auto dim = getAffineDimExpr(0, ctx()); in TEST()
433 auto d0 = getAffineDimExpr(0, ctx()); in TEST()
434 auto d1 = getAffineDimExpr(1, ctx()); in TEST()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 102 newUbExprs[i] = b.getAffineDimExpr(0) + b.getAffineDimExpr(i + 1); in getCleanupLoopLowerBound()
638 AffineExpr lbLoopIvExpr = b.getAffineDimExpr(lbOperands.size() - 1); in setIntraTileBoundsParametric()
639 AffineExpr ubLoopIvExpr = b.getAffineDimExpr(ubOperands.size() - 1); in setIntraTileBoundsParametric()
879 AffineExpr dim = b.getAffineDimExpr(origUbMap.getNumDims()); in constructTiledIndexSetHyperRect()
891 AffineExpr dim = b.getAffineDimExpr(0); in constructTiledIndexSetHyperRect()
1150 auto d0 = b.getAffineDimExpr(0); in loopUnrollByFactor()
1385 auto d0 = builder.getAffineDimExpr(0); in loopUnrollJamByFactor()
1633 bounds.push_back(b.getAffineDimExpr(map->getNumDims()) + offset); in augmentMapAndBounds()
2254 fastBufExprs.push_back(b.getAffineDimExpr(2 * d + 1) - in generatePointWiseCopy()
2255 b.getAffineDimExpr(2 * d)); in generatePointWiseCopy()
[all …]
|
| /external/llvm-project/mlir/lib/CAPI/IR/ |
| D | AffineExpr.cpp | 60 return wrap(getAffineDimExpr(position, unwrap(ctx))); in mlirAffineDimExprGet()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
| D | LinalgToSPIRV.cpp | 101 AffineMap::get(1, 0, getAffineDimExpr(0, op->getContext()))) in matchAsPerformingReduction()
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | AffineParser.cpp | 304 : getAffineDimExpr(numDimOperands++, getContext()); in parseSSAIdExpr()
482 auto dimension = getAffineDimExpr(numDims++, getContext()); in parseDimIdList()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 428 rewriter.getAffineDimExpr(0) * rewriter.getAffineSymbolExpr(0) + in processParallelLoop()
472 ((rewriter.getAffineDimExpr(0) - rewriter.getAffineSymbolExpr(0)) in processParallelLoop()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | PipelineDataTransfer.cpp | 99 auto d0 = bInner.getAffineDimExpr(0); in doubleBuffer()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorUtils.cpp | 196 perm[kvp.second] = getAffineDimExpr(dim, context); in makePermutationMap()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestVectorTransforms.cpp | 206 perm.push_back(getAffineDimExpr(i, ctx)); in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 593 reassociations.push_back(getAffineDimExpr(currDim++, context)); in collapseReassociationMaps()
830 reassociationMap.push_back(b.getAffineDimExpr(index)); in convertReassociationIndicesToMaps()
1335 res.push_back(getAffineDimExpr(dim, ctx)); in getDimsOfType()
1356 res.push_back(getAffineDimExpr(startIdx++, context)); in makeAffineDimExprs()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/EDSC/ |
| D | Builders.cpp | 97 d = getAffineDimExpr(numDims++, context); in categorizeValueByAffineType()
|