From cccace731712ab192448ff9bc3ce7259fd462e74 Mon Sep 17 00:00:00 2001 From: Nupur Garg Date: Wed, 26 Sep 2018 11:18:09 -0700 Subject: Internal change. PiperOrigin-RevId: 214636032 --- tensorflow/contrib/lite/python/BUILD | 2 +- tensorflow/contrib/lite/python/convert_saved_model.py | 12 ++++++------ 2 files changed, 7 insertions(+), 7 deletions(-) (limited to 'tensorflow/contrib/lite/python') diff --git a/tensorflow/contrib/lite/python/BUILD b/tensorflow/contrib/lite/python/BUILD index 57e1290e07..916788f215 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 = ["//visibility:public"], + visibility = ["//tensorflow/contrib/lite:__subpackages__"], 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 1553464b9f..d18b60d0ea 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 -- cgit v1.2.3