Lines Matching defs:groupId

254 /// \p groupId-th reassociation group.
259 ///     baseSizes#groupId / product(expandShapeSizes#j,
261 /// Where reassIdx#i is the reassociation index at index i in \p groupId.
263 /// \post result.size() == expandShape.getReassociationIndices()[groupId].size()
270                  ArrayRef<OpFoldResult> origSizes, unsigned groupId) {
272       expandShape.getReassociationIndices()[groupId];
302         origSizes[groupId]);
309 /// \p groupId-th reassociation group.
319 /// Where reassIdx#i is the reassociation index for at index i in \p groupId
327 /// \post result.size() == expandShape.getReassociationIndices()[groupId].size()
336                                              unsigned groupId) {
338       expandShape.getReassociationIndices()[groupId];
368   OpFoldResult origStride = ShapedType::isDynamic(strides[groupId])
369                                 ? origStrides[groupId]
370                                 : builder.getIndexAttr(strides[groupId]);
378     assert(ShapedType::isDynamic(sourceType.getDimSize(groupId)) &&
380     OpFoldResult origSize = origSizes[groupId];
440 /// \p groupId-th reassociation group.
445 /// `prod(origSizes#i, for i in {ressociationGroup[groupId]})`.
455                  ArrayRef<OpFoldResult> origSizes, unsigned groupId) {
460   uint64_t size = collapseShapeType.getDimSize(groupId);
473       collapseShape.getReassociationIndices()[groupId];
483 /// \p groupId-th reassociation group.
497                    ArrayRef<OpFoldResult> origStrides, unsigned groupId) {
499       collapseShape.getReassociationIndices()[groupId];
532     int64_t finalStride = collapsedStrides[groupId];
566         ArrayRef<OpFoldResult> /*origSizes*/, unsigned /*groupId*/)>
571         ArrayRef<OpFoldResult> /*origStrides*/, unsigned /*groupId*/)>
613         getReshapedSizes(reshape, rewriter, origSizes, /*groupId=*/idx);
615         reshape, rewriter, origSizes, origStrides, /*groupId=*/idx);
652               ArrayRef<OpFoldResult> /*origSizes*/, unsigned /*groupId*/),
656               ArrayRef<OpFoldResult> /*origStrides*/, unsigned /*groupId*/)>