Lines Matching refs:numDims

30   AffineExprConstantFolder(unsigned numDims, ArrayRef<Attribute> operandConsts)  in AffineExprConstantFolder()  argument
31 : numDims(numDims), operandConsts(operandConsts) {} in AffineExprConstantFolder()
67 if (auto attr = operandConsts[numDims + in constantFoldImpl()
87 unsigned numDims; member in __anon1d589f0a0111::AffineExprConstantFolder
244 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims, in getMultiDimIdentityMap() argument
247 dimExprs.reserve(numDims); in getMultiDimIdentityMap()
248 for (unsigned i = 0; i < numDims; ++i) in getMultiDimIdentityMap()
250 return get(/*dimCount=*/numDims, /*symbolCount=*/0, dimExprs, context); in getMultiDimIdentityMap()
259 for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) { in isIdentity() local
296 return map->numDims; in getNumDims()
305 return map->numDims + map->numSymbols; in getNumInputs()
442 unsigned numDims = map.getNumDims(); in compose() local
445 SmallVector<AffineExpr, 8> newDims(numDims); in compose()
446 for (unsigned idx = 0; idx < numDims; ++idx) { in compose()
455 map.replaceDimsAndSymbols(newDims, newSymbols, numDims, numSymbols); in compose()
460 return AffineMap::get(numDims, numSymbols, exprs, map.getContext()); in compose()
545 unsigned numDims = 0; in compressDims() local
553 dimReplacements.push_back(getAffineDimExpr(numDims++, context)); in compressDims()
559 return AffineMap::get(numDims, map.getNumSymbols(), resultExprs, context); in compressDims()
573 unsigned numDims = maps.front().getNumDims(), in compressUnusedImpl() local
576 assert(numDims == m.getNumDims() && numSymbols == m.getNumSymbols() && in compressUnusedImpl()
581 AffineMap::get(numDims, numSymbols, allExprs, maps.front().getContext())); in compressUnusedImpl()
699 unsigned numResults = 0, numDims = 0, numSymbols = 0; in concatAffineMaps() local
709 numDims = std::max(m.getNumDims(), numDims); in concatAffineMaps()
711 return AffineMap::get(numDims, numSymbols, results, in concatAffineMaps()
721 unsigned numDims = maps[0].getNumDims(); in getUnusedDimsBitVector() local
722 llvm::SmallBitVector numDimsBitVector(numDims, true); in getUnusedDimsBitVector()
724 for (unsigned i = 0; i < numDims; ++i) { in getUnusedDimsBitVector()
738 numDims(map.getNumDims()), numSymbols(map.getNumSymbols()), in MutableAffineMap()
743 numDims = map.getNumDims(); in reset()
764 results[i] = simplifyAffineExpr(getResult(i), numDims, numSymbols); in simplify()
769 return AffineMap::get(numDims, numSymbols, results, context); in getAffineMap()