Lines Matching refs:indexTy
97 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
103 RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
114 rewriter.create<SubIOp>(loc, indexTy, greaterRank, lesserRank); in matchAndRewrite()
129 loc, TypeRange{indexTy}, isUnchallengedDimension, in matchAndRewrite()
142 b.create<SubIOp>(loc, indexTy, outputDimension, rankDiff); in matchAndRewrite()
184 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
186 rewriter.create<TensorFromElementsOp>(loc, indexTy, extentOperands); in matchAndRewrite()
187 Type resultTy = RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
240 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
246 RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
256 rewriter.create<SubIOp>(loc, indexTy, greaterRank, lesserRank); in matchAndRewrite()
269 Value ivShifted = b.create<SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
372 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
373 Value rank = rewriter.create<DimOp>(loc, indexTy, transformed.shape(), zero); in matchAndRewrite()
451 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
453 Value lhsRank = rewriter.create<DimOp>(loc, indexTy, transformed.lhs(), zero); in matchAndRewrite()
454 Value rhsRank = rewriter.create<DimOp>(loc, indexTy, transformed.rhs(), zero); in matchAndRewrite()