Lines Matching refs:reassociation
669 static bool isReassociationValid(ArrayRef<AffineMap> reassociation, in isReassociationValid() argument
671 if (reassociation.empty()) in isReassociationValid()
673 unsigned nDims = reassociation[0].getNumDims(); in isReassociationValid()
675 for (auto it : llvm::enumerate(reassociation)) { in isReassociationValid()
693 *invalidIndex = reassociation.size() - 1; in isReassociationValid()
728 ArrayRef<AffineMap> reassociation) { in computeReshapeCollapsedType() argument
737 newSizes.reserve(reassociation.size()); in computeReshapeCollapsedType()
739 newStrides.reserve(reassociation.size()); in computeReshapeCollapsedType()
743 assert(isReassociationValid(reassociation) && "invalid reassociation"); in computeReshapeCollapsedType()
745 for (AffineMap m : reassociation) { in computeReshapeCollapsedType()
809 getSymbolLessAffineMaps(ArrayRef<ReassociationExprs> reassociation) { in getSymbolLessAffineMaps() argument
810 unsigned maxDim = getMaxPosOfType<AffineDimExpr>(reassociation); in getSymbolLessAffineMaps()
811 assert(getMaxPosOfType<AffineSymbolExpr>(reassociation) == 0 && in getSymbolLessAffineMaps()
814 maps.reserve(reassociation.size()); in getSymbolLessAffineMaps()
815 for (const auto &exprs : reassociation) { in getSymbolLessAffineMaps()
838 ArrayRef<ReassociationExprs> reassociation, in build() argument
840 auto maps = getSymbolLessAffineMaps(reassociation); in build()
850 ArrayRef<ReassociationExprs> reassociation, in build() argument
852 auto maps = getSymbolLessAffineMaps(reassociation); in build()
889 if (collapsedRank != op.reassociation().size()) in verifyReshapeLikeTypes()
892 << op.reassociation().size() << ")"; in verifyReshapeLikeTypes()
893 auto maps = getAffineMaps(op.reassociation()); in verifyReshapeLikeTypes()
910 auto maps = getAffineMaps(op.reassociation()); in verify()
930 ArrayRef<AffineMap> reassociation) { in computeTensorReshapeCollapsedType() argument
933 newShape.reserve(reassociation.size()); in computeTensorReshapeCollapsedType()
937 assert(isReassociationValid(reassociation) && "invalid reassociation"); in computeTensorReshapeCollapsedType()
939 for (AffineMap m : reassociation) { in computeTensorReshapeCollapsedType()
957 ArrayRef<ReassociationExprs> reassociation, in build() argument
959 auto maps = getSymbolLessAffineMaps(reassociation); in build()
969 ArrayRef<ReassociationExprs> reassociation, in build() argument
971 auto maps = getSymbolLessAffineMaps(reassociation); in build()
981 auto maps = getAffineMaps(op.reassociation()); in verify()