chore: adding imports for types in LRO annotations

Google APIs · copybara-github · commit cd537206c2bd · 2023-02-09T13:06:22.000-08:00
PiperOrigin-RevId: 508454659
diff --git 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.
