Lines Matching refs:sliceOp

208   auto sliceOp = currOpOperand->get().getDefiningOp<tensor::ExtractSliceOp>();
209   if (!sliceOp)
212   // Compute the dropped dimensions if `sliceOp` is ranke-reducing.
213   llvm::SmallBitVector droppedDims = sliceOp.getDroppedDims();
214   OffsetSizeAndStrideOpInterface shapedOp = sliceOp;
216   // Upper bound the `sliceOp` sizes to obtain a static bounding box.
815     tensor::ExtractSliceOp sliceOp, PatternRewriter &rewriter) const {
816   if (!sliceOp.hasUnitStride())
819   auto padOp = sliceOp.getSource().getDefiningOp<tensor::PadOp>();
825     if (Optional<bool> control = controlFn(sliceOp))
832       tensor::bubbleUpPadSlice(rewriter, padOp, sliceOp.getMixedOffsets(),
833                                sliceOp.getMixedSizes(), zeroSliceGuard);
836   rewriter.replaceOp(sliceOp, tiledPadOp->getResults());