experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor tweaks to tf.saved_model.save's docstring

Browse Source 
Subclass Module instead of Checkpoint, remove the "this may change" paragraph.

PiperOrigin-RevId: 249724699
This commit is contained in:
Allen Lavoie 2019-05-23 15:13:15 -07:00 committed by TensorFlower Gardener
parent 07ade963a5
commit c02e3f9af9
2 changed files with 2 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/python/saved_model/save.py
View File

@ -630,7 +630,7 @@ def save(obj, export_dir, signatures=None):
Example usage:
```python
class Adder(tf.train.Checkpoint):
class Adder(tf.Module):
@tf.function(input_signature=[tf.TensorSpec(shape=None, dtype=tf.float32)])
def add(self, x):
@ -747,17 +747,6 @@ def save(obj, export_dir, signatures=None):
handled automatically, such as when the exported model contains operations
which the consumer does not have definitions for.
The current implementation of `tf.saved_model.save` targets serving use-cases,
but omits information which will be necessary for the planned future
implementation of `tf.saved_model.load`. Exported models using the current
`save` implementation, and other existing SavedModels, will not be compatible
with `tf.saved_model.load` when it is implemented. Further, `save` will in the
future attempt to export `@tf.function`-decorated methods which it does not
currently inspect, so some objects which are exportable today will raise
exceptions on export in the future (e.g. due to complex/non-serializable
default arguments). Such backwards-incompatible API changes are expected only
prior to the TensorFlow 2.0 release.
Args:
obj: A trackable object to export.
export_dir: A directory in which to write the SavedModel.

2
tensorflow/python/saved_model/save_test.py
View File

@ -262,7 +262,7 @@ class SaveTest(test.TestCase):
def test_docstring(self):
class Adder(util.Checkpoint):
class Adder(module.Module):
@def_function.function(input_signature=[tensor_spec.TensorSpec(
shape=None, dtype=dtypes.float32)])