Lines Matching refs:padOp
425 auto padOp = opOperand->get().getDefiningOp<tensor::PadOp>(); in returningMatchAndRewrite() local
426 if (!padOp || en.value() == 0) in returningMatchAndRewrite()
441 padOp, en.value(), transposeVector, hoistedOp, transposeOps); in returningMatchAndRewrite()
444 rewriter.replaceOp(padOp, *newResult); in returningMatchAndRewrite()
672 PadOpTransformationPattern::matchAndRewrite(tensor::PadOp padOp, in matchAndRewrite() argument
675 auto inputShapedType = padOp.getSource().getType().cast<ShapedType>(); in matchAndRewrite()
676 auto resultShapedType = padOp.getResult().getType().cast<ShapedType>(); in matchAndRewrite()
687 Block &block = padOp.getRegion().front(); in matchAndRewrite()
697 Location loc = padOp.getLoc(); in matchAndRewrite()
714 padOp.getStaticLow()[i].cast<IntegerAttr>().getInt()); in matchAndRewrite()
723 padOp, resultShapedType, padOp.getSource(), tmpTensor, transferMaps, in matchAndRewrite()
735 PatternRewriter &rewriter, tensor::PadOp padOp, Value dest, in createFillOrGenerateOp() argument
737 auto padValue = padOp.getConstantPaddingValue(); in createFillOrGenerateOp()
739 return rewriter.create<FillOp>(padOp.getLoc(), padValue, dest).result(); in createFillOrGenerateOp()
743 padOp.getLoc(), padOp.getResultType(), dynSizes); in createFillOrGenerateOp()
746 padOp.getRegion().cloneInto(&generateOp.getRegion(), bvm); in createFillOrGenerateOp()
751 GeneralizePadOpPattern::matchAndRewrite(tensor::PadOp padOp, in matchAndRewrite() argument
759 padOp.getLoc(), ofr.get<Attribute>().cast<IntegerAttr>().getInt()) in matchAndRewrite()
763 auto resultType = padOp.getResultType(); in matchAndRewrite()
771 padOp.getLoc(), padOp.getSource(), dim); in matchAndRewrite()
774 padOp.getLoc(), srcSize, getIdxValue(padOp.getMixedLowPad()[dim])); in matchAndRewrite()
776 padOp.getLoc(), plusLow, getIdxValue(padOp.getMixedHighPad()[dim])); in matchAndRewrite()
784 padOp.getLoc(), dynSizes, staticSizes, resultType.getElementType()); in matchAndRewrite()
785 Value fill = createFillOrGenerateOp(rewriter, padOp, init, dynSizes); in matchAndRewrite()
788 if (optimizeCopyFn && optimizeCopyFn(rewriter, padOp, fill).succeeded()) in matchAndRewrite()
793 auto sourceType = padOp.getSourceType(); in matchAndRewrite()
799 padOp.getLoc(), padOp.getSource(), dim)); in matchAndRewrite()
808 padOp, padOp.getSource(), fill, padOp.getMixedLowPad(), srcSizes, in matchAndRewrite()
819 auto padOp = sliceOp.getSource().getDefiningOp<tensor::PadOp>(); in matchAndRewrite() local
820 if (!padOp) in matchAndRewrite()
832 tensor::bubbleUpPadSlice(rewriter, padOp, sliceOp.getMixedOffsets(), in matchAndRewrite()