Searched refs:padValue (Results 1 – 7 of 7) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | FusePadOpWithLinalgProducer.cpp | 47 Value padValue = padOp.getConstantPaddingValue(); in matchAndRewrite() local
48 if (!padValue) in matchAndRewrite()
86 rewriter.create<linalg::FillOp>(loc, padValue, initTensor.getResult()); in matchAndRewrite()
|
| H A D | Vectorization.cpp | 698 auto padValue = padOp.getConstantPaddingValue(); in tryVectorizeCopy() local
699 if (!padValue) { in tryVectorizeCopy()
704 padValue = rewriter.create<arith::ConstantOp>( in tryVectorizeCopy()
743 padOp.getLoc(), vecType, padOp.getSource(), readIndices, padValue, in tryVectorizeCopy()
814 auto padValue = padOp.getConstantPaddingValue(); in rewriteUser() local
815 if (!padValue) in rewriteUser()
826 xferOp.getPaddingMutable().assign(padValue); in rewriteUser()
880 auto padValue = padOp.getConstantPaddingValue(); in rewriteUser() local
881 if (!padValue) in rewriteUser()
1033 auto padValue = padOp.getConstantPaddingValue(); in rewriteUser() local
[all …]
|
| H A D | Transforms.cpp | 689 Value padValue = yieldOp.getValue(); in matchAndRewrite() local
690 Operation *definingOp = padValue.getDefiningOp(); in matchAndRewrite()
693 if (!definingOp && padValue.cast<BlockArgument>().getOwner() == &block) in matchAndRewrite()
705 .create<linalg::FillOp>(loc, ValueRange{padValue}, in matchAndRewrite()
737 auto padValue = padOp.getConstantPaddingValue(); in createFillOrGenerateOp() local
738 if (padValue) in createFillOrGenerateOp()
739 return rewriter.create<FillOp>(padOp.getLoc(), padValue, dest).result(); in createFillOrGenerateOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 81 Value padValue = padOp.getConstantPaddingValue(); in bubbleUpPadSlice() local
82 if (!padValue) in bubbleUpPadSlice()
239 builder.create<tensor::YieldOp>(gLoc, padValue); in bubbleUpPadSlice()
|
| H A D | TensorOps.cpp | 2214 Value padValue = yieldOp.getValue(); in getConstantPaddingValue() local
2216 if (matchPattern(padValue, m_Constant())) in getConstantPaddingValue()
2217 return padValue; in getConstantPaddingValue()
2219 if (padValue.getParentBlock() == &getRegion().front()) in getConstantPaddingValue()
2222 return padValue; in getConstantPaddingValue()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 530 Value padValue = padOp.getConstantPaddingValue(); in matchAndRewrite() local
531 if (!padValue || fillOp.value() != padValue) in matchAndRewrite()
548 fillOp.getLoc(), ValueRange{padValue}, ValueRange{newInitOp}); in matchAndRewrite()
615 Value padValue = srcPadOp.getConstantPaddingValue(); in matchAndRewrite() local
616 if (!padValue || dstFillOp.value() != padValue) in matchAndRewrite()
633 Value padValue = getValueOrCreateConstantIndexOp( in matchAndRewrite() local
638 applyMapToValues(rewriter, loc, addMap, {offsetValue, padValue})[0]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 59 Value padValue = rewriter.create<arith::ConstantOp>(loc, padAttr); in applyPad() local
62 input, padValue, lowIndices, highIndices, in applyPad()
|