diff --git a/tensorflow/lite/python/op_hint.py b/tensorflow/lite/python/op_hint.py
index 90c87503f48..ae1f5f094b9 100644
--- a/tensorflow/lite/python/op_hint.py
+++ b/tensorflow/lite/python/op_hint.py
@@ -1140,7 +1140,6 @@ def _get_correct_mapping(original_index, nodes):
   return original_index
 
 
-@_tf_export("lite.convert_op_hints_to_stubs")
 def _convert_op_hints_to_stubs_helper(
     graph_def, write_callback=lambda sess, graph_def: None):
   """Converts a graph_def to a new graph_def where all op hints are stubbed.
@@ -1248,6 +1247,7 @@ def find_all_hinted_output_nodes(session=None, graph_def=None):
   return hinted_outputs_nodes
 
 
+@_tf_export("lite.experimental.convert_op_hints_to_stubs")
 def convert_op_hints_to_stubs(session=None,
                               graph_def=None,
                               write_callback=lambda graph_def, comments: None):
diff --git a/tensorflow/tools/api/golden/v1/tensorflow.lite.experimental.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.lite.experimental.pbtxt
index 8d585ea6643..354a7086d60 100644
--- a/tensorflow/tools/api/golden/v1/tensorflow.lite.experimental.pbtxt
+++ b/tensorflow/tools/api/golden/v1/tensorflow.lite.experimental.pbtxt
@@ -4,4 +4,8 @@ tf_module {
     name: "nn"
     mtype: "<type \'module\'>"
   }
+  member_method {
+    name: "convert_op_hints_to_stubs"
+    argspec: "args=[\'session\', \'graph_def\', \'write_callback\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'<function <lambda> instance>\'], "
+  }
 }
diff --git a/tensorflow/tools/api/golden/v2/tensorflow.lite.experimental.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.lite.experimental.pbtxt
index 8d585ea6643..354a7086d60 100644
--- a/tensorflow/tools/api/golden/v2/tensorflow.lite.experimental.pbtxt
+++ b/tensorflow/tools/api/golden/v2/tensorflow.lite.experimental.pbtxt
@@ -4,4 +4,8 @@ tf_module {
     name: "nn"
     mtype: "<type \'module\'>"
   }
+  member_method {
+    name: "convert_op_hints_to_stubs"
+    argspec: "args=[\'session\', \'graph_def\', \'write_callback\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'<function <lambda> instance>\'], "
+  }
 }