message: Enable remote functions for TPU_SYSTEM.
diff: @@ -144,7 +144,10 @@ Status ProcessFunctionLibraryRuntime::GetDeviceContext(
  }
  Device* device = flr->device();
  string device_type = device->parsed_name().type;
  - if (device_type == "CPU") return Status::OK();
  + if (device_type == "CPU" || device_type == "TPU_SYSTEM") {
  + // "TPU_SYSTEM" indicates that `device` is a CPU.
  + return Status::OK();
  + }
  if (device_type == "GPU") {
  auto* dev_info = flr->device()->tensorflow_gpu_device_info();
  if (dev_info) {
  