Lines Matching defs:padOp

276   auto padOp =
304         isRankReducedType(packedTensorType, padOp.getResultType());
319           loc, /*source=*/padOp, /*dest=*/packOp.getDest(),
326       return LowerPackResult{padOp, /*reshapeOp=*/nullptr,
335       loc, expandShapeResultType, padOp.getResult(),
352   return LowerPackResult{padOp, reshapeOp, transposeOp};
927     RewriterBase &rewriter, tensor::PadOp padOp, Value dest,
929   auto padValue = padOp.getConstantPaddingValue();
931     return rewriter.create<FillOp>(padOp.getLoc(), padValue, dest).result();
935       padOp.getLoc(), padOp.getResultType(), dynSizes);
938   padOp.getRegion().cloneInto(&generateOp.getRegion(), bvm);
943 DecomposePadOpPattern::matchAndRewrite(tensor::PadOp padOp,
951             padOp.getLoc(), cast<IntegerAttr>(cast<Attribute>(ofr)).getInt())
955   auto resultType = padOp.getResultType();
961       auto srcSize = getIdxValue(tensor::getMixedSize(rewriter, padOp.getLoc(),
962                                                       padOp.getSource(), dim));
965           padOp.getLoc(), srcSize, getIdxValue(padOp.getMixedLowPad()[dim]));
967           padOp.getLoc(), plusLow, getIdxValue(padOp.getMixedHighPad()[dim]));
975       padOp.getLoc(), staticSizes, resultType.getElementType(), dynSizes);
976   Value fill = createFillOrGenerateOp(rewriter, padOp, emptyTensor, dynSizes);
979   auto sourceType = padOp.getSourceType();
982       tensor::getMixedSizes(rewriter, padOp.getLoc(), padOp.getSource());
987       padOp, padOp.getSource(), fill, padOp.getMixedLowPad(), srcSizes,
998   auto padOp = sliceOp.getSource().getDefiningOp<tensor::PadOp>();
999   if (!padOp)
1011       tensor::bubbleUpPadSlice(rewriter, padOp, sliceOp.getMixedOffsets(),