From 331ce4d2032bd9942783178a274b2cd2ff85b178 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Mon, 6 Jul 2020 17:18:38 -0700
Subject: [PATCH] Update ops-related pbtxt files.

PiperOrigin-RevId: 319887474
Change-Id: I03c2bd78d98491ab55a8d17deb0a9bf572f91015
---
 .../ops_history_v2/TPUReplicateMetadata.pbtxt | 82 +++++++++++++++++++
 tensorflow/core/ops/ops.pbtxt                 |  7 ++
 2 files changed, 89 insertions(+)

diff --git a/tensorflow/core/ops/compat/ops_history_v2/TPUReplicateMetadata.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/TPUReplicateMetadata.pbtxt
index 1d1b1bad889..6feb35cc3c9 100644
--- a/tensorflow/core/ops/compat/ops_history_v2/TPUReplicateMetadata.pbtxt
+++ b/tensorflow/core/ops/compat/ops_history_v2/TPUReplicateMetadata.pbtxt
@@ -202,3 +202,85 @@ op {
     }
   }
 }
+op {
+  name: "TPUReplicateMetadata"
+  attr {
+    name: "num_replicas"
+    type: "int"
+    has_minimum: true
+  }
+  attr {
+    name: "num_cores_per_replica"
+    type: "int"
+    default_value {
+      i: 1
+    }
+  }
+  attr {
+    name: "topology"
+    type: "string"
+    default_value {
+      s: ""
+    }
+  }
+  attr {
+    name: "use_tpu"
+    type: "bool"
+    default_value {
+      b: true
+    }
+  }
+  attr {
+    name: "device_assignment"
+    type: "list(int)"
+    default_value {
+      list {
+      }
+    }
+  }
+  attr {
+    name: "computation_shape"
+    type: "list(int)"
+    default_value {
+      list {
+      }
+    }
+  }
+  attr {
+    name: "host_compute_core"
+    type: "list(string)"
+    default_value {
+      list {
+      }
+    }
+  }
+  attr {
+    name: "padding_map"
+    type: "list(string)"
+    default_value {
+      list {
+      }
+    }
+  }
+  attr {
+    name: "step_marker_location"
+    type: "string"
+    default_value {
+      s: "STEP_MARK_AT_ENTRY"
+    }
+  }
+  attr {
+    name: "allow_soft_placement"
+    type: "bool"
+    default_value {
+      b: false
+    }
+  }
+  attr {
+    name: "use_spmd_for_xla_partitioning"
+    type: "bool"
+    default_value {
+      b: false
+    }
+  }
+}
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt
index b9e6dd8d0c6..c9ad5aa2dc6 100644
--- a/tensorflow/core/ops/ops.pbtxt
+++ b/tensorflow/core/ops/ops.pbtxt
@@ -50770,6 +50770,13 @@ op {
       b: false
     }
   }
+  attr {
+    name: "use_spmd_for_xla_partitioning"
+    type: "bool"
+    default_value {
+      b: false
+    }
+  }
 }
 op {
   name: "TPUReplicatedInput"