[PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.

The two are aliases and the former is a forwarding header pointing to the latter.

Cleanup only, no functional changes.

PiperOrigin-RevId: 621341188

Author: hawkinsp

examples/jax_cpp/BUILD

@@ -27,7 +27,7 @@ cc_binary(
         "@xla//xla:status",
         "@xla//xla:statusor",
         "@xla//xla/pjrt:pjrt_client",
-        "@xla//xla/pjrt:tfrt_cpu_pjrt_client",
+        "@xla//xla/pjrt/cpu:cpu_client",
         "@xla//xla/service:hlo_proto_cc",
         "@xla//xla/tools:hlo_module_loader",
         "@tsl//tsl/platform:logging",

examples/jax_cpp/main.cc

@@ -27,10 +27,10 @@ limitations under the License.
 //
 // To load and run the HloModule,
 //
-// $ bazel build examples/jax_cpp:main --experimental_repo_remote_exec --check_visibility=false
-// $ bazel-bin/examples/jax_cpp/main
-// 2021-01-12 15:35:28.316880: I examples/jax_cpp/main.cc:65] result = (
-// f32[2,2] {
+// $ bazel build examples/jax_cpp:main --experimental_repo_remote_exec \
+//    --check_visibility=false
+// $ bazel-bin/examples/jax_cpp/main 2021-01-12
+// 15:35:28.316880: I examples/jax_cpp/main.cc:65] result = ( f32[2,2] {
 //   { 1.5, 1.5 },
 //   { 3.5, 3.5 }
 // }
@@ -42,8 +42,8 @@ limitations under the License.
 
 #include "xla/literal.h"
 #include "xla/literal_util.h"
+#include "xla/pjrt/cpu/cpu_client.h"
 #include "xla/pjrt/pjrt_client.h"
-#include "xla/pjrt/tfrt_cpu_pjrt_client.h"
 #include "xla/status.h"
 #include "xla/statusor.h"
 #include "xla/tools/hlo_module_loader.h"