| /llvm-project-15.0.7/mlir/lib/Conversion/LinalgToStandard/ |
| H A D | LinalgToStandard.cpp | 60 SmallVector<Type, 4> inputTypes(extractOperandTypes(op)); in getLibraryCallSymbolRef() local
64 auto libFnType = rewriter.getFunctionType(inputTypes, {}); in getLibraryCallSymbolRef()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | NormalizeMemRefs.cpp | 342 SmallVector<Type, 8> inputTypes; in normalizeFuncOpMemRefs() local
351 inputTypes.push_back(argType); in normalizeFuncOpMemRefs()
361 inputTypes.push_back(newMemRefType); in normalizeFuncOpMemRefs()
470 FunctionType::get(&getContext(), /*inputs=*/inputTypes, in normalizeFuncOpMemRefs()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/ |
| H A D | SparseTensorDialect.cpp | 269 TypeRange inputTypes, Type outputType) { in verifyNumBlockArgs() argument
271 unsigned expectedNum = inputTypes.size(); in verifyNumBlockArgs()
278 if (typ != inputTypes[i]) in verifyNumBlockArgs()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 60 TypeRange inputTypes, TypeRange outputTypes, in fillStructuredOpRegion() argument
69 for (auto containers : {inputTypes, outputTypes}) { in fillStructuredOpRegion()
127 SmallVectorImpl<Type> &inputTypes, in parseCommonStructuredOpParts() argument
142 parser.parseColonTypeList(inputTypes) || parser.parseRParen()) in parseCommonStructuredOpParts()
153 if (parser.resolveOperands(inputsOperands, inputTypes, inputsOperandsLoc, in parseCommonStructuredOpParts()
182 if (numRegionArgs != inputTypes.size() + outputTypes.size()) { in parseNamedStructuredOpRegion()
187 numRegionArgs, inputTypes.size() + outputTypes.size())); in parseNamedStructuredOpRegion()
191 fillStructuredOpRegion(opBuilder, region, inputTypes, outputTypes, attrs, in parseNamedStructuredOpRegion()
209 SmallVector<Type, 1> inputTypes, outputTypes; in parseNamedStructuredOp() local
210 if (parseCommonStructuredOpParts(parser, result, inputTypes, outputTypes)) in parseNamedStructuredOp()
[all …]
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | Allocatable.cpp | 148 llvm::ArrayRef<mlir::Type> inputTypes = callee.getFunctionType().getInputs(); in genRuntimeInitCharacter() local
149 if (inputTypes.size() != 5) in genRuntimeInitCharacter()
153 args.push_back(builder.createConvert(loc, inputTypes[0], box.getAddr())); in genRuntimeInitCharacter()
154 args.push_back(builder.createConvert(loc, inputTypes[1], len)); in genRuntimeInitCharacter()
156 args.push_back(builder.createIntegerConstant(loc, inputTypes[2], kind)); in genRuntimeInitCharacter()
158 args.push_back(builder.createIntegerConstant(loc, inputTypes[3], rank)); in genRuntimeInitCharacter()
161 args.push_back(builder.createIntegerConstant(loc, inputTypes[4], corank)); in genRuntimeInitCharacter()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncParallelFor.cpp | 474 SmallVector<Type> inputTypes; in createAsyncDispatchFunction() local
475 inputTypes.push_back(async::GroupType::get(rewriter.getContext())); in createAsyncDispatchFunction()
476 inputTypes.push_back(rewriter.getIndexType()); // add blockStart argument in createAsyncDispatchFunction()
477 inputTypes.append(computeFuncInputTypes.begin(), computeFuncInputTypes.end()); in createAsyncDispatchFunction()
479 FunctionType type = rewriter.getFunctionType(inputTypes, TypeRange()); in createAsyncDispatchFunction()
|
| H A D | AsyncToAsyncRuntime.cpp | 258 SmallVector<Type, 4> inputTypes(typesRange.begin(), typesRange.end()); in outlineExecuteOp() local
261 auto funcType = FunctionType::get(ctx, inputTypes, outputTypes); in outlineExecuteOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | FunctionInterfaces.td | 139 ArrayRef<NamedAttribute> attrs, ArrayRef<Type> inputTypes) {
150 for (Type input : inputTypes)
213 ArrayRef<Type> inputTypes = $_op.getArgumentTypes();
214 SmallVector<Location> locations(inputTypes.size(),
216 entry->addArguments(inputTypes, locations);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 4951 SmallVector<Type> inputTypes; in parse() local
4958 parser.parseColonTypeList(inputTypes) || parser.parseRParen()) in parse()
4961 if (parser.resolveOperands(inputsOperands, inputTypes, inputsOperandsLoc, in parse()
|