Mislim da je to maknuto još u 9-ki zajedno s opcijom za isključivanje mobilnih podataka u sleepu. Ne znam kako je kod tebe, na Huaweiju na 10-ki imam opciju za isključiti vezu u sleepu, ali mislim da to obuhvaća i mobile data i WiFi.
Ako hoćeš, možeš se probati poigrati ADB-om, prvo komandom "adb shell settings list global" vidiš trenutno stanje, a onda komandom "adb shell settings put global wifi_sleep_policy 1" mijenjaš da WiFi radi samo na punjaču. Ovo su moguće opcije:
Keep WiFi on during sleep: Always = 2
Keep WiFi on during sleep: Only When Plugged In = 1
Keep WiFi on during sleep: Never = 0