Home
last modified time | relevance | path

Searched refs:paddingValue (Results 1 – 8 of 8) sorted by relevance

/llvm-project/mlir/lib/Dialect/Tensor/Transforms/
H A DRewriteAsConstant.cpp156 Value paddingValue = padTensorOp.getConstantPaddingValue(); in matchAndRewrite() local
160 if (!paddingValue || !matchPattern(paddingValue, m_Constant(&paddingAttr))) in matchAndRewrite()
H A DPackAndUnpackPatterns.cpp213 if (auto paddingValue = packOp.getPaddingValue()) in matchAndRewrite()
214 if (!isEqualConstantIntOrValue(paddingValue, constantPaddingValue)) in matchAndRewrite()
218 if (auto paddingValue = packOp.getPaddingValue()) matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DPadding.cpp132 Value paddingValue; in padOperandToSmallestStaticBoundingBox() local
136 paddingValue = rewriter.create<complex::ConstantOp>(opToPad.getLoc(), in padOperandToSmallestStaticBoundingBox()
139 paddingValue = rewriter.create<arith::ConstantOp>( in padOperandToSmallestStaticBoundingBox()
149 opOperand->get(), paddingValue, nofold); in padOperandToSmallestStaticBoundingBox()
H A DHoistPadding.cpp322 Value paddingValue = opToHoist.getConstantPaddingValue(); in finalizeHoistPaddingAnalysis() local
323 if (!paddingValue || in finalizeHoistPaddingAnalysis()
324 !isa_and_nonnull<arith::ConstantOp>(paddingValue.getDefiningOp())) { in finalizeHoistPaddingAnalysis()
H A DTransforms.cpp271 Value paddingValue = packOp.getPaddingValue(); in lowerPack()
272 if (!paddingValue) { in lowerPack()
273 paddingValue = rewriter.create<arith::ConstantOp>( in lowerPack()
278 highs, paddingValue, /*nofold=*/false); in lowerPack()
270 Value paddingValue = packOp.getPaddingValue(); lowerPack() local
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp4089 std::optional<Value> paddingValue, in areTilesAndTiledDimsAllConstant()
4098 paddingValue ? *paddingValue : nullptr, in areTilesAndTiledDimsAllConstant()
4176 auto paddingValue = getPaddingValue(); in inferStaticShape()
4177 if (paddingValue && in inferStaticShape()
4178 paddingValue.getType() != getSourceType().getElementType()) { in inferStaticShape()
4181 << " but got: " << paddingValue.getType(); in inferStaticShape()
4184 if (!paddingValue && in inferStaticShape()
4536 std::optional<Attribute> paddingValue; in matchAndRewrite()
4538 paddingValue in matchAndRewrite()
3862 build(OpBuilder & builder,OperationState & state,Value source,Value dest,ArrayRef<int64_t> innerDimsPos,ArrayRef<OpFoldResult> innerTiles,std::optional<Value> paddingValue,ArrayRef<int64_t> outerDimsPerm) build() argument
3932 auto paddingValue = getPaddingValue(); verify() local
4283 std::optional<Attribute> paddingValue; fold() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp838 if (auto paddingValue = packOp.getPaddingValue()) in matchAndRewrite()
839 if (!isEqualConstantIntOrValue(paddingValue, fillOp.value())) in matchAndRewrite()
817 if (auto paddingValue = packOp.getPaddingValue()) foldFillPackIntoFillOp() local
/llvm-project/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensorOps.td1972 CArg<"std::optional<Value>", "std::nullopt">:$paddingValue,