Use `:gtest` instead of `:gtest_main` in PJRT API test libraries
Use `:gtest` instead of `:gtest_main` in PJRT API test libraries
- This is to allow actual test definitions which depend on the test library to override the heap checker behavior.
