diff options
| author |  Michael Case <mikecase@google.com> | 2018-06-29 13:50:37 -0700 |
|---|---|---|
| committer | Michael Case <mikecase@google.com> | 2018-06-29 13:50:37 -0700 |
| commit | 7be4245d629510ed3d1c2edd7a2598167017f33b (patch) | |
| tree | 6585d143160249ae282044790d4589145a79efa2 /tensorflow/contrib/lite/python | |
| parent | 01c36c3d7b3e230c865e71d67e138a8dc765e7a6 (diff) | |
| parent | 79dab9ced650d69bdf3f312bd902bd52de5bdad8 (diff) | |
Merge commit for internal changes
Diffstat (limited to 'tensorflow/contrib/lite/python')
| -rw-r--r-- | tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h | 2 | ||||
| -rw-r--r-- | tensorflow/contrib/lite/python/lite.py | 4 | ||||
| -rw-r--r-- | tensorflow/contrib/lite/python/tflite_convert.py | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h index e7343cb388..681448be20 100644 --- a/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h +++ b/tensorflow/contrib/lite/python/interpreter_wrapper/interpreter_wrapper.h @@ -20,8 +20,8 @@ limitations under the License. #include <vector> // Place `<locale>` before <Python.h> to avoid build failures in macOS. -#include <locale> #include <Python.h> +#include <locale> // We forward declare TFLite classes here to avoid exposing them to SWIG. namespace tflite { diff --git a/tensorflow/contrib/lite/python/lite.py b/tensorflow/contrib/lite/python/lite.py index a4229f91f5..29a1487c1f 100644 --- a/tensorflow/contrib/lite/python/lite.py +++ b/tensorflow/contrib/lite/python/lite.py @@ -132,7 +132,7 @@ class TocoConverter(object): Args: - graph_def: TensorFlow GraphDef. + graph_def: Frozen TensorFlow GraphDef. input_tensors: List of input tensors. Type and shape are computed using `foo.get_shape()` and `foo.dtype`. output_tensors: List of output tensors (only .name is used from this). @@ -178,7 +178,7 @@ class TocoConverter(object): """Creates a TocoConverter class from a file containing a frozen GraphDef. Args: - graph_def_file: Full filepath of file containing TensorFlow GraphDef. + graph_def_file: Full filepath of file containing frozen GraphDef. input_arrays: List of input tensors to freeze graph with. output_arrays: List of output tensors to freeze graph with. input_shapes: Dict of strings representing input tensor names to list of diff --git a/tensorflow/contrib/lite/python/tflite_convert.py b/tensorflow/contrib/lite/python/tflite_convert.py index 0a60477c6d..9bd1f4f76e 100644 --- a/tensorflow/contrib/lite/python/tflite_convert.py +++ b/tensorflow/contrib/lite/python/tflite_convert.py @@ -225,7 +225,7 @@ def run_main(_): input_file_group.add_argument( "--graph_def_file", type=str, - help="Full filepath of file containing TensorFlow GraphDef.") + help="Full filepath of file containing frozen TensorFlow GraphDef.") input_file_group.add_argument( "--saved_model_dir", type=str, |