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: Ia5faeed2d4cec8c135ea37d5b0d45a338830fec0