diff options
| author |  Sukriti Ramesh <sukritiramesh@google.com> | 2017-08-09 16:36:20 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-09 16:39:29 -0700 |
| commit | 5aff53032be4cdf98844251de2e1d917367250c2 (patch) | |
| tree | 99b643b6fef186c92f2df489925a14c6cacb5f12 /tensorflow/python/saved_model | |
| parent | b5c59851f79553c8a29166ca1b384767617b8d78 (diff) | |
Comment update for main-op in the SavedModel builder APIs.
PiperOrigin-RevId: 164791375
Diffstat (limited to 'tensorflow/python/saved_model')
| -rw-r--r-- | tensorflow/python/saved_model/builder_impl.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/python/saved_model/builder_impl.py b/tensorflow/python/saved_model/builder_impl.py index 5c988bf95a..73a3f9075d 100644 --- a/tensorflow/python/saved_model/builder_impl.py +++ b/tensorflow/python/saved_model/builder_impl.py @@ -252,7 +252,9 @@ class SavedModelBuilder(object): restore op upon a load. clear_devices: Set to true if the device info on the default graph should be cleared. - main_op: Op or group of ops to execute when the graph is loaded. + main_op: Op or group of ops to execute when the graph is loaded. Note + that when the main_op is specified it is run after the restore op at + load-time. Raises: AssertionError: If the variables for the SavedModel have not been saved @@ -324,7 +326,9 @@ class SavedModelBuilder(object): restore op upon a load. clear_devices: Set to true if the device info on the default graph should be cleared. - main_op: Op or group of ops to execute when the graph is loaded. + main_op: Op or group of ops to execute when the graph is loaded. Note + that when the main_op is specified it is run after the restore op at + load-time. """ if self._has_saved_variables: raise AssertionError("Graph state including variables and assets has " |