Lines Matching defs:reassociation

1719   llvm_unreachable("could not find reassociation group");
1725                                 ArrayRef<ReassociationIndices> reassociation,
1728       inferExpandShapeOutputShape(b, loc, expandedType, reassociation,
1741                           ArrayRef<ReassociationIndices> reassociation,
1746         getReassociationIndicesAttribute(builder, reassociation),
1752                           ArrayRef<ReassociationIndices> reassociation) {
1757       builder, result.location, tensorResultTy, reassociation, inputShape);
1762   build(builder, result, tensorResultTy, src, reassociation,
1783     RankedTensorType type, SmallVector<ReassociationIndices> reassociation) {
1786                 type.getContext(), reassociation)));
1789 /// Compute the RankedTensorType obtained by applying `reassociation` to
1793                                     ArrayRef<AffineMap> reassociation) {
1796   newShape.reserve(reassociation.size());
1798   // Use the fact that reassociation is valid to simplify the logic: only use
1800   assert(isReassociationValid(reassociation) && "invalid reassociation");
1802   for (AffineMap m : reassociation) {
1819                             ArrayRef<ReassociationIndices> reassociation,
1824           convertReassociationIndicesToExprs(b.getContext(), reassociation)));
1826                       getReassociationIndicesAttribute(b, reassociation));
1983     // Find reassociation group that contains this result dimension.
2028     // Get reassociation group of the result dimension.