From cd537206c2bdaa928e6fa7154cc7ca485bb386cc Mon Sep 17 00:00:00 2001
From: Google APIs <noreply@google.com>
Date: Thu, 9 Feb 2023 13:05:39 -0800
Subject: [PATCH] chore: adding imports for types in LRO annotations

PiperOrigin-RevId: 508454659
---
 google/cloud/tpu/v2alpha1/cloud_tpu.proto | 31 +++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/google/cloud/tpu/v2alpha1/cloud_tpu.proto b/google/cloud/tpu/v2alpha1/cloud_tpu.proto
index e3a9ab2460..f4967a8014 100644
--- a/google/cloud/tpu/v2alpha1/cloud_tpu.proto
+++ b/google/cloud/tpu/v2alpha1/cloud_tpu.proto
@@ -22,6 +22,7 @@ import "google/api/field_behavior.proto";
 import "google/api/resource.proto";
 import "google/longrunning/operations.proto";
 import "google/protobuf/duration.proto";
+import "google/protobuf/empty.proto";
 import "google/protobuf/field_mask.proto";
 import "google/protobuf/timestamp.proto";
 import "google/rpc/status.proto";
@@ -519,6 +520,9 @@ message Node {
   // Node.
   string queued_resource = 43 [(google.api.field_behavior) = OUTPUT_ONLY];
 
+  // The AccleratorConfig for the TPU Node.
+  AcceleratorConfig accelerator_config = 44;
+
   // Shielded Instance options.
   ShieldedInstanceConfig shielded_instance_config = 45;
 }
@@ -943,6 +947,9 @@ message AcceleratorType {
 
   // The accelerator type.
   string type = 2;
+
+  // The accelerator config.
+  repeated AcceleratorConfig accelerator_configs = 3;
 }
 
 // Request for
@@ -1163,6 +1170,30 @@ message SimulateMaintenanceEventRequest {
   repeated string worker_ids = 2;
 }
 
+// A TPU accelerator configuration.
+message AcceleratorConfig {
+  // TPU type.
+  enum Type {
+    // Unspecified version.
+    TYPE_UNSPECIFIED = 0;
+
+    // TPU v2.
+    V2 = 2;
+
+    // TPU v3.
+    V3 = 4;
+
+    // TPU v4.
+    V4 = 7;
+  }
+
+  // Required. Type of TPU.
+  Type type = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Required. Topology of TPU in chips.
+  string topology = 2 [(google.api.field_behavior) = REQUIRED];
+}
+
 // A set of Shielded Instance options.
 message ShieldedInstanceConfig {
   // Defines whether the instance has Secure Boot enabled.