Lines Matching refs:indexTy
90 Type indexTy = lb.getIndexType(); in getBroadcastedDim() local
93 TypeRange{indexTy}, outOfBounds, in getBroadcastedDim()
106 b.create<SubIOp>(loc, indexTy, outputDimension, rankDiff); in getBroadcastedDim()
135 Type indexTy = lb.getIndexType(); in matchAndRewrite() local
154 return lb.create<SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
198 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
200 rewriter.create<tensor::FromElementsOp>(loc, indexTy, extentOperands); in matchAndRewrite()
201 Type resultTy = RankedTensorType::get({ShapedType::kDynamicSize}, indexTy); in matchAndRewrite()
249 Type indexTy = lb.getIndexType(); in matchAndRewrite() local
268 return lb.create<SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
301 b.create<SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
407 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
409 rewriter.create<tensor::DimOp>(loc, indexTy, transformed.shape(), zero); in matchAndRewrite()
492 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
496 rewriter.create<tensor::DimOp>(loc, indexTy, firstShape, zero); in matchAndRewrite()
500 Value rank = rewriter.create<tensor::DimOp>(loc, indexTy, shape, zero); in matchAndRewrite()