diff options
author | Anna R <annarev@google.com> | 2018-09-25 15:56:57 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-25 16:02:53 -0700 |
commit | c274706224fc977fa4a8b8721c3e0b10814e638e (patch) | |
tree | 42904e810edc57218189cdefafddb7c132753b29 /tensorflow/contrib/lite/python | |
parent | f0f475690cb728d7988328d32a5955c55ab1fb22 (diff) |
Automated rollback of commit 20c71535c5f1ed1d918d6cc6e327ffbba49ecbd6
PiperOrigin-RevId: 214519671
Diffstat (limited to 'tensorflow/contrib/lite/python')
-rw-r--r-- | tensorflow/contrib/lite/python/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/contrib/lite/python/convert_saved_model.py | 12 |
2 files changed, 7 insertions, 7 deletions
diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index 916788f215..57e1290e07 100644 --- a/tensorflow/contrib/lite/python/BUILD +++ b/tensorflow/contrib/lite/python/BUILD @@ -144,7 +144,7 @@ py_library( name = "convert_saved_model", srcs = ["convert_saved_model.py"], srcs_version = "PY2AND3", - visibility = ["//tensorflow/contrib/lite:__subpackages__"], + visibility = ["//visibility:public"], deps = [ ":convert", "//tensorflow/contrib/saved_model:saved_model_py", diff --git a/tensorflow/contrib/lite/python/convert_saved_model.py b/tensorflow/contrib/lite/python/convert_saved_model.py index d18b60d0ea..1553464b9f 100644 --- a/tensorflow/contrib/lite/python/convert_saved_model.py +++ b/tensorflow/contrib/lite/python/convert_saved_model.py @@ -44,7 +44,7 @@ def _log_tensor_details(tensor_info): dtype) -def get_meta_graph_def(saved_model_dir, tag_set): +def _get_meta_graph_def(saved_model_dir, tag_set): """Validate saved_model and extract MetaGraphDef. Args: @@ -61,7 +61,7 @@ def get_meta_graph_def(saved_model_dir, tag_set): return loader.load(sess, tag_set, saved_model_dir) -def get_signature_def(meta_graph, signature_key): +def _get_signature_def(meta_graph, signature_key): """Get the signature def from meta_graph with given signature_key. Args: @@ -86,7 +86,7 @@ def get_signature_def(meta_graph, signature_key): return signature_def_map[signature_key] -def get_inputs_outputs(signature_def): +def _get_inputs_outputs(signature_def): """Get inputs and outputs from SignatureDef. Args: @@ -236,9 +236,9 @@ def freeze_saved_model(saved_model_dir, input_arrays, input_shapes, input_arrays or output_arrays are not valid. """ # Read SignatureDef. - meta_graph = get_meta_graph_def(saved_model_dir, tag_set) - signature_def = get_signature_def(meta_graph, signature_key) - inputs, outputs = get_inputs_outputs(signature_def) + meta_graph = _get_meta_graph_def(saved_model_dir, tag_set) + signature_def = _get_signature_def(meta_graph, signature_key) + inputs, outputs = _get_inputs_outputs(signature_def) # Check SavedModel for assets directory. collection_def = meta_graph.collection_def |