diff options
| author |  Sukriti Ramesh <sukritiramesh@google.com> | 2017-07-20 15:40:58 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-07-20 15:44:45 -0700 |
| commit | 856f2a5c1296a4256053553b6fd7f9d087f7d7ed (patch) | |
| tree | abf53539cb0799d6b12b35c1e7dcf44acee33791 /tensorflow/examples/saved_model | |
| parent | 99e439297174c6c109d5eaa97bc727dd5188e481 (diff) | |
Update example to use tf.saved_model API.
PiperOrigin-RevId: 162676113
Diffstat (limited to 'tensorflow/examples/saved_model')
| -rw-r--r-- | tensorflow/examples/saved_model/saved_model_half_plus_two.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/examples/saved_model/saved_model_half_plus_two.py b/tensorflow/examples/saved_model/saved_model_half_plus_two.py index 02afa0b0fc..0d6f1ef655 100644 --- a/tensorflow/examples/saved_model/saved_model_half_plus_two.py +++ b/tensorflow/examples/saved_model/saved_model_half_plus_two.py @@ -46,7 +46,6 @@ import sys import tensorflow as tf from tensorflow.python.lib.io import file_io -from tensorflow.python.saved_model import main_op FLAGS = None @@ -205,7 +204,8 @@ def _generate_saved_model_for_half_plus_two(export_dir, sess, [tf.saved_model.tag_constants.SERVING], signature_def_map=signature_def_map, assets_collection=tf.get_collection(tf.GraphKeys.ASSET_FILEPATHS), - main_op=tf.group(main_op.main_op(), assign_filename_op)) + main_op=tf.group(tf.saved_model.main_op.main_op(), + assign_filename_op)) else: builder.add_meta_graph_and_variables( sess, [tf.saved_model.tag_constants.SERVING], |