[SavedModel] simple_save api support dump incremental graph info. (DeepRec-AI#115)

shanshanpt · web-flow · commit ccf65cf56786 · 2022-03-17T13:15:27.000+08:00
diff --git a/tensorflow/python/saved_model/simple_save.py b/tensorflow/python/saved_model/simple_save.py
@@ -32,7 +32,8 @@
     None,
     'This function will only be available through the v1 compatibility '
     'library as tf.compat.v1.saved_model.simple_save.')
-def simple_save(session, export_dir, inputs, outputs, legacy_init_op=None):
+def simple_save(session, export_dir, inputs, outputs,
+                legacy_init_op=None, save_incr_model=False):
   """Convenience function to build a SavedModel suitable for serving.
 
   In many common cases, saving models for serving will be as simple as:
@@ -80,7 +81,7 @@ def simple_save(session, export_dir, inputs, outputs, legacy_init_op=None):
       signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY:
           signature_def_utils.predict_signature_def(inputs, outputs)
   }
-  b = builder.SavedModelBuilder(export_dir)
+  b = builder.SavedModelBuilder(export_dir, save_incr_model=save_incr_model)
   b.add_meta_graph_and_variables(
       session,
       tags=[tag_constants.SERVING],
diff --git a/tensorflow/tools/api/golden/v1/tensorflow.saved_model.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.saved_model.pbtxt
@@ -190,6 +190,6 @@ tf_module {
   }
   member_method {
     name: "simple_save"
-    argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\'], varargs=None, keywords=None, defaults=[\'None\'], "
+    argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\', \'save_incr_model\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], "
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -190,6 +190,6 @@ tf_module {`
`190`	`190`	`}`
`191`	`191`	`member_method {`
`192`	`192`	`name: "simple_save"`
`193`		`- argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\'], varargs=None, keywords=None, defaults=[\'None\'], "`
	`193`	`+ argspec: "args=[\'session\', \'export_dir\', \'inputs\', \'outputs\', \'legacy_init_op\', \'save_incr_model\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], "`
`194`	`194`	`}`
`195`	`195`	`}`