Lines Matching refs:create
80 Value one = lb.create<arith::ConstantIndexOp>(1); in getBroadcastedDim()
85 Value outOfBounds = lb.create<arith::CmpIOp>(arith::CmpIPredicate::ult, in getBroadcastedDim()
89 lb.create<IfOp>( in getBroadcastedDim()
92 b.create<scf::YieldOp>(loc, broadcastedDim); in getBroadcastedDim()
102 Value lesserRankOperandDimension = b.create<arith::SubIOp>( in getBroadcastedDim()
104 Value lesserRankOperandExtent = b.create<tensor::ExtractOp>( in getBroadcastedDim()
108 b.create<arith::CmpIOp>(loc, arith::CmpIPredicate::eq, in getBroadcastedDim()
110 Value dim = b.create<arith::SelectOp>( in getBroadcastedDim()
112 b.create<scf::YieldOp>(loc, dim); in getBroadcastedDim()
131 Value zero = lb.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
139 return lb.create<tensor::DimOp>(v, zero); in matchAndRewrite()
146 lb.create<arith::CmpIOp>(arith::CmpIPredicate::ugt, v, maxRank); in matchAndRewrite()
147 maxRank = lb.create<arith::SelectOp>(rankIsGreater, v, maxRank); in matchAndRewrite()
152 return lb.create<arith::SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
155 Value replacement = lb.create<tensor::GenerateOp>( in matchAndRewrite()
162 b.create<tensor::YieldOp>(loc, broadcastedDim); in matchAndRewrite()
165 replacement = lb.create<tensor::CastOp>(op.getType(), replacement); in matchAndRewrite()
194 rewriter.create<arith::ConstantIndexOp>(loc, extent.getLimitedValue())); in matchAndRewrite()
199 rewriter.create<tensor::FromElementsOp>(loc, resultTy, extentOperands); in matchAndRewrite()
245 Value zero = lb.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
246 Value one = lb.create<arith::ConstantIndexOp>(1); in matchAndRewrite()
254 return lb.create<tensor::DimOp>(v, zero); in matchAndRewrite()
261 lb.create<arith::CmpIOp>(arith::CmpIPredicate::ugt, v, maxRank); in matchAndRewrite()
262 maxRank = lb.create<arith::SelectOp>(rankIsGreater, v, maxRank); in matchAndRewrite()
267 return lb.create<arith::SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
272 rewriter.create<arith::ConstantOp>(loc, i1Ty, rewriter.getBoolAttr(true)); in matchAndRewrite()
274 auto reduceResult = lb.create<ForOp>( in matchAndRewrite()
287 Value outOfBounds = b.create<arith::CmpIOp>( in matchAndRewrite()
290 b.create<IfOp>( in matchAndRewrite()
294 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
300 b.create<arith::SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
301 Value dimensionExtent = b.create<tensor::ExtractOp>( in matchAndRewrite()
304 Value equalOne = b.create<arith::CmpIOp>( in matchAndRewrite()
306 Value equalBroadcasted = b.create<arith::CmpIOp>( in matchAndRewrite()
309 Value result = b.create<arith::AndIOp>( in matchAndRewrite()
311 b.create<arith::OrIOp>(loc, equalOne, in matchAndRewrite()
313 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
318 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
401 Value zero = rewriter.create<arith::ConstantIndexOp>(loc, 0); in matchAndRewrite()
402 Value one = rewriter.create<arith::ConstantIndexOp>(loc, 1); in matchAndRewrite()
405 rewriter.create<tensor::DimOp>(loc, indexTy, adaptor.getShape(), zero); in matchAndRewrite()
407 auto loop = rewriter.create<scf::ForOp>( in matchAndRewrite()
410 Value extent = b.create<tensor::ExtractOp>(loc, adaptor.getShape(), iv); in matchAndRewrite()
424 b.create<scf::YieldOp>(loc, mappedResults); in matchAndRewrite()
487 Value zero = rewriter.create<arith::ConstantIndexOp>(loc, 0); in matchAndRewrite()
490 rewriter.create<tensor::DimOp>(loc, indexTy, firstShape, zero); in matchAndRewrite()
494 Value rank = rewriter.create<tensor::DimOp>(loc, indexTy, shape, zero); in matchAndRewrite()
495 Value eqRank = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::eq, in matchAndRewrite()
497 auto same = rewriter.create<IfOp>( in matchAndRewrite()
500 Value one = b.create<arith::ConstantIndexOp>(loc, 1); in matchAndRewrite()
502 b.create<arith::ConstantOp>(loc, i1Ty, b.getBoolAttr(true)); in matchAndRewrite()
503 auto loop = b.create<scf::ForOp>( in matchAndRewrite()
508 b.create<tensor::ExtractOp>(loc, firstShape, iv); in matchAndRewrite()
509 Value rhsExtent = b.create<tensor::ExtractOp>(loc, shape, iv); in matchAndRewrite()
510 Value eqExtent = b.create<arith::CmpIOp>( in matchAndRewrite()
512 Value conjNext = b.create<arith::AndIOp>(loc, conj, eqExtent); in matchAndRewrite()
513 b.create<scf::YieldOp>(loc, ValueRange({conjNext})); in matchAndRewrite()
515 b.create<scf::YieldOp>(loc, loop.getResults()); in matchAndRewrite()
519 b.create<arith::ConstantOp>(loc, i1Ty, b.getBoolAttr(false)); in matchAndRewrite()
520 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
523 : rewriter.create<arith::AndIOp>(loc, result, in matchAndRewrite()
561 Value extent = rewriter.create<tensor::DimOp>(loc, tensor, i); in matchAndRewrite()
564 Value extent = rewriter.create<arith::ConstantIndexOp>( in matchAndRewrite()
571 Value staticExtentTensor = rewriter.create<tensor::FromElementsOp>( in matchAndRewrite()
581 Value rank = rewriter.create<tensor::RankOp>(loc, tensor); in matchAndRewrite()
586 Value extent = b.create<tensor::DimOp>(loc, tensor, dim); in matchAndRewrite()
587 b.create<tensor::YieldOp>(loc, extent); in matchAndRewrite()
614 Value zero = b.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
615 Value rank = b.create<tensor::DimOp>(adaptor.getOperand(), zero); in matchAndRewrite()
619 Value add = b.create<arith::AddIOp>(originalIndex, rank); in matchAndRewrite()
621 b.create<arith::CmpIOp>(arith::CmpIPredicate::slt, originalIndex, zero); in matchAndRewrite()
622 Value index = b.create<arith::SelectOp>(indexIsNegative, add, originalIndex); in matchAndRewrite()
624 Value one = b.create<arith::ConstantIndexOp>(1); in matchAndRewrite()
626 b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), zero, index, one); in matchAndRewrite()
627 Value tailSize = b.create<arith::SubIOp>(rank, index); in matchAndRewrite()
628 Value tail = b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), index, in matchAndRewrite()