Lines Matching defs:OpFoldResult

43   OpFoldResult offset;
44 SmallVector<OpFoldResult> sizes;
45 SmallVector<OpFoldResult> strides;
79 SmallVector<OpFoldResult> strides;
80 SmallVector<OpFoldResult> subStrides = subview.getMixedStrides();
84 SmallVector<OpFoldResult> values(2 * sourceRank + 1);
92 SmallVector<OpFoldResult> subOffsets = subview.getMixedOffsets();
98 OpFoldResult origStride =
101 : OpFoldResult(rewriter.getIndexAttr(sourceStrides[i]));
115 OpFoldResult finalOffset =
138 SmallVector<OpFoldResult> subSizes = subview.getMixedSizes();
141 SmallVector<OpFoldResult> finalSizes;
144 SmallVector<OpFoldResult> finalStrides;
268 static SmallVector<OpFoldResult>
270 ArrayRef<OpFoldResult> origSizes, unsigned groupId) {
277 SmallVector<OpFoldResult> expandedSizes(groupSize);
332 SmallVector<OpFoldResult> getExpandedStrides(memref::ExpandShapeOp expandShape,
334 ArrayRef<OpFoldResult> origSizes,
335 ArrayRef<OpFoldResult> origStrides,
350 SmallVector<OpFoldResult> expandedStrides(groupSize);
368 OpFoldResult origStride = ShapedType::isDynamic(strides[groupId])
380 OpFoldResult origSize = origSizes[groupId];
408 /// Produce an OpFoldResult object with \p builder at \p loc representing
415 static OpFoldResult
418 ArrayRef<OpFoldResult> values,
421 SmallVector<OpFoldResult> inputValues;
453 static SmallVector<OpFoldResult>
455 ArrayRef<OpFoldResult> origSizes, unsigned groupId) {
456 SmallVector<OpFoldResult> collapsedSize;
494 static SmallVector<OpFoldResult>
496 ArrayRef<OpFoldResult> origSizes,
497 ArrayRef<OpFoldResult> origStrides, unsigned groupId) {
508 SmallVector<OpFoldResult> groupStrides;
511 OpFoldResult lastValidStride = nullptr;
564 function_ref<SmallVector<OpFoldResult>(
566 ArrayRef<OpFoldResult> /*origSizes*/, unsigned /*groupId*/)>
568 function_ref<SmallVector<OpFoldResult>(
570 ArrayRef<OpFoldResult> /*origSizes*/,
571 ArrayRef<OpFoldResult> /*origStrides*/, unsigned /*groupId*/)>
588 OpFoldResult offsetOfr =
595 SmallVector<OpFoldResult> ones(reshapeRank, rewriter.getIndexAttr(1));
600 SmallVector<OpFoldResult> finalSizes;
602 SmallVector<OpFoldResult> finalStrides;
606 SmallVector<OpFoldResult> origSizes =
608 SmallVector<OpFoldResult> origStrides =
612 SmallVector<OpFoldResult> reshapedSizes =
614 SmallVector<OpFoldResult> reshapedStrides = getReshapedStrides(
650 SmallVector<OpFoldResult> (*getReshapedSizes)(
652 ArrayRef<OpFoldResult> /*origSizes*/, unsigned /*groupId*/),
653 SmallVector<OpFoldResult> (*getReshapedStrides)(
655 ArrayRef<OpFoldResult> /*origSizes*/,
656 ArrayRef<OpFoldResult> /*origStrides*/, unsigned /*groupId*/)>
797 SmallVector<OpFoldResult> sizes;
808 SmallVector<OpFoldResult> strides(rank, rewriter.getIndexAttr(1));
815 ArrayRef<OpFoldResult> sizesInvolvedInStride(&sizes[i + 1], symbolNumber);
841 for (OpFoldResult size : sizes)
844 for (OpFoldResult stride : strides)
980 SmallVector<OpFoldResult> results;
997 SmallVector<OpFoldResult> sizes = reinterpretCastOp.getMixedSizes();
998 SmallVector<OpFoldResult> strides = reinterpretCastOp.getMixedStrides();
1054 SmallVector<OpFoldResult> results;
1065 OpFoldResult ofr) -> OpFoldResult {
1067 ? OpFoldResult(rewriter.getIndexAttr(constant))
1082 SmallVector<OpFoldResult> sizes = newExtractStridedMetadata.getSizes();
1083 SmallVector<OpFoldResult> strides = newExtractStridedMetadata.getStrides();