Lines Matching refs:indexTy
85 Type indexTy = lb.getIndexType(); in getBroadcastedDim() local
88 TypeRange{indexTy}, outOfBounds, in getBroadcastedDim()
101 b.create<SubIOp>(loc, indexTy, outputDimension, rankDiff); in getBroadcastedDim()
129 Type indexTy = lb.getIndexType(); in matchAndRewrite() local
148 return lb.create<SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
191 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
193 rewriter.create<tensor::FromElementsOp>(loc, indexTy, extentOperands); in matchAndRewrite()
194 Type resultTy = RankedTensorType::get({op.shape().size()}, indexTy); in matchAndRewrite()
241 Type indexTy = lb.getIndexType(); in matchAndRewrite() local
260 return lb.create<SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
293 b.create<SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
394 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
396 rewriter.create<tensor::DimOp>(loc, indexTy, adaptor.shape(), zero); in matchAndRewrite()
477 Type indexTy = rewriter.getIndexType(); in matchAndRewrite() local
481 rewriter.create<tensor::DimOp>(loc, indexTy, firstShape, zero); in matchAndRewrite()
485 Value rank = rewriter.create<tensor::DimOp>(loc, indexTy, shape, zero); in matchAndRewrite()