Home
last modified time | relevance | path

Searched refs:ConstantOp (Results 1 – 25 of 184) sorted by relevance

12345678

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/IR/
H A DArithmetic.h43 class ConstantIntOp : public arith::ConstantOp {
45 using arith::ConstantOp::ConstantOp;
57 return arith::ConstantOp::getValue().cast<IntegerAttr>().getInt(); in value()
64 class ConstantFloatOp : public arith::ConstantOp {
66 using arith::ConstantOp::ConstantOp;
73 return arith::ConstantOp::getValue().cast<FloatAttr>().getValue(); in value()
80 class ConstantIndexOp : public arith::ConstantOp {
82 using arith::ConstantOp::ConstantOp;
88 return arith::ConstantOp::getValue().cast<IntegerAttr>().getInt(); in value()
/llvm-project-15.0.7/mlir/lib/Conversion/MathToLLVM/
H A DMathToLLVM.cpp63 LLVM::ConstantOp zero = in matchAndRewrite()
64 rewriter.create<LLVM::ConstantOp>(loc, boolType, boolZero); in matchAndRewrite()
77 LLVM::ConstantOp zero = in matchAndRewrite()
78 rewriter.create<LLVM::ConstantOp>(loc, boolType, boolZero); in matchAndRewrite()
109 LLVM::ConstantOp one; in matchAndRewrite()
111 one = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
115 one = rewriter.create<LLVM::ConstantOp>(loc, operandType, floatOne); in matchAndRewrite()
162 LLVM::ConstantOp one = in matchAndRewrite()
164 ? rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
216 LLVM::ConstantOp one; in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/
H A Dscf.py64 lb = arith.ConstantOp.create_index(0)
65 ub = arith.ConstantOp.create_index(42)
66 step = arith.ConstantOp.create_index(2)
96 one = arith.ConstantOp(i32, 1)
118 x_true = arith.ConstantOp(i32, 0)
119 y_true = arith.ConstantOp(i32, 1)
122 x_false = arith.ConstantOp(i32, 2)
123 y_false = arith.ConstantOp(i32, 3)
H A Dfunc.py24 c1 = arith.ConstantOp(IntegerType.get_signless(32), 42)
25 c2 = arith.ConstantOp(IntegerType.get_signless(64), 100)
26 c3 = arith.ConstantOp(F32Type.get(), 3.14)
27 c4 = arith.ConstantOp(F64Type.get(), 1.23)
52 c1 = arith.ConstantOp(
69 c1 = arith.ConstantOp.create_index(10)
H A Dtensor.py33 c0 = arith.ConstantOp(indexType, 0)
34 c1 = arith.ConstantOp(indexType, 1)
/llvm-project-15.0.7/mlir/lib/Dialect/Complex/IR/
H A DComplexDialect.cpp27 if (complex::ConstantOp::isBuildableWith(value, type)) { in materializeConstant()
28 return builder.create<complex::ConstantOp>(loc, type, in materializeConstant()
31 if (arith::ConstantOp::isBuildableWith(value, type)) in materializeConstant()
32 return builder.create<arith::ConstantOp>(loc, type, value); in materializeConstant()
H A DComplexOps.cpp20 OpFoldResult ConstantOp::fold(ArrayRef<Attribute> operands) { in fold()
25 void ConstantOp::getAsmResultNames( in getAsmResultNames()
30 bool ConstantOp::isBuildableWith(Attribute value, Type type) { in isBuildableWith()
42 LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/lib/Conversion/MathToSPIRV/
H A DMathToSPIRV.cpp39 return builder.create<spirv::ConstantOp>(loc, type, in getScalarOrVectorI32Constant()
43 return builder.create<spirv::ConstantOp>(loc, type, in getScalarOrVectorI32Constant()
83 Value signMask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
85 Value valueMask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
183 auto one = spirv::ConstantOp::getOne(type, loc, rewriter); in matchAndRewrite()
203 auto one = spirv::ConstantOp::getOne(type, operation.getLoc(), rewriter); in matchAndRewrite()
226 spirv::ConstantOp::getZero(adaptor.getLhs().getType(), loc, rewriter); in matchAndRewrite()
249 auto zero = spirv::ConstantOp::getZero(ty, loc, rewriter); in matchAndRewrite()
250 auto one = spirv::ConstantOp::getOne(ty, loc, rewriter); in matchAndRewrite()
253 half = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DExpandPatterns.cpp30 Value one = rewriter.create<arith::ConstantOp>(loc, floatOne); in convertTanhOp()
31 Value two = rewriter.create<arith::ConstantOp>(loc, floatTwo); in convertTanhOp()
49 Value zero = rewriter.create<arith::ConstantOp>(loc, floatZero); in convertTanhOp()
66 rewriter.create<arith::ConstantOp>(loc, IntegerAttr::get(elementTy, 0)); in convertCtlzOp()
67 auto leadingZeros = rewriter.create<arith::ConstantOp>( in convertCtlzOp()
99 rewriter.create<arith::ConstantOp>(loc, IntegerAttr::get(elementTy, 1)); in convertCtlzOp()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp34 static arith::ConstantOp
39 return rewriter.create<arith::ConstantOp>( in createConstFromIntAttribute()
56 auto zero = rewriter.create<arith::ConstantOp>( in createLinalgBodyCalculationForElementwiseOp()
269 auto one = rewriter.create<arith::ConstantOp>( in createLinalgBodyCalculationForElementwiseOp()
379 auto min = rewriter.create<arith::ConstantOp>( in createLinalgBodyCalculationForElementwiseOp()
381 auto max = rewriter.create<arith::ConstantOp>( in createLinalgBodyCalculationForElementwiseOp()
424 auto n = rewriter.create<arith::ConstantOp>( in createLinalgBodyCalculationForElementwiseOp()
1387 auto hMax = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
1389 auto wMax = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
1411 yStride = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
[all …]
H A DTosaToLinalgNamed.cpp59 Value padValue = rewriter.create<arith::ConstantOp>(loc, padAttr); in applyPad()
81 auto one = rewriter.create<arith::ConstantOp>( in getConvOutputDim()
237 rewriter.create<arith::ConstantOp>(loc, weightPermAttr); in matchAndRewrite()
280 auto iZpVal = rewriter.create<arith::ConstantOp>(loc, iZp); in matchAndRewrite()
281 auto kZpVal = rewriter.create<arith::ConstantOp>(loc, kZp); in matchAndRewrite()
466 auto iZpVal = rewriter.create<arith::ConstantOp>(loc, iZp); in matchAndRewrite()
467 auto kZpVal = rewriter.create<arith::ConstantOp>(loc, kZp); in matchAndRewrite()
544 auto aZp = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
546 auto bZp = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
654 auto inputZp = rewriter.create<arith::ConstantOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToSPIRV/
H A DArithmeticToSPIRV.cpp32 : public OpConversionPattern<arith::ConstantOp> {
33 using OpConversionPattern<arith::ConstantOp>::OpConversionPattern;
36 matchAndRewrite(arith::ConstantOp op, OpAdaptor adaptor,
42 : public OpConversionPattern<arith::ConstantOp> {
43 using OpConversionPattern<arith::ConstantOp>::OpConversionPattern;
46 matchAndRewrite(arith::ConstantOp op, OpAdaptor adaptor,
300 arith::ConstantOp constOp, OpAdaptor adaptor, in matchAndRewrite()
386 arith::ConstantOp constOp, OpAdaptor adaptor, in matchAndRewrite()
584 Value one = spirv::ConstantOp::getOne(dstType, loc, rewriter); in matchAndRewrite()
605 Value one = spirv::ConstantOp::getOne(dstType, loc, rewriter); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUOpsLowering.cpp89 zero = rewriter.create<LLVM::ConstantOp>(loc, i32Type, in matchAndRewrite()
126 Value numElements = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
199 Value zeroI64 = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
229 Value zero = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
233 Value stringLen = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
236 Value oneI32 = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
238 Value zeroI32 = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
256 arguments.push_back(rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
328 Value zero = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DCodegenUtils.h113 return builder.create<complex::ConstantOp>(loc, tp, zeroa); in constantZero()
115 return builder.create<arith::ConstantOp>(loc, tp, builder.getZeroAttr(tp)); in constantZero()
125 return builder.create<complex::ConstantOp>(loc, tp, zeroa); in constantOne()
127 return builder.create<arith::ConstantOp>(loc, tp, getOneAttr(builder, tp)); in constantOne()
/llvm-project-15.0.7/mlir/lib/Dialect/Func/IR/
H A DFuncOps.cpp116 if (ConstantOp::isBuildableWith(value, type)) in materializeConstant()
117 return builder.create<ConstantOp>(loc, type, in materializeConstant()
188 LogicalResult ConstantOp::verify() { in verify()
205 OpFoldResult ConstantOp::fold(ArrayRef<Attribute> operands) { in fold()
210 void ConstantOp::getAsmResultNames( in getAsmResultNames()
215 bool ConstantOp::isBuildableWith(Attribute value, Type type) { in isBuildableWith()
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToSPIRV/
H A DMemRefToSPIRV.cpp45 auto idx = builder.create<spirv::ConstantOp>(loc, targetType, idxAttr); in getOffsetForBitwidth()
48 builder.create<spirv::ConstantOp>(loc, targetType, srcBitsAttr); in getOffsetForBitwidth()
70 auto idx = builder.create<spirv::ConstantOp>(loc, targetType, attr); in adjustAccessChainForBitwidth()
140 auto one = spirv::ConstantOp::getOne(srcInt.getType(), loc, builder); in castIntNToBool()
150 Value zero = spirv::ConstantOp::getZero(dstType, loc, builder); in castBoolToIntN()
151 Value one = spirv::ConstantOp::getOne(dstType, loc, builder); in castBoolToIntN()
381 Value mask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
391 rewriter.create<spirv::ConstantOp>(loc, dstType, shiftValueAttr); in matchAndRewrite()
399 mask = spirv::ConstantOp::getOne(result.getType(), loc, rewriter); in matchAndRewrite()
491 Value mask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
/llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/
H A DDialect.cpp98 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
102 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
112 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
125 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
133 mlir::LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/
H A DDialect.cpp98 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
102 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
112 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
125 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
133 mlir::LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToLLVM/
H A DArithmeticToLLVM.cpp77 struct ConstantOpLowering : public ConvertOpToLLVMPattern<arith::ConstantOp> {
78 using ConvertOpToLLVMPattern<arith::ConstantOp>::ConvertOpToLLVMPattern;
81 matchAndRewrite(arith::ConstantOp op, OpAdaptor adaptor,
120 ConstantOpLowering::matchAndRewrite(arith::ConstantOp op, OpAdaptor adaptor, in matchAndRewrite()
122 return LLVM::detail::oneToOneRewrite(op, LLVM::ConstantOp::getOperationName(), in matchAndRewrite()
/llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/
H A DDialect.cpp147 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
151 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
161 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
174 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
238 mlir::LogicalResult ConstantOp::verify() { in verify()
248 void ConstantOp::inferShapes() { getResult().setType(getValue().getType()); } in inferShapes()
638 return builder.create<ConstantOp>(loc, type, in materializeConstant()
/llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/
H A DDialect.cpp160 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
164 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
174 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
187 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
195 mlir::LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/
H A DDialect.cpp160 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
164 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
174 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
187 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
195 mlir::LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/
H A DDialect.cpp160 void ConstantOp::build(mlir::OpBuilder &builder, mlir::OperationState &state, in build()
164 ConstantOp::build(builder, state, dataType, dataAttribute); in build()
174 mlir::ParseResult ConstantOp::parse(mlir::OpAsmParser &parser, in parse()
187 void ConstantOp::print(mlir::OpAsmPrinter &printer) { in print()
195 mlir::LogicalResult ConstantOp::verify() { in verify()
/llvm-project-15.0.7/mlir/lib/Conversion/SPIRVToLLVM/
H A DSPIRVToLLVM.cpp88 return rewriter.create<LLVM::ConstantOp>( in createConstantAllBitsSet()
93 return rewriter.create<LLVM::ConstantOp>( in createConstantAllBitsSet()
102 return rewriter.create<LLVM::ConstantOp>( in createFPConstant()
108 return rewriter.create<LLVM::ConstantOp>( in createFPConstant()
147 auto index = rewriter.create<LLVM::ConstantOp>( in broadcast()
214 return rewriter.create<LLVM::ConstantOp>( in createI32ConstantOf()
316 Value zero = rewriter.create<LLVM::ConstantOp>( in matchAndRewrite()
385 : public SPIRVToLLVMConversion<spirv::ConstantOp> {
411 rewriter.replaceOpWithNewOp<LLVM::ConstantOp>( in matchAndRewrite()
422 rewriter.replaceOpWithNewOp<LLVM::ConstantOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/X86Vector/Transforms/
H A DLegalizeForLLVMExport.cpp79 src = rewriter.create<arith::ConstantOp>(op.getLoc(), opType, in matchAndRewrite()
83 src = rewriter.create<arith::ConstantOp>(op->getLoc(), opType, zeroAttr); in matchAndRewrite()
116 rewriter.create<LLVM::ConstantOp>(op.getLoc(), llvmIntType, attr); in matchAndRewrite()

12345678