Lines Matching defs:fold

467 OpFoldResult MultiDimReductionOp::fold(FoldAdaptor adaptor) {
1250 OpFoldResult vector::ExtractElementOp::fold(FoldAdaptor adaptor) {
1358         "incorrect fold/rewrite");
1427   /// Iterate over producing insert and transpose ops until we find a fold.
1428   Value fold();
1481   /// // can fold to vector.extract %source[0, 3]
1489   /// Try to fold in place to extract(source, extractPosition) and return the
1553   // Case 2.a. early-exit fold.
1577   // Case 3.a. early-exit fold (break and delegate to post-while path).
1583 /// Try to fold in place to extract(source, extractPosition) and return the
1592   // If we can't fold (either internal transposition, or nothing to fold), bail.
1597   // Otherwise, fold by updating the op inplace and return its result.
1605 /// Iterate over producing insert and transpose ops until we find a fold.
1606 Value ExtractFromInsertTransposeChainState::fold() {
1643   // If after all this we can fold, go for it.
1950 /// Try to fold the extraction of a scalar from a vector defined by
1955 /// ==> fold to %a
1979   // Compute flattened/linearized index and fold to operand.
2000 OpFoldResult ExtractOp::fold(FoldAdaptor adaptor) {
2002   // Note: Do not fold "vector.extract %v[] : f32 from vector<f32>" (type
2011   if (auto res = ExtractFromInsertTransposeChainState(*this).fold())
2559 OpFoldResult BroadcastOp::fold(FoldAdaptor adaptor) {
2675 OpFoldResult vector::ShuffleOp::fold(FoldAdaptor adaptor) {
2682   // fold shuffle V1, V2, [0, 1, 2, 3] : <4xi32>, <2xi32> -> V1
2686   // fold shuffle V1, V2, [4, 5] : <4xi32>, <2xi32> -> V2
2840 OpFoldResult vector::InsertElementOp::fold(FoldAdaptor adaptor) {
3066 OpFoldResult vector::InsertOp::fold(FoldAdaptor adaptor) {
3068   // %v. Note: Do not fold "vector.insert %v, %dest [] : f32 into vector<f32>"
3382 OpFoldResult InsertStridedSliceOp::fold(FoldAdaptor adaptor) {
3662       // fold.
3669 OpFoldResult ExtractStridedSliceOp::fold(FoldAdaptor adaptor) {
4419 OpFoldResult TransferReadOp::fold(FoldAdaptor) {
4809 LogicalResult TransferWriteOp::fold(FoldAdaptor adaptor,
5049 OpFoldResult LoadOp::fold(FoldAdaptor) {
5082 LogicalResult StoreOp::fold(FoldAdaptor adaptor,
5135 OpFoldResult MaskedLoadOp::fold(FoldAdaptor) {
5186 LogicalResult MaskedStoreOp::fold(FoldAdaptor adaptor,
5545 OpFoldResult ShapeCastOp::fold(FoldAdaptor adaptor) {
5637 /// to fold shape_cast into create_mask.
5799 OpFoldResult BitCastOp::fold(FoldAdaptor adaptor) {
5931 OpFoldResult vector::TransposeOp::fold(FoldAdaptor adaptor) {
6250         // Must be all-true to fold to a ConstantMask.
6475 LogicalResult MaskOp::fold(FoldAdaptor adaptor,
6593 OpFoldResult SplatOp::fold(FoldAdaptor adaptor) {
6724 /// usually able to match op + select patterns and fold them into a native