Home
last modified time | relevance | path

Searched refs:resultTypes (Results 1 – 25 of 69) sorted by relevance

123

/llvm-project-15.0.7/mlir/lib/IR/
H A DFunctionImplementation.cpp91 resultTypes.push_back(ty); in parseFunctionResultList()
102 resultTypes.emplace_back(); in parseFunctionResultList()
105 if (parser.parseType(resultTypes.back()) || in parseFunctionResultList()
119 SmallVectorImpl<Type> &resultTypes, in parseFunctionSignature() argument
169 SmallVector<Type> resultTypes; in parseFunctionOp() local
185 resultTypes, resultAttrs)) in parseFunctionOp()
222 assert(resultAttrs.size() == resultTypes.size()); in parseFunctionOp()
266 ArrayRef<Type> resultTypes) { in printFunctionSignature() argument
296 if (!resultTypes.empty()) { in printFunctionSignature()
299 printFunctionResultList(p, resultTypes, resultAttrs); in printFunctionSignature()
[all …]
H A DOperation.cpp35 TypeRange resultTypes, ValueRange operands, in create() argument
39 Operation *op = create(location, name, resultTypes, operands, in create()
50 TypeRange resultTypes, ValueRange operands, in create() argument
53 assert(llvm::all_of(resultTypes, [](Type t) { return t; }) && in create()
58 unsigned numInlineResults = OpResult::getNumInline(resultTypes.size()); in create()
61 unsigned numResults = resultTypes.size(); in create()
89 auto resultTypeIt = resultTypes.begin(); in create()
1159 auto resultTypes = op->getResultTypes(); in verifyCastInterfaceOp() local
1160 if (llvm::empty(resultTypes)) in verifyCastInterfaceOp()
1165 if (!areCastCompatible(operandTypes, resultTypes)) { in verifyCastInterfaceOp()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp200 SmallVector<Type, 4> resultTypes; in updateFunctionSignature() local
230 resultTypes[operandEn.index()] = memrefType; in updateFunctionSignature()
451 SmallVector<Type, 4> resultTypes; in normalizeFuncOpMemRefs() local
459 resultTypes.push_back(resType); in normalizeFuncOpMemRefs()
466 resultTypes.push_back(newMemRefType); in normalizeFuncOpMemRefs()
471 /*results=*/resultTypes); in normalizeFuncOpMemRefs()
491 SmallVector<Type, 4> resultTypes; in createOpResultsNormalized() local
499 resultTypes.push_back(resultType); in createOpResultsNormalized()
508 resultTypes.push_back(memrefType); in createOpResultsNormalized()
511 resultTypes.push_back(newMemRefType); in createOpResultsNormalized()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAsyncRegionRewriter.cpp103 SmallVector<Type, 1> resultTypes; in rewriteAsyncOp() local
104 resultTypes.reserve(1 + op->getNumResults()); in rewriteAsyncOp()
105 copy(op->getResultTypes(), std::back_inserter(resultTypes)); in rewriteAsyncOp()
106 resultTypes.push_back(tokenType); in rewriteAsyncOp()
107 auto *newOp = Operation::create(op->getLoc(), op->getName(), resultTypes, in rewriteAsyncOp()
147 SmallVector<Type, 2> resultTypes; in addExecuteResults() local
148 resultTypes.reserve(executeOp.getNumResults() + results.size()); in addExecuteResults()
149 transform(executeOp.getResultTypes(), std::back_inserter(resultTypes), in addExecuteResults()
157 transform(results, std::back_inserter(resultTypes), in addExecuteResults()
163 executeOp.getLoc(), TypeRange{resultTypes}.drop_front() /*drop token*/, in addExecuteResults()
/llvm-project-15.0.7/mlir/test/mlir-tblgen/
H A Dop-result.td20 // CHECK: ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands
21 // CHECK: assert(resultTypes.size() == 1u && "mismatched number of return types");
22 // CHECK-NEXT: odsState.addTypes(resultTypes);
50 …d(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes) {
51 // CHECK-NEXT: assert(resultTypes.size() == 3u && "mismatched number of results");
52 // CHECK-NEXT: odsState.addTypes(resultTypes);
94 // CHECK: ::mlir::TypeRange resultTypes
95 // CHECK: assert(resultTypes.size() >= 1u && "mismatched number of return types");
96 // CHECK-NEXT: odsState.addTypes(resultTypes);
H A Dop-decl-and-defs.td204 …ilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value …
205 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
217 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
229 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
243 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
253 …ilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value …
254 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
265 …ilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value …
266 …mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueR…
298 …ilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::Value …
[all …]
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DFunctionImplementation.h62 bool &isVariadic, SmallVectorImpl<Type> &resultTypes,
68 ArrayRef<Type> resultTypes);
88 ArrayRef<Type> resultTypes);
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DGeneralization.cpp52 SmallVector<RankedTensorType> resultTypes = linalgOp.getOutputTensorTypes(); in generalizeNamedOp() local
53 SmallVector<Type> types(resultTypes.begin(), resultTypes.end()); in generalizeNamedOp()
H A DElementwiseToLinalg.cpp101 auto resultTypes = llvm::to_vector<6>( in matchAndRewrite() local
108 resultTypes, op->getAttrs()); in matchAndRewrite()
H A DFusion.cpp153 SmallVector<Type, 4> resultTypes; in fuse() local
154 resultTypes.reserve(producer->getNumResults()); in fuse()
162 resultTypes.push_back(tensor::ExtractSliceOp::inferResultType( in fuse()
167 Operation *clonedOp = producer.clone(b, loc, resultTypes, clonedShapes); in fuse()
H A DDecomposeLinalgOps.cpp198 SmallVector<Type> resultTypes = llvm::to_vector(genericOp.getResultTypes()); in createPeeledGenericOp() local
199 resultTypes.append(newResultTypes.begin(), newResultTypes.end()); in createPeeledGenericOp()
203 loc, resultTypes, genericOp.inputs(), outsOperands, indexingMapAttr, in createPeeledGenericOp()
H A DDropUnitDims.cpp428 SmallVector<Type, 4> resultTypes; in matchAndRewrite() local
429 resultTypes.reserve(genericOp.getNumResults()); in matchAndRewrite()
431 resultTypes.push_back(newInputOutputTypes[i + genericOp.getNumInputs()]); in matchAndRewrite()
433 loc, resultTypes, newInputs, newOutputs, newIndexingMaps, in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/IR/
H A DAsync.cpp91 TypeRange resultTypes, ValueRange dependencies, in build() argument
108 for (Type type : resultTypes) in build()
125 if (resultTypes.empty() && !bodyBuilder) { in build()
212 SmallVector<Type, 4> resultTypes; in parse() local
214 if (parser.parseOptionalArrowTypeList(resultTypes) || in parse()
217 parser.addTypesToList(resultTypes, result.types) || in parse()
/llvm-project-15.0.7/mlir/lib/Dialect/PDLInterp/IR/
H A DPDLInterp.cpp106 SmallVectorImpl<Type> &resultTypes, UnitAttr &inferredResultTypes) { in parseCreateOperationOpResults() argument
120 p.parseColonTypeList(resultTypes) || p.parseRParen()); in parseCreateOperationOpResults()
125 TypeRange resultTypes, in printCreateOperationOpResults() argument
134 if (!resultTypes.empty()) in printCreateOperationOpResults()
135 p << " -> (" << resultOperands << " : " << resultTypes << ")"; in printCreateOperationOpResults()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp53 return rewriter.create<math::AbsOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
66 return rewriter.create<arith::AddFOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
69 return rewriter.create<arith::AddIOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
73 return rewriter.create<arith::SubFOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
76 return rewriter.create<arith::SubIOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
85 return rewriter.create<arith::MulFOp>(loc, resultTypes, args); in createLinalgBodyCalculationForElementwiseOp()
125 int cWidth = resultTypes[0].getIntOrFloatBitWidth(); in createLinalgBodyCalculationForElementwiseOp()
128 a = rewriter.create<arith::ExtSIOp>(loc, resultTypes[0], a); in createLinalgBodyCalculationForElementwiseOp()
130 b = rewriter.create<arith::ExtSIOp>(loc, resultTypes[0], b); in createLinalgBodyCalculationForElementwiseOp()
132 return rewriter.create<arith::MulIOp>(loc, resultTypes, a, b); in createLinalgBodyCalculationForElementwiseOp()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DFuncBufferizableOpInterfaceImpl.cpp269 SmallVector<Type> resultTypes; in bufferize() local
285 retValMapping[returnValIdx] = resultTypes.size(); in bufferize()
286 resultTypes.push_back(returnType); in bufferize()
291 retValMapping[returnValIdx] = resultTypes.size(); in bufferize()
292 resultTypes.push_back(funcType.getResult(resultTypes.size())); in bufferize()
336 callOp.getLoc(), funcOp.getSymName(), resultTypes, newOperands); in bufferize()
H A DOneShotModuleBufferize.cpp340 SmallVector<Type> resultTypes; in foldMemRefCasts() local
345 resultTypes.push_back(castOp.getSource().getType()); in foldMemRefCasts()
347 resultTypes.push_back(operand.get().getType()); in foldMemRefCasts()
352 funcOp.getContext(), funcOp.getFunctionType().getInputs(), resultTypes); in foldMemRefCasts()
/llvm-project-15.0.7/mlir/unittests/TableGen/
H A DOpBuildGen.cpp49 void verifyOp(OpTy &&concreteOp, std::vector<Type> resultTypes, in verifyOp() argument
55 EXPECT_EQ(op->getNumResults(), resultTypes.size()); in verifyOp()
57 EXPECT_EQ(op->getResult(idx).getType(), resultTypes[idx]); in verifyOp()
/llvm-project-15.0.7/mlir/lib/Target/SPIRV/Deserialization/
H A DDeserializeOps.cpp219 SmallVector<Type, 1> resultTypes; in processOpWithoutGrammarAttr() local
234 resultTypes.push_back(type); in processOpWithoutGrammarAttr()
284 opState.addTypes(resultTypes); in processOpWithoutGrammarAttr()
508 SmallVector<Type, 1> resultTypes; in processOp() local
570 opBuilder.create<spirv::CopyMemoryOp>(loc, resultTypes, operands, attributes); in processOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp198 SmallVectorImpl<Type> &resultTypes) { in parseNamedStructuredOpResults() argument
199 if (parser.parseOptionalArrowTypeList(resultTypes)) in parseNamedStructuredOpResults()
231 TypeRange resultTypes) { in printNamedStructuredOpResults() argument
232 if (resultTypes.empty()) in printNamedStructuredOpResults()
234 p.printOptionalArrowTypeList(resultTypes); in printNamedStructuredOpResults()
1746 SmallVector<Type> resultTypes(linalgOp->result_type_begin(), in matchAndRewrite() local
1748 resultTypes[resultNumber] = resultType; in matchAndRewrite()
1816 resultTypes.push_back(resultType); in createNewOperandWithStaticSizes()
1849 resultTypes.push_back(resultType); in createNewOperandWithStaticSizes()
1879 SmallVector<Type> resultTypes; in matchAndRewrite() local
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/PDL/IR/
H A DPDL.cpp182 OperandRange resultTypes) { in verifyResultTypesAreInferrable() argument
203 if (resultTypes.empty()) { in verifyResultTypesAreInferrable()
233 for (const auto &it : llvm::enumerate(resultTypes)) { in verifyResultTypesAreInferrable()
/llvm-project-15.0.7/mlir/lib/Tools/PDLL/AST/
H A DNodes.cpp303 ArrayRef<Expr *> resultTypes, in create() argument
307 operands.size() + resultTypes.size(), attributes.size()); in create()
313 OperationExpr(loc, resultType, name, operands.size(), resultTypes.size(), in create()
317 std::uninitialized_copy(resultTypes.begin(), resultTypes.end(), in create()
/llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/
H A DTypeConverter.cpp463 SmallVector<Type, 8> resultTypes; in packFunctionResults() local
464 resultTypes.reserve(types.size()); in packFunctionResults()
469 resultTypes.push_back(converted); in packFunctionResults()
472 return LLVM::LLVMStructType::getLiteral(&getContext(), resultTypes); in packFunctionResults()
/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/
H A DFunctionCallUtils.cpp155 ArrayRef<Type> resultTypes) { in createLLVMCall() argument
157 .create<LLVM::CallOp>(loc, resultTypes, SymbolRefAttr::get(fn), in createLLVMCall()
/llvm-project-15.0.7/mlir/lib/Conversion/FuncToLLVM/
H A DFuncToLLVM.cpp500 auto resultTypes = llvm::to_vector<4>(callOp.getResultTypes()); in matchAndRewrite() local
504 this->getTypeConverter()->packFunctionResults(resultTypes))) in matchAndRewrite()
535 assert(results.size() == resultTypes.size() && in matchAndRewrite()
538 rewriter, callOp.getLoc(), resultTypes, results); in matchAndRewrite()
540 resultTypes, results, in matchAndRewrite()

123