absl::StatusOr<llvm::SmallVector<ParsedDevice, 8>> GetTPUSystemDevices(
    ParsedDevices devices) {
  ParsedDevice spec;
  spec.type = kDeviceTPUSystem;
  spec.has_type = true;
  spec.id = 0;
  spec.has_id = true;

  llvm::SmallVector<ParsedDevice, 8> system_devices =
      FindMatchingDevices(devices, spec);
  if (system_devices.empty())
    return absl::InvalidArgumentError(

  const tensorflow::DeviceBase* device_base =
      reinterpret_cast<tensorflow::OpKernelContext*>(ctx)->device();
#if defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
  if (!device_base->parsed_name().has_id) return -1;
  return device_base->parsed_name().id;
#else
  const auto* device = reinterpret_cast<const tensorflow::Device*>(
      device_base->UnderlyingDevice());