Lines Matching full:create
85 Value one = lb.create<arith::ConstantIndexOp>(1); in getBroadcastedDim()
90 Value outOfBounds = lb.create<arith::CmpIOp>(arith::CmpIPredicate::ult, in getBroadcastedDim()
94 lb.create<IfOp>( in getBroadcastedDim()
97 b.create<scf::YieldOp>(loc, broadcastedDim); in getBroadcastedDim()
107 Value lesserRankOperandDimension = b.create<arith::SubIOp>( in getBroadcastedDim()
109 Value lesserRankOperandExtent = b.create<tensor::ExtractOp>( in getBroadcastedDim()
113 b.create<arith::CmpIOp>(loc, arith::CmpIPredicate::eq, in getBroadcastedDim()
115 Value dim = b.create<arith::SelectOp>( in getBroadcastedDim()
117 b.create<scf::YieldOp>(loc, dim); in getBroadcastedDim()
136 Value zero = lb.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
144 return lb.create<tensor::DimOp>(v, zero); in matchAndRewrite()
150 maxRank = lb.create<arith::MaxUIOp>(v, maxRank); in matchAndRewrite()
155 return lb.create<arith::SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
158 Value replacement = lb.create<tensor::GenerateOp>( in matchAndRewrite()
165 b.create<tensor::YieldOp>(loc, broadcastedDim); in matchAndRewrite()
168 replacement = lb.create<tensor::CastOp>(op.getType(), replacement); in matchAndRewrite()
197 rewriter.create<arith::ConstantIndexOp>(loc, extent.getLimitedValue())); in matchAndRewrite()
202 rewriter.create<tensor::FromElementsOp>(loc, resultTy, extentOperands); in matchAndRewrite()
248 Value zero = lb.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
249 Value one = lb.create<arith::ConstantIndexOp>(1); in matchAndRewrite()
257 return lb.create<tensor::DimOp>(v, zero); in matchAndRewrite()
263 maxRank = lb.create<arith::MaxUIOp>(v, maxRank); in matchAndRewrite()
268 return lb.create<arith::SubIOp>(indexTy, maxRank, v); in matchAndRewrite()
273 rewriter.create<arith::ConstantOp>(loc, i1Ty, rewriter.getBoolAttr(true)); in matchAndRewrite()
275 auto reduceResult = lb.create<ForOp>( in matchAndRewrite()
288 Value outOfBounds = b.create<arith::CmpIOp>( in matchAndRewrite()
291 b.create<IfOp>( in matchAndRewrite()
295 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
301 b.create<arith::SubIOp>(loc, indexTy, iv, rankDiff); in matchAndRewrite()
302 Value dimensionExtent = b.create<tensor::ExtractOp>( in matchAndRewrite()
305 Value equalOne = b.create<arith::CmpIOp>( in matchAndRewrite()
307 Value equalBroadcasted = b.create<arith::CmpIOp>( in matchAndRewrite()
310 Value result = b.create<arith::AndIOp>( in matchAndRewrite()
312 b.create<arith::OrIOp>(loc, equalOne, in matchAndRewrite()
314 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
319 b.create<scf::YieldOp>(loc, broadcastable); in matchAndRewrite()
342 auto shapeOf = rewriter.create<shape::ShapeOfOp>(op.getLoc(), op.getValue()); in matchAndRewrite()
424 Value zero = rewriter.create<arith::ConstantIndexOp>(loc, 0); in matchAndRewrite()
425 Value one = rewriter.create<arith::ConstantIndexOp>(loc, 1); in matchAndRewrite()
428 rewriter.create<tensor::DimOp>(loc, indexTy, adaptor.getShape(), zero); in matchAndRewrite()
430 auto loop = rewriter.create<scf::ForOp>( in matchAndRewrite()
433 Value extent = b.create<tensor::ExtractOp>(loc, adaptor.getShape(), iv); in matchAndRewrite()
447 b.create<scf::YieldOp>(loc, mappedResults); in matchAndRewrite()
510 Value zero = rewriter.create<arith::ConstantIndexOp>(loc, 0); in matchAndRewrite()
513 rewriter.create<tensor::DimOp>(loc, indexTy, firstShape, zero); in matchAndRewrite()
517 Value rank = rewriter.create<tensor::DimOp>(loc, indexTy, shape, zero); in matchAndRewrite()
518 Value eqRank = rewriter.create<arith::CmpIOp>(loc, arith::CmpIPredicate::eq, in matchAndRewrite()
520 auto same = rewriter.create<IfOp>( in matchAndRewrite()
523 Value one = b.create<arith::ConstantIndexOp>(loc, 1); in matchAndRewrite()
525 b.create<arith::ConstantOp>(loc, i1Ty, b.getBoolAttr(true)); in matchAndRewrite()
526 auto loop = b.create<scf::ForOp>( in matchAndRewrite()
531 b.create<tensor::ExtractOp>(loc, firstShape, iv); in matchAndRewrite()
532 Value rhsExtent = b.create<tensor::ExtractOp>(loc, shape, iv); in matchAndRewrite()
533 Value eqExtent = b.create<arith::CmpIOp>( in matchAndRewrite()
535 Value conjNext = b.create<arith::AndIOp>(loc, conj, eqExtent); in matchAndRewrite()
536 b.create<scf::YieldOp>(loc, ValueRange({conjNext})); in matchAndRewrite()
538 b.create<scf::YieldOp>(loc, loop.getResults()); in matchAndRewrite()
542 b.create<arith::ConstantOp>(loc, i1Ty, b.getBoolAttr(false)); in matchAndRewrite()
543 b.create<scf::YieldOp>(loc, result); in matchAndRewrite()
546 : rewriter.create<arith::AndIOp>(loc, result, in matchAndRewrite()
584 Value extent = rewriter.create<tensor::DimOp>(loc, tensor, i); in matchAndRewrite()
587 Value extent = rewriter.create<arith::ConstantIndexOp>( in matchAndRewrite()
594 Value staticExtentTensor = rewriter.create<tensor::FromElementsOp>( in matchAndRewrite()
604 Value rank = rewriter.create<tensor::RankOp>(loc, tensor); in matchAndRewrite()
609 Value extent = b.create<tensor::DimOp>(loc, tensor, dim); in matchAndRewrite()
610 b.create<tensor::YieldOp>(loc, extent); in matchAndRewrite()
637 Value zero = b.create<arith::ConstantIndexOp>(0); in matchAndRewrite()
638 Value rank = b.create<tensor::DimOp>(adaptor.getOperand(), zero); in matchAndRewrite()
642 Value add = b.create<arith::AddIOp>(originalIndex, rank); in matchAndRewrite()
644 b.create<arith::CmpIOp>(arith::CmpIPredicate::slt, originalIndex, zero); in matchAndRewrite()
645 Value index = b.create<arith::SelectOp>(indexIsNegative, add, originalIndex); in matchAndRewrite()
647 Value one = b.create<arith::ConstantIndexOp>(1); in matchAndRewrite()
649 b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), zero, index, one); in matchAndRewrite()
650 Value tailSize = b.create<arith::SubIOp>(rank, index); in matchAndRewrite()
651 Value tail = b.create<tensor::ExtractSliceOp>(adaptor.getOperand(), index, in matchAndRewrite()