From 3b4f970c3b32d7900650fd238024ad2dfd04c613 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 18 Feb 2021 16:36:49 -0800
Subject: [PATCH] Update ops-related pbtxt files.

PiperOrigin-RevId: 358292541
Change-Id: Ieb03481abb80ee0e0c0e295fcf4e0726d27f4e02
---
 .../ops_history_v2/TPUReshardVariables.pbtxt  | 22 +++++++++++++++++++
 tensorflow/core/ops/ops.pbtxt                 | 22 +++++++++++++++++++
 2 files changed, 44 insertions(+)
 create mode 100644 tensorflow/core/ops/compat/ops_history_v2/TPUReshardVariables.pbtxt

diff --git a/tensorflow/core/ops/compat/ops_history_v2/TPUReshardVariables.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/TPUReshardVariables.pbtxt
new file mode 100644
index 00000000000..fecd05d06eb
--- /dev/null
+++ b/tensorflow/core/ops/compat/ops_history_v2/TPUReshardVariables.pbtxt
@@ -0,0 +1,22 @@
+op {
+  name: "TPUReshardVariables"
+  input_arg {
+    name: "vars"
+    type: DT_RESOURCE
+    number_attr: "N"
+  }
+  input_arg {
+    name: "new_format_key"
+    type: DT_STRING
+  }
+  input_arg {
+    name: "format_state_var"
+    type: DT_RESOURCE
+  }
+  attr {
+    name: "N"
+    type: "int"
+    has_minimum: true
+  }
+  is_stateful: true
+}
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt
index d892e75263f..69f1f47b3c0 100644
--- a/tensorflow/core/ops/ops.pbtxt
+++ b/tensorflow/core/ops/ops.pbtxt
@@ -54108,6 +54108,28 @@ op {
     type: "type"
   }
 }
+op {
+  name: "TPUReshardVariables"
+  input_arg {
+    name: "vars"
+    type: DT_RESOURCE
+    number_attr: "N"
+  }
+  input_arg {
+    name: "new_format_key"
+    type: DT_STRING
+  }
+  input_arg {
+    name: "format_state_var"
+    type: DT_RESOURCE
+  }
+  attr {
+    name: "N"
+    type: "int"
+    has_minimum: true
+  }
+  is_stateful: true
+}
 op {
   name: "TakeDataset"
   input_arg {