Home
last modified time | relevance | path

Searched defs:dimIndex (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DResolveShapedTypeResultDims.cpp51 std::optional<int64_t> dimIndex = dimOp.getConstantIndex(); matchAndRewrite() local
88 std::optional<int64_t> dimIndex = dimOp.getConstantIndex(); matchAndRewrite() local
/llvm-project/flang/lib/Evaluate/
H A Dfold-reduction.cpp13 ActualArguments &arg, std::optional<int> dimIndex, int rank) { in CheckReductionDIM()
H A Dshape.cpp918 std::size_t dimIndex{intrinsic->name == "findloc" ? 2u : 1u}; operator ()() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp42 getCollapsedOutputDimFromInputShape(OpBuilder & builder,Location loc,int64_t dimIndex,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociationMap) getCollapsedOutputDimFromInputShape() argument
82 getExpandedOutputDimFromInputShape(OpBuilder & builder,Location loc,int64_t dimIndex,Value src,ArrayRef<int64_t> dstStaticShape,ArrayRef<AffineMap> reassociation,llvm::DenseMap<int64_t,int64_t> & expandedDimToCollapsedDim) getExpandedOutputDimFromInputShape() argument
H A DTensorOps.cpp493 size_t dimIndex = 0; matchAndRewrite() local
/llvm-project/flang/lib/Optimizer/Builder/Runtime/
H A DAllocatable.cpp60 mlir::Value dimIndex, in genAllocatableSetBounds()
/llvm-project/flang/lib/Lower/
H A DAllocatable.cpp130 genRuntimeSetBounds(fir::FirOpBuilder & builder,mlir::Location loc,const fir::MutableBoxValue & box,mlir::Value dimIndex,mlir::Value lowerBound,mlir::Value upperBound) genRuntimeSetBounds() argument
561 mlir::Value dimIndex = genAllocateObjectBounds() local
582 mlir::Value dimIndex = builder.createIntegerConstant(loc, i32Ty, i); genAllocateObjectBounds() local
/llvm-project/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp356 std::optional<int64_t> dimIndex; in getUniqueNonUnitDim() local
/llvm-project/flang/lib/Optimizer/HLFIR/Transforms/
H A DBufferizeHLFIR.cpp137 mlir::Value dimIndex = builder.createIntegerConstant(loc, idxTy, dim++); createArrayTemp() local