diff options
| author |  Zongheng Yang <zongheng@google.com> | 2016-10-14 11:36:54 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-14 12:51:19 -0700 |
| commit | 68af6c9848defff9c9e77bf92302fd266f019cd5 (patch) | |
| tree | 9b8154a97ada994b896c4a07c4a8c9402c86afdd | |
| parent | fb016110ea162f5afc981c7d59982c371c390b3f (diff) | |
Change for internal compatibility.
| -rw-r--r-- | tensorflow/contrib/session_bundle/exporter_test.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/session_bundle/exporter_test.py b/tensorflow/contrib/session_bundle/exporter_test.py index ee7173c5f0..6e06446d61 100644 --- a/tensorflow/contrib/session_bundle/exporter_test.py +++ b/tensorflow/contrib/session_bundle/exporter_test.py @@ -93,7 +93,8 @@ class SaveRestoreShardedTest(tf.test.TestCase): save = tf.train.Saver({"v0": v0, "v1": v1}, restore_sequentially=True, - sharded=sharded) + sharded=sharded, + write_version=tf.train.SaverDef.V1) export = exporter.Exporter(save) compare_def = tf.get_default_graph().as_graph_def() export.init(compare_def, |