From 0ead9dec6030d91f2c7f61065347adb3c029cb66 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 12 Jan 2021 14:18:33 -0800
Subject: [PATCH] Update ops-related pbtxt files.

PiperOrigin-RevId: 351448163
Change-Id: I1a499506395d81dcb752c2018e1ddd3666fe61d8
---
 .../StatelessRandomGetAlg.pbtxt               |  8 +++++
 .../StatelessRandomGetKeyCounter.pbtxt        | 29 +++++++++++++++++++
 tensorflow/core/ops/ops.pbtxt                 |  2 ++
 3 files changed, 39 insertions(+)

diff --git a/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetAlg.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetAlg.pbtxt
index f54eef5b5fb..e80fdc09ccf 100644
--- a/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetAlg.pbtxt
+++ b/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetAlg.pbtxt
@@ -5,3 +5,11 @@ op {
     type: DT_INT32
   }
 }
+op {
+  name: "StatelessRandomGetAlg"
+  output_arg {
+    name: "alg"
+    type: DT_INT32
+  }
+  is_stateful: true
+}
diff --git a/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetKeyCounter.pbtxt b/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetKeyCounter.pbtxt
index 1ee936dcb36..78d7322af24 100644
--- a/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetKeyCounter.pbtxt
+++ b/tensorflow/core/ops/compat/ops_history_v2/StatelessRandomGetKeyCounter.pbtxt
@@ -26,3 +26,32 @@ op {
     }
   }
 }
+op {
+  name: "StatelessRandomGetKeyCounter"
+  input_arg {
+    name: "seed"
+    type_attr: "Tseed"
+  }
+  output_arg {
+    name: "key"
+    type: DT_UINT64
+  }
+  output_arg {
+    name: "counter"
+    type: DT_UINT64
+  }
+  attr {
+    name: "Tseed"
+    type: "type"
+    default_value {
+      type: DT_INT64
+    }
+    allowed_values {
+      list {
+        type: DT_INT32
+        type: DT_INT64
+      }
+    }
+  }
+  is_stateful: true
+}
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt
index babff1d9bc7..b4e1a968627 100644
--- a/tensorflow/core/ops/ops.pbtxt
+++ b/tensorflow/core/ops/ops.pbtxt
@@ -51346,6 +51346,7 @@ op {
     name: "alg"
     type: DT_INT32
   }
+  is_stateful: true
 }
 op {
   name: "StatelessRandomGetKeyCounter"
@@ -51374,6 +51375,7 @@ op {
       }
     }
   }
+  is_stateful: true
 }
 op {
   name: "StatelessRandomGetKeyCounterAlg"