Lines Matching refs:forOp

36     AffineForOp forOp, AffineMap *tripCountMap,  in getTripCountMapAndOperands()  argument
38 MLIRContext *context = forOp.getContext(); in getTripCountMapAndOperands()
39 int64_t step = forOp.getStep(); in getTripCountMapAndOperands()
41 if (forOp.hasConstantBounds()) { in getTripCountMapAndOperands()
42 int64_t lb = forOp.getConstantLowerBound(); in getTripCountMapAndOperands()
43 int64_t ub = forOp.getConstantUpperBound(); in getTripCountMapAndOperands()
51 auto lbMap = forOp.getLowerBoundMap(); in getTripCountMapAndOperands()
52 auto ubMap = forOp.getUpperBoundMap(); in getTripCountMapAndOperands()
61 AffineValueMap ubValueMap(ubMap, forOp.getUpperBoundOperands()); in getTripCountMapAndOperands()
67 AffineValueMap lbSplatValueMap(lbMapSplat, forOp.getLowerBoundOperands()); in getTripCountMapAndOperands()
83 Optional<uint64_t> mlir::getConstantTripCount(AffineForOp forOp) { in getConstantTripCount() argument
86 getTripCountMapAndOperands(forOp, &map, &operands); in getConstantTripCount()
109 uint64_t mlir::getLargestDivisorOfTripCount(AffineForOp forOp) { in getLargestDivisorOfTripCount() argument
112 getTripCountMapAndOperands(forOp, &map, &operands); in getLargestDivisorOfTripCount()
269 auto *forOp = loop.getOperation(); in isVectorizableLoopBodyWithOpCond() local
274 conditionals.match(forOp, &conditionalsMatched); in isVectorizableLoopBodyWithOpCond()
284 regions.match(forOp, &regionsMatched); in isVectorizableLoopBodyWithOpCond()
290 vectorTransferMatcher.match(forOp, &vectorTransfersMatched); in isVectorizableLoopBodyWithOpCond()
297 loadAndStores.match(forOp, &loadAndStoresMatched); in isVectorizableLoopBodyWithOpCond()
349 bool mlir::isOpwiseShiftValid(AffineForOp forOp, ArrayRef<uint64_t> shifts) { in isOpwiseShiftValid() argument
350 auto *forBody = forOp.getBody(); in isOpwiseShiftValid()