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"