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*/)>