Expose saved_model.SavedModelBuilder as the internal API

PiperOrigin-RevId: 367720276
Change-Id: I7f660a52e8fba185ca4a54bc48e3ed29a507adb0

tensorflow/python/saved_model/builder_impl.py

@@ -43,10 +43,11 @@
 
 # Base class for the SavedModelBuilder that is only used by Tensorflow
 # internally. Please use tf.compat.v1.saved_model.SavedModelBuilder instead.
+@tf_export("__internal__.saved_model.SavedModelBuilder", v1=[])
 class _SavedModelBuilder(object):
   """Builds the `SavedModel` protocol buffer and saves variables and assets.
 
-  The `SavedModelBuilder` class provides the functionality to build a 
+  The `SavedModelBuilder` class provides the functionality to build a
   `SavedModel` protocol buffer. Specifically, this allows multiple meta
   graphs to be saved as part of a single language-neutral `SavedModel`,
   while sharing variables and assets.

tensorflow/tools/api/golden/v2/tensorflow.__internal__.saved_model.-saved-model-builder.pbtxt

@@ -0,0 +1,21 @@
+path: "tensorflow.__internal__.saved_model.SavedModelBuilder"
+tf_class {
+  is_instance: "<class \'tensorflow.python.saved_model.builder_impl._SavedModelBuilder\'>"
+  is_instance: "<type \'object\'>"
+  member_method {
+    name: "__init__"
+    argspec: "args=[\'self\', \'export_dir\'], varargs=None, keywords=None, defaults=None"
+  }
+  member_method {
+    name: "add_meta_graph"
+    argspec: "args=[\'self\', \'tags\', \'signature_def_map\', \'assets_list\', \'clear_devices\', \'init_op\', \'train_op\', \'saver\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'None\', \'None\', \'None\'], "
+  }
+  member_method {
+    name: "add_meta_graph_and_variables"
+    argspec: "args=[\'self\', \'sess\', \'tags\', \'signature_def_map\', \'assets_list\', \'clear_devices\', \'init_op\', \'train_op\', \'strip_default_attrs\', \'saver\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'None\', \'None\', \'False\', \'None\'], "
+  }
+  member_method {
+    name: "save"
+    argspec: "args=[\'self\', \'as_text\'], varargs=None, keywords=None, defaults=[\'False\'], "
+  }
+}

tensorflow/tools/api/golden/v2/tensorflow.__internal__.saved_model.pbtxt

@@ -1,5 +1,9 @@
 path: "tensorflow.__internal__.saved_model"
 tf_module {
+  member {
+    name: "SavedModelBuilder"
+    mtype: "<type \'type\'>"
+  }
   member {
     name: "StructureCoder"
     mtype: "<type \'type\'>"