Add missing ROCm dependencies in xla/tests
Add missing ROCm dependencies in xla/tests

All the changed targets include `rocm_config.h`,
therefore they need to depend on `:rocm_headers`
directly to pass the layering check.
