Update ops-related pbtxt files.

PiperOrigin-RevId: 632644501
tensorflow · May 11, 2024 · 02fa87f · 02fa87f
1 parent c56a5c7
commit 02fa87f
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt
@@ -1,4 +1,4 @@
-go/debugproto    
+go/nodeserialize   
 op {
   name: "Abort"
   attr {
@@ -22416,6 +22416,13 @@ op {
     minimum: 1
   }
 }
+op {
+  name: "GetTpuTaskId"
+  output_arg {
+    name: "tpu_task_id"
+    type: DT_INT32
+  }
+}
 op {
   name: "GlobalIterId"
   output_arg {
@@ -65954,6 +65961,21 @@ op {
     type: DT_VARIANT
   }
 }
+op {
+  name: "UpdateTaskIdAndGlobalCoreArray"
+  input_arg {
+    name: "tpu_task_id_to_shard_id"
+    type: DT_INT32
+    number_attr: "task_count"
+  }
+  attr {
+    name: "task_count"
+    type: "int"
+    has_minimum: true
+    minimum: 1
+  }
+  is_stateful: true
+}
 op {
   name: "UpperBound"
   input_arg {