diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-05 17:35:00 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-05 17:37:43 -0700 |
commit | 0b87efcbae8cde976a302415c6df2189958e7a8b (patch) | |
tree | 38b5645c3a9e48a3291153d27475bf8082625b54 /tensorflow/contrib/lite/toco/tooling_util.cc | |
parent | 165a87a0f938aef649f7e817193716349e42072a (diff) |
Add a command line parameter to toco to change the way toco rescales input and output tensors.
PiperOrigin-RevId: 191825756
Diffstat (limited to 'tensorflow/contrib/lite/toco/tooling_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/tooling_util.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/tooling_util.cc b/tensorflow/contrib/lite/toco/tooling_util.cc index 61d08fa13f..b72f5fa2a7 100644 --- a/tensorflow/contrib/lite/toco/tooling_util.cc +++ b/tensorflow/contrib/lite/toco/tooling_util.cc @@ -1413,7 +1413,8 @@ void ResolveModelFlags(const ModelFlags& model_flags, Model* model) { CHECK(input_array.shape().dims_size()); } } - + model->flags.set_change_concat_input_ranges( + model_flags.change_concat_input_ranges()); model->flags.set_allow_nonascii_arrays(model_flags.allow_nonascii_arrays()); model->flags.set_allow_nonexistent_arrays( model_flags.allow_nonexistent_arrays()); |