Lines Matching defs:rtp
682 Type rtp = s0.getType();
684 auto semiring = rewriter.create<sparse_tensor::UnaryOp>(loc, rtp, s0);
686 rewriter.createBlock(&semiring.getPresentRegion(), {}, rtp, loc);
692 rewriter.create<arith::ConstantOp>(loc, rewriter.getZeroAttr(rtp));
699 loc, rtp, semiring.getResult(), s1, identity);
701 rewriter.createBlock(&custom.getRegion(), {}, {rtp, rtp}, {loc, loc});
1079 auto rtp = getRankedTensorType(op.getSrc());
1081 RankedTensorType::get(rtp.getShape(), rtp.getElementType());
1087 auto rtp = getRankedTensorType(op.getResult());
1089 RankedTensorType::get(rtp.getShape(), rtp.getElementType());
1099 Value convert = rewriter.create<ConvertOp>(loc, rtp, reshape);
1126 Value finalize(OpBuilder &builder, Location loc, RankedTensorType rtp) const {