Lines Matching defs:padOp
745 auto padOp = cast<tensor::PadOp>(op);
747 padOp.getSource(), options, invocationStack);
751 return MemRefType::get(padOp.getResultType().getShape(),
752 padOp.getResultType().getElementType(), layout,
758 auto padOp = cast<tensor::PadOp>(op);
759 Location loc = padOp.getLoc();
760 RankedTensorType resultType = padOp.getResultType();
761 RankedTensorType srcType = padOp.getSourceType();
772 SmallVector<OpFoldResult> mixedLowPad = padOp.getMixedLowPad();
773 SmallVector<OpFoldResult> mixedHighPad = padOp.getMixedHighPad();
778 Value srcDim = rewriter.create<tensor::DimOp>(loc, padOp.getSource(), i);
791 allocateTensorForShapedValue(rewriter, loc, padOp.getResult(), options,
800 rewriter, loc, *tensorAlloc, dynamicSizes, padOp.getBodyRegion());
804 getMixedSizes(rewriter, loc, padOp.getSource());
808 padOp, padOp.getSource(), filledBuffer,
809 /*offsets=*/padOp.getMixedLowPad(), sliceSizes, sliceStrides);