| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | StructuredOpsUtils.cpp | 24 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isRowMajorMatmul()
25 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isRowMajorMatmul()
50 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isColumnMajorMatmul()
51 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isColumnMajorMatmul()
76 map2.getNumResults() != 3 || map0.getNumInputs() != 4 || in isRowMajorBatchMatmul()
77 map1.getNumInputs() != 4 || map2.getNumInputs() != 4) { in isRowMajorBatchMatmul()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/IR/ |
| H A D | AffineOps.h | 126 getSrcMap().getNumInputs()}; in getSrcIndices()
136 return getSrcMemRefOperandIndex() + 1 + getSrcMap().getNumInputs(); in getDstMemRefOperandIndex()
165 getDstMap().getNumInputs()}; in getDstIndices()
170 return getDstMemRefOperandIndex() + 1 + getDstMap().getNumInputs(); in getTagMemRefOperandIndex()
194 getTagMap().getNumInputs()}; in getTagIndices()
200 getTagMap().getNumInputs()); in getNumElements()
252 getTagMemRefOperandIndex() + 1 + getTagMap().getNumInputs() + 1; in isStrided()
311 operand_begin() + 1 + getTagMap().getNumInputs()}; in getTagIndices()
328 Value getNumElements() { return getOperand(1 + getTagMap().getNumInputs()); } in getNumElements()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | AffineMap.cpp | 160 getNumResults() < getNumInputs() ? getNumInputs() - getNumResults() : 0; in isPermutationOfMinorIdentityWithBroadcasting()
168 getNumResults() > getNumInputs() ? getNumResults() - getNumInputs() : 0; in isPermutationOfMinorIdentityWithBroadcasting()
169 llvm::SmallBitVector dimFound(std::max(getNumInputs(), getNumResults()), in isPermutationOfMinorIdentityWithBroadcasting()
303 unsigned AffineMap::getNumInputs() const { in getNumInputs() function in AffineMap
352 assert(getNumInputs() == operandConstants.size()); in partialConstantFold()
484 if (getNumResults() > getNumInputs()) in isProjectedPermutation()
487 SmallVector<bool, 8> seen(getNumInputs(), false); in isProjectedPermutation()
672 if (seenExprs.size() != map.getNumInputs()) in inversePermutation()
682 SmallVector<AffineExpr, 4> exprs(map.getNumInputs(), zero); in inverseAndBroadcastProjectedPermutation()
|
| H A D | IntegerSet.cpp | 17 unsigned IntegerSet::getNumInputs() const { in getNumInputs() function in IntegerSet
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 1500 if (getLowerBoundMap().getNumInputs() > 0) in verifyRegions()
1505 if (getUpperBoundMap().getNumInputs() > 0) in verifyRegions()
1735 return getNumOperands() - lbMap.getNumInputs() - ubMap.getNumInputs(); in getNumIterOperands()
1996 lbMap.getNumInputs() + ubMap.getNumInputs(), ubMap); in getUpperBound()
2000 assert(lbOperands.size() == map.getNumInputs()); in setLowerBound()
2015 assert(ubOperands.size() == map.getNumInputs()); in setUpperBound()
2076 getUpperBoundMap().getNumInputs()}; in getUpperBoundOperands()
2091 unsigned numOperands = lbMap.getNumInputs(); in matchingBoundOperandList()
2365 if (getNumOperands() != condition.getNumInputs()) in verify()
2612 if (map.getNumInputs() != numIndexOperands) in verifyMemoryOpIndexing()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Func/IR/ |
| H A D | FuncOps.cpp | 138 if (fnType.getNumInputs() != getNumOperands()) in verifySymbolUses()
141 for (unsigned i = 0, e = fnType.getNumInputs(); i != e; ++i) in verifySymbolUses()
255 assert(type.getNumInputs() == argAttrs.size()); in build()
309 unsigned oldNumArgs = oldType.getNumInputs(); in clone()
|
| /llvm-project-15.0.7/mlir/include/mlir/Analysis/Presburger/ |
| H A D | PWMAFunction.h | 52 unsigned getNumInputs() const { return domainSet.getNumDimAndSymbolVars(); } in getNumInputs() function
153 unsigned getNumInputs() const { return space.getNumVars(); } in getNumInputs() function
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.td | 308 /*methodName=*/"getNumInputs",
361 int64_t numInputs = getNumInputs();
380 assert(i >= 0 && i < getNumInputs());
394 result.reserve(getNumInputs());
413 result.reserve(getNumInputs());
455 return &this->getOperation()->getOpOperand(getNumInputs() + i);
468 this->getOperation()->setOperand(getNumInputs() + i, value);
596 if (opOperand->getOperandNumber() < $_op.getNumInputs())
612 if (opOperand->getOperandNumber() >= $_op.getNumInputs())
743 return *(indexingMaps.begin() + getNumInputs() +
[all …]
|
| /llvm-project-15.0.7/flang/unittests/Optimizer/ |
| H A D | RTBuilder.cpp | 31 EXPECT_EQ(c99_cacosf_funcTy.getNumInputs(), 1u); in TEST()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Loops.cpp | 36 assert(map.getNumInputs() == vals.size()); in makeCanonicalAffineApplies()
281 if (map.getNumResults() != 1 || map.getNumInputs() > 1) in matchAndRewrite()
285 if (map.getNumInputs() == 0) { in matchAndRewrite()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 41 newInputBuffers.reserve(op.getNumInputs()); in bufferizeLinalgOp()
|
| H A D | ElementwiseOpFusion.cpp | 187 producerBlock.getArguments().take_front(producer.getNumInputs())) in generateFusedElementwiseOpRegion()
195 .drop_front(producer.getNumInputs()) in generateFusedElementwiseOpRegion()
203 .take_front(consumer.getNumInputs()) in generateFusedElementwiseOpRegion()
717 expandedOpOperands.reserve(genericOp.getNumInputs()); in fuseWithReshapeByExpansion()
1538 fusedOperands.reserve(genericOp.getNumInputs()); in matchAndRewrite()
1539 fusedLocs.reserve(fusedLocs.size() + genericOp.getNumInputs()); in matchAndRewrite()
|
| H A D | DropUnitDims.cpp | 431 resultTypes.push_back(newInputOutputTypes[i + genericOp.getNumInputs()]); in matchAndRewrite()
443 unsigned index = result.index() + replacementOp.getNumInputs(); in matchAndRewrite()
|
| H A D | Vectorization.cpp | 147 outputOperand->getOperandNumber() - linalgOp.getNumInputs(); in matchLinalgReduction()
369 if (!arg || arg.getArgNumber() < linalgOp.getNumInputs()) in vectorizeOneOp()
374 arg.getArgNumber() - linalgOp.getNumInputs(), reductionOps); in vectorizeOneOp()
474 if (opOperand->getOperandNumber() < linalgOp.getNumInputs()) { in vectorizeAsLinalgGeneric()
1322 if (linalgOp.getNumInputs() != 2 || linalgOp.getNumOutputs() != 1) in Conv1DNwcGenerator()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | AffineStructures.cpp | 479 assert(flatExpr.size() >= other.getNumInputs() + 1); in composeMatchingMap()
486 for (unsigned i = 0, f = other.getNumInputs(); i < f; i++) { in composeMatchingMap()
493 for (unsigned i = other.getNumInputs(); i < end; i++, j++) { in composeMatchingMap()
654 assert(!lbMap || lbMap.getNumInputs() == operands.size()); in addDomainFromSliceMaps()
655 assert(!ubMap || ubMap.getNumInputs() == operands.size()); in addDomainFromSliceMaps()
1196 for (unsigned j = 0, e = boundMap.getNumInputs(); j < e; j++) { in addBound()
1208 for (unsigned i = boundMap.getNumInputs(); i < end; i++, j++) { in addBound()
1232 assert(map.getNumInputs() == operands.size() && "number of inputs mismatch"); in computeAlignedMap()
1299 assert(!lbMap || lbMap.getNumInputs() == operands.size()); in addSliceBounds()
1300 assert(!ubMap || ubMap.getNumInputs() == operands.size()); in addSliceBounds()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgInterfaces.cpp | 121 if (linalgOp.getNumInputs() != 2 || linalgOp.getNumOutputs() != 1) in isContractionInterfaceImpl()
280 if (linalgOp.getNumInputs() < 2 || linalgOp.getNumOutputs() != 1) in isConvolutionInterfaceImpl()
445 if (linalgOp.getNumInputs() != 1 || linalgOp.getNumOutputs() != 1) in isFillInterfaceImpl()
619 int64_t numInputs = linalgOp.getNumInputs(); in verifyStructuredOpInterface()
714 << (opOperand->getOperandNumber() - linalgOp.getNumInputs()) in verifyStructuredOpInterface()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | IntegerSet.h | 82 unsigned getNumInputs() const;
|
| H A D | AffineMap.h | 160 unsigned getNumInputs() const;
550 assert(map.getNumInputs() == source.size()); in applyPermutationMap()
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | IntegerSet.cpp | 82 return static_cast<intptr_t>(unwrap(set).getNumInputs()); in mlirIntegerSetGetNumInputs()
|
| H A D | AffineMap.cpp | 111 return unwrap(affineMap).getNumInputs(); in mlirAffineMapGetNumInputs()
|
| /llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/ |
| H A D | TestLinalgFusionTransforms.cpp | 59 if (opOperand->getOperandNumber() >= linalgOp.getNumInputs()) in fuseLinalgOpsGreedily()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/ |
| H A D | GPUToSPIRV.cpp | 193 if (!argABIInfo.empty() && fnType.getNumInputs() != argABIInfo.size()) { in lowerAsEntryFunction()
202 TypeConverter::SignatureConversion signatureConverter(fnType.getNumInputs()); in lowerAsEntryFunction()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Func/Transforms/ |
| H A D | DecomposeCallGraphTypes.cpp | 70 TypeConverter::SignatureConversion conversion(functionType.getNumInputs()); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/LinalgToSPIRV/ |
| H A D | LinalgToSPIRV.cpp | 73 if (genericOp.getNumInputs() != 1 || genericOp.getNumOutputs() != 1) in matchAsPerformingReduction()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | BufferResultsToOutParams.cpp | 73 func.setArgAttrs(functionType.getNumInputs() + i, in updateFuncOp()
|