Add highlight menu keys for all toggle pref controllers and custom slices

Bug: 204695404
Test: manual, robotest build pass, unit
Change-Id: Iad2e28ada8ddb766ffa7eff2c8dc581250cbd6f8

src/com/android/settings/display/AmbientDisplayAlwaysOnPreferenceController.java

@@ -67,6 +67,11 @@ public class AmbientDisplayAlwaysOnPreferenceController extends TogglePreference
         return true;
     }
 
+    @Override
+    public int getSliceHighlightMenuRes() {
+        return R.string.menu_key_display;
+    }
+
     @Override
     public boolean isChecked() {
         return getConfig().alwaysOnEnabled(MY_USER);