Use androidx FlakyTest annotation for device tests.

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

Bug: 293837621
Test: presubmit
Change-Id: I835c4c837fa2074c937d7068f17c75467fb213dc