diff options
| author |  Nupur Garg <nupurgarg@google.com> | 2018-09-26 11:18:09 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-26 11:23:05 -0700 |
| commit | cccace731712ab192448ff9bc3ce7259fd462e74 (patch) | |
| tree | b6b9a6b361ff38b788f05e2aecdb21f524570767 /tensorflow/contrib/lite/python | |
| parent | 5cedb0427bd4db4117182da8bc0680dd555b4f49 (diff) | |
Internal change.
PiperOrigin-RevId: 214636032
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 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 |