if (perm_index != match_perm_index) {
        return failure();
      }
    }

    auto reshape_op = op.getInput().getDefiningOp<ReshapeOp>();
    if (!reshape_op ||
        !InsertOneInSecondInnermostDim(reshape_op.getInput().getType(),
                                       reshape_op.getType())) {
      return failure();
    }