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: build
Change-Id: Ic3efbe5c44235226aa1d5db3f10175775a01e992