Lines Matching refs:groups

93   // Traverse all reassociation groups to determine the appropriate indices
95   for (ArrayRef<int64_t> groups : expandShapeOp.getReassociationIndices()) {
96     assert(!groups.empty() && "association indices groups cannot be empty");
99     int64_t groupSize = groups.size();
105       sizesVal[i] = (*outputShape)[groups[i]];
127       dynamicIndices[i] = indices[groups[i]];
173   for (ArrayRef<int64_t> groups : collapseShapeOp.getReassociationIndices()) {
174     assert(!groups.empty() && "association indices groups cannot be empty");
176     int64_t groupSize = groups.size();
184       sizes[i] = collapseShapeOp.getSrcType().getDimSize(groups[i]);