Home
last modified time | relevance | path

Searched refs:inputTypes (Results 1 – 14 of 14) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseAssembler.cpp178 SmallVector<Type> inputTypes; in matchAndRewrite()
182 convTypes(hasAnnotation, funcOp.getArgumentTypes(), inputTypes, nullptr, in matchAndRewrite()
202 unsigned extra = inputTypes.size(); in matchAndRewrite()
203 inputTypes.append(extraTypes); in matchAndRewrite()
205 loc, orgName, FunctionType::get(context, inputTypes, outputTypes)); in matchAndRewrite()
176 SmallVector<Type> inputTypes; matchAndRewrite() local
/llvm-project/flang/lib/Optimizer/Builder/Runtime/
H A DDerived.cpp85 llvm::ArrayRef<mlir::Type> inputTypes = callee.getFunctionType().getInputs(); in genSameTypeAs()
87 args.push_back(builder.createConvert(loc, inputTypes[0], box)); in genSameTypeAs()
88 args.push_back(builder.createConvert(loc, inputTypes[1], typeDesc)); in genSameTypeAs()
89 mlir::Value rankCst = builder.createIntegerConstant(loc, inputTypes[2], rank); in genSameTypeAs()
90 mlir::Value c0 = builder.createIntegerConstant(loc, inputTypes[3], 0);
70 llvm::ArrayRef<mlir::Type> inputTypes = callee.getFunctionType().getInputs(); genNullifyDerivedType() local
/llvm-project/flang/lib/Lower/
H A DAllocatable.cpp161 llvm::ArrayRef<mlir::Type> inputTypes = callee.getFunctionType().getInputs(); in genRuntimeInitCharacter()
162 if (inputTypes.size() != 5) in genRuntimeInitCharacter()
166 args.push_back(builder.createConvert(loc, inputTypes[0], box.getAddr())); in genRuntimeInitCharacter()
167 args.push_back(builder.createConvert(loc, inputTypes[1], len)); in genRuntimeInitCharacter()
170 args.push_back(builder.createIntegerConstant(loc, inputTypes[2], kind)); in genRuntimeInitCharacter()
172 args.push_back(builder.createIntegerConstant(loc, inputTypes[3], rank)); in genRuntimeInitCharacter()
175 args.push_back(builder.createIntegerConstant(loc, inputTypes[4], corank));
661 llvm::ArrayRef<mlir::Type> inputTypes = in genInitIntrinsic()
664 args.push_back(builder.createConvert(loc, inputTypes[0], box.getAddr())); in genInitIntrinsic()
665 args.push_back(builder.createConvert(loc, inputTypes[ in genInitIntrinsic()
158 llvm::ArrayRef<mlir::Type> inputTypes = callee.getFunctionType().getInputs(); genRuntimeInitCharacter() local
633 llvm::ArrayRef<mlir::Type> inputTypes = genInitDerived() local
660 llvm::ArrayRef<mlir::Type> inputTypes = genInitIntrinsic() local
[all...]
/llvm-project/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp372 SmallVector<Type, 8> inputTypes; in normalizeFuncOpMemRefs()
381 inputTypes.push_back(argType); in normalizeFuncOpMemRefs()
390 inputTypes.push_back(newMemRefType); in normalizeFuncOpMemRefs()
498 FunctionType::get(&getContext(), /*inputs=*/inputTypes, in createOpResultsNormalized()
356 SmallVector<Type, 8> inputTypes; normalizeFuncOpMemRefs() local
/llvm-project/mlir/lib/Conversion/LinalgToStandard/
H A DLinalgToStandard.cpp69 SmallVector<Type, 4> inputTypes(extractOperandTypes(op)); in getLibraryCallSymbolRef() local
76 auto libFnType = rewriter.getFunctionType(inputTypes, {}); in getLibraryCallSymbolRef()
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp124 /// types of `inputTypes` and `outputTypes`. in fillStructuredOpRegion()
126 TypeRange inputTypes, TypeRange outputTypes, in fillStructuredOpRegion()
131 for (auto containers : {inputTypes, outputTypes}) { in fillStructuredOpRegion()
207 SmallVectorImpl<Type> &inputTypes, in parseCommonStructuredOpParts()
228 parser.parseColonTypeList(inputTypes) || parser.parseRParen()) in parseCommonStructuredOpParts()
239 if (parser.resolveOperands(inputsOperands, inputTypes, inputsOperandsLoc, in parseCommonStructuredOpParts()
294 TypeRange inputTypes, TypeRange outputTypes, ArrayRef<NamedAttribute> attrs, in parseNamedStructuredOpResults()
296 if (numRegionArgs != inputTypes.size() + outputTypes.size()) {
301 numRegionArgs, inputTypes.size() + outputTypes.size())); in parseNamedStructuredOp()
305 fillStructuredOpRegion(opBuilder, region, inputTypes, outputType in parseNamedStructuredOp()
120 fillStructuredOpRegion(OpBuilder & opBuilder,Region & region,TypeRange inputTypes,TypeRange outputTypes,ArrayRef<NamedAttribute> attrs,RegionBuilderFn regionBuilder) fillStructuredOpRegion() argument
186 parseCommonStructuredOpParts(OpAsmParser & parser,OperationState & result,SmallVectorImpl<Type> & inputTypes,SmallVectorImpl<Type> & outputTypes,bool addOperandSegmentSizes=true) parseCommonStructuredOpParts() argument
273 parseNamedStructuredOpRegion(OpAsmParser & parser,Region & region,unsigned numRegionArgs,TypeRange inputTypes,TypeRange outputTypes,ArrayRef<NamedAttribute> attrs,RegionBuilderFn regionBuilder) parseNamedStructuredOpRegion() argument
302 SmallVector<Type, 1> inputTypes, outputTypes; parseNamedStructuredOp() local
1095 SmallVector<Type, 1> inputTypes, outputTypes; parse() local
1256 SmallVector<Type, 4> inputTypes, outputTypes; parseDstStyleOp() local
[all...]
/llvm-project/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp475 SmallVector<Type> inputTypes; in createAsyncDispatchFunction()
476 inputTypes.push_back(async::GroupType::get(rewriter.getContext())); in createAsyncDispatchFunction()
477 inputTypes.push_back(rewriter.getIndexType()); // add blockStart argument in createAsyncDispatchFunction()
478 inputTypes.append(computeFuncInputTypes.begin(), computeFuncInputTypes.end()); in createAsyncDispatchFunction()
480 FunctionType type = rewriter.getFunctionType(inputTypes, TypeRange()); in createAsyncDispatchFunction() local
H A DAsyncToAsyncRuntime.cpp316 SmallVector<Type, 4> inputTypes(typesRange.begin(), typesRange.end()); in outlineExecuteOp() local
319 auto funcType = FunctionType::get(ctx, inputTypes, outputTypes); in outlineExecuteOp()
/llvm-project/mlir/include/mlir/Interfaces/
H A DFunctionInterfaces.td133 ArrayRef<NamedAttribute> attrs, TypeRange inputTypes) {
144 for (Type input : inputTypes)
207 ::llvm::ArrayRef<::mlir::Type> inputTypes = $_op.getArgumentTypes();
208 ::llvm::SmallVector<::mlir::Location> locations(inputTypes.size(),
210 entry->addArguments(inputTypes, locations);
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp521 RankedTensorType ConcatOp::inferResultType(int64_t dim, TypeRange inputTypes) {
522 assert(!inputTypes.empty() && "cannot concatenate 0 tensors"); in inferResultType() argument
524 llvm::to_vector<4>(llvm::map_range(inputTypes, [](Type type) { in inferResultType()
561 SmallVector<RankedTensorType> inputTypes; in verify()
563 inputTypes.push_back(cast<RankedTensorType>(input.getType())); in verify()
567 if (llvm::any_of(inputTypes, [resultRank](RankedTensorType type) { in verify()
573 if (llvm::any_of(inputTypes, [&](RankedTensorType type) { in verify()
587 for (auto tensorType : inputTypes) { in verify()
598 for (auto tensorType : inputTypes) in verify()
603 RankedTensorType::get(sizes, inputTypes[ in verify()
562 SmallVector<RankedTensorType> inputTypes; verify() local
[all...]
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp1555 SmallVector<Type> inputTypes; in matchAndRewrite()
1568 inputTypes.push_back(operand->get().getType()); in matchAndRewrite()
1614 warpInputType.push_back(inputTypes[i]); in matchAndRewrite()
1677 SmallVector<Type> inputTypes; matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp1739 TypeRange inputTypes, Type outputType) { in verifyNumBlockArgs()
1741 unsigned expectedNum = inputTypes.size(); in verifyNumBlockArgs()
1748 if (typ != inputTypes[i]) in verifyNumBlockArgs()
1732 verifyNumBlockArgs(T * op,Region & region,const char * regionName,TypeRange inputTypes,Type outputType) verifyNumBlockArgs() argument
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp6380 SmallVector<Type> inputTypes; parse() local
/llvm-project/mlir/include/mlir/Dialect/Tensor/IR/
H A DTensorOps.td174 static RankedTensorType inferResultType(int64_t dim, TypeRange inputTypes);