Lines Matching defs:numDims

41   AffineExprConstantFolder(unsigned numDims, ArrayRef<Attribute> operandConsts)
42       : numDims(numDims), operandConsts(operandConsts) {}
99               operandConsts[numDims +
119   unsigned numDims;
143     MLIRContext *ctx, unsigned numDims,
145   auto identityMap = getMultiDimIdentityMap(numDims, ctx);
148   llvm::SmallBitVector dropDimResults(numDims);
280 AffineMap AffineMap::getMultiDimMapWithTargets(unsigned numDims,
286   AffineMap result = AffineMap::get(/*dimCount=*/numDims, /*symbolCount=*/0,
334 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims,
337   dimExprs.reserve(numDims);
338   for (unsigned i = 0; i < numDims; ++i)
340   return get(/*dimCount=*/numDims, /*symbolCount=*/0, dimExprs, context);
349   for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) {
396   return map->numDims;
405   return map->numDims + map->numSymbols;
559   unsigned numDims = map.getNumDims();
562   SmallVector<AffineExpr, 8> newDims(numDims);
563   for (unsigned idx = 0; idx < numDims; ++idx) {
572       map.replaceDimsAndSymbols(newDims, newSymbols, numDims, numSymbols);
577   return AffineMap::get(numDims, numSymbols, exprs, map.getContext());
694   unsigned numDims = maps.front().getNumDims(),
697     assert(numDims == m.getNumDims() && numSymbols == m.getNumSymbols() &&
702       AffineMap::get(numDims, numSymbols, allExprs, maps.front().getContext()));
749   int64_t numDims = 0;
755       dimReplacements.push_back(b.getAffineDimExpr(numDims++));
769   return map.replaceDimsAndSymbols(dimReplacements, symReplacements, numDims,
840   unsigned numResults = 0, numDims = 0, numSymbols = 0;
850     numDims = std::max(m.getNumDims(), numDims);
852   return AffineMap::get(numDims, numSymbols, results, context);
901   int64_t numDims = (compress && isDim) ? newNumDimOrSym : map.getNumDims();
903   return AffineMap::get(numDims, numSyms, resultExprs, context);
931   unsigned numDims = maps[0].getNumDims();
932   llvm::SmallBitVector numDimsBitVector(numDims, true);
934     for (unsigned i = 0; i < numDims; ++i) {
967     : results(map.getResults()), numDims(map.getNumDims()),
972   numDims = map.getNumDims();
988     results[i] = simplifyAffineExpr(getResult(i), numDims, numSymbols);
993   return AffineMap::get(numDims, numSymbols, results, context);