The functionality as you describe still exists. So go ahead, change the setting back and make your All mailbox how you want it for yourself. It’s a one-and-done in your settings. Just tap the link they have in the announcement, sign in and you’ll be on the right page. That setting does reflect properly in the Android app, even if you aren’t able to toggle it there at this time.