Use androidx Flaky annotation for device tests.

Platform_testing flaky annotation is for
host side tests and will be moved, device
tests should use androidx annotation.

Bug: 293837621
Test: presubmit
Change-Id: I3eb1ec225bcdfa5eb24cd98f2d1f73caeb02fb93