diff options
author | David Soergel <soergel@google.com> | 2017-09-26 08:59:34 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-26 09:06:50 -0700 |
commit | b6238a1b44c80c7dcb9930350ba53e2f33e3f81b (patch) | |
tree | 6f9fe9cfd7062d77e027151741d6bd81979af998 /tensorflow/contrib/meta_graph_transform | |
parent | f5ceb90e7f08fbe7605a002a546b22ef893f248c (diff) |
Add OWNERS for MetaGraphDef Transform Tool.
(Also, a docstring nit re the sparsify_gather transform).
PiperOrigin-RevId: 170059603
Diffstat (limited to 'tensorflow/contrib/meta_graph_transform')
-rw-r--r-- | tensorflow/contrib/meta_graph_transform/meta_graph_transform.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py b/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py index ff4afbb4ce..303c02dfa4 100644 --- a/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py +++ b/tensorflow/contrib/meta_graph_transform/meta_graph_transform.py @@ -706,7 +706,8 @@ def meta_graph_transform( output_names: Names of output nodes. transforms: A list of strings naming the graph transforms to be applied in order. These transform names are exactly those supported by the Graph - Transform Tool, with the addition of the 'freeze_graph' transform. + Transform Tool, with the addition of the 'freeze_graph' and + 'sparsify_gather' transforms. tags: A list of tags with which to annotate the transformed MetaGraphDef. checkpoint_path: A path to a checkpoint to restore during freezing, if needed (default None). |