feat: add TPU_V2 & TPU_V3 values to AcceleratorType in aiplatform v1/v1beta1 accelerator_type.proto

Google APIs · copybara-github · commit f873e7f132a5 · 2022-02-22T11:44:29.000-08:00
PiperOrigin-RevId: 430259767
diff --git a/google/cloud/aiplatform/v1/accelerator_type.proto b/google/cloud/aiplatform/v1/accelerator_type.proto
@@ -16,8 +16,6 @@ syntax = "proto3";
 
 package google.cloud.aiplatform.v1;
 
-import "google/api/annotations.proto";
-
 option csharp_namespace = "Google.Cloud.AIPlatform.V1";
 option go_package = "google.golang.org/genproto/googleapis/cloud/aiplatform/v1;aiplatform";
 option java_multiple_files = true;
@@ -48,4 +46,10 @@ enum AcceleratorType {
 
   // Nvidia Tesla A100 GPU.
   NVIDIA_TESLA_A100 = 8;
+
+  // TPU v2.
+  TPU_V2 = 6;
+
+  // TPU v3.
+  TPU_V3 = 7;
 }
diff --git a/google/cloud/aiplatform/v1beta1/accelerator_type.proto b/google/cloud/aiplatform/v1beta1/accelerator_type.proto
@@ -16,8 +16,6 @@ syntax = "proto3";
 
 package google.cloud.aiplatform.v1beta1;
 
-import "google/api/annotations.proto";
-
 option csharp_namespace = "Google.Cloud.AIPlatform.V1Beta1";
 option go_package = "google.golang.org/genproto/googleapis/cloud/aiplatform/v1beta1;aiplatform";
 option java_multiple_files = true;
@@ -48,4 +46,10 @@ enum AcceleratorType {
 
   // Nvidia Tesla A100 GPU.
   NVIDIA_TESLA_A100 = 8;
+
+  // TPU v2.
+  TPU_V2 = 6;
+
+  // TPU v3.
+  TPU_V3 = 7;
 }
