| /llvm-project-15.0.7/flang/lib/Evaluate/ |
| H A D | fold-reduction.cpp | 13 ActualArguments &arg, std::optional<int> dimIndex, int rank) { in CheckReductionDIM() argument
14 if (dimIndex && static_cast<std::size_t>(*dimIndex) < arg.size()) { in CheckReductionDIM()
16 Folder<SubscriptInteger>{context}.Folding(arg[*dimIndex])}) { in CheckReductionDIM()
|
| H A D | fold-reduction.h | 20 ActualArguments &, std::optional<int> dimIndex, int rank);
36 int arrayIndex, std::optional<int> dimIndex = std::nullopt,
45 if (!CheckReductionDIM(dim, context, arg, dimIndex, folded->Rank())) {
|
| H A D | fold-implementation.h | 618 ConstantSubscript &dimIndex{arrayAt[zbDim]}; in CSHIFT()
623 ConstantSubscript origDimIndex{dimIndex}; in CSHIFT()
634 dimIndex = dimLB + ((dimIndex - dimLB + shiftCount) % dimExtent); in CSHIFT()
635 if (dimIndex < dimLB) { in CSHIFT()
636 dimIndex += dimExtent; in CSHIFT()
638 dimIndex -= dimExtent; in CSHIFT()
641 dimIndex = origDimIndex; in CSHIFT()
735 ConstantSubscript origDimIndex{dimIndex}; in EOSHIFT()
746 dimIndex += shiftCount; in EOSHIFT()
747 if (dimIndex >= dimLB && dimIndex < dimLB + dimExtent) { in EOSHIFT()
[all …]
|
| H A D | shape.cpp | 811 std::size_t dimIndex{intrinsic->name == "findloc" ? 2u : 1u}; in operator ()() local
812 if (call.arguments().size() > dimIndex) { in operator ()()
817 UnwrapExpr<Expr<SomeType>>(call.arguments()[dimIndex])}) { in operator ()()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | ResolveShapedTypeResultDims.cpp | 41 Optional<int64_t> dimIndex = dimOp.getConstantIndex(); in matchAndRewrite() local
42 if (!dimIndex) in matchAndRewrite()
61 rewriter.createOrFold<arith::ConstantIndexOp>(loc, *dimIndex)); in matchAndRewrite()
81 Optional<int64_t> dimIndex = dimOp.getConstantIndex(); in matchAndRewrite() local
82 if (!dimIndex) in matchAndRewrite()
99 rewriter.replaceOp(dimOp, reifiedResultShapes[resultNumber][*dimIndex]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorInferTypeOpInterfaceImpl.cpp | 42 int64_t dimIndex, Value src, in getCollapsedOutputDimFromInputShape() argument
44 AffineMap map = reassociationMap[dimIndex]; in getCollapsedOutputDimFromInputShape()
75 OpBuilder &builder, Location loc, int64_t dimIndex, Value src, in getExpandedOutputDimFromInputShape() argument
78 if (!ShapedType::isDynamic(dstStaticShape[dimIndex])) { in getExpandedOutputDimFromInputShape()
79 return builder.getI64IntegerAttr(dstStaticShape[dimIndex]); in getExpandedOutputDimFromInputShape()
81 unsigned sourceDimPos = expandedDimToCollapsedDim[dimIndex]; in getExpandedOutputDimFromInputShape()
95 if (d.index() + startPos == static_cast<unsigned>(dimIndex)) in getExpandedOutputDimFromInputShape()
|
| H A D | TensorOps.cpp | 263 size_t dimIndex = 0; in matchAndRewrite() local
267 int64_t dim = tensorCast.getType().getShape()[dimIndex++]; in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | Allocatable.cpp | 120 mlir::Value dimIndex, mlir::Value lowerBound, in genRuntimeSetBounds() argument
128 llvm::SmallVector<mlir::Value> args{box.getAddr(), dimIndex, lowerBound, in genRuntimeSetBounds()
413 mlir::Value dimIndex = in genSimpleAllocation() local
416 genRuntimeSetBounds(builder, loc, box, dimIndex, lb, ub); in genSimpleAllocation()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1616 for (auto dimIndex : in matchAndRewrite() local
1618 dstDesc.setStride(rewriter, loc, dimIndex, stride); in matchAndRewrite()
1619 stride = rewriter.create<LLVM::MulOp>(loc, dstShape[dimIndex], stride); in matchAndRewrite()
|