Lines Matching defs:dimsPos

569 /// Project dimsPos to the inner-most non-unit dim pos with reassocIndices.
571 /// For example, given dimsPos [0, 2], reassocIndices [[0, 1], [2, 3]], and
579 projectToInnerMostNonUnitDimsPos(ArrayRef<int64_t> dimsPos,
583   for (auto pos : dimsPos) {
598 /// Check if all dims in dimsPos are divisible by the corresponding tile sizes.
599 static bool isDimsDivisibleByTileSizes(ArrayRef<int64_t> dimsPos,
602   for (auto [pos, tileSize] : llvm::zip_equal(dimsPos, tileSizes)) {
711 /// Project dimsPos to their collapsed positions in the reassocIndices.
713 /// For example, given dimsPos [0, 1, 2, 4], and matching reassocIndices
718 projectDimsPosIntoReassocPos(ArrayRef<int64_t> dimsPos,
723   for (auto pos : dimsPos) {
734   assert(projectedPos.size() == dimsPos.size() && "Invalid dim pos projection");