Use androidx FlakyTest annotation for device tests.

Platform_testing large test annotation is for
host side tests and will be restricted, device
tests should use androidx annotation.

Bug: 293837621
Test: presubmit
Change-Id: Ie7059450d2904ce424a6338069873020b78cc6dd