diff options
author | Andrew Selle <aselle@google.com> | 2018-07-13 10:59:05 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-13 11:04:28 -0700 |
commit | b5b5cc0e248b6fbd5025765f592a43af158a1cb2 (patch) | |
tree | 82deb1f8db93f157e1dbc45662a63313aa5bd717 /tensorflow/contrib/lite/interpreter.cc | |
parent | 2449da260ccd7dc075b890729f7a6cbb08e10882 (diff) |
Make an option that allows making the memory plan not reuse space.
This is called 'allow_intermediates'. In the future we should have
a way for users to enable this functionality from the interpreter API.
Also, preserve_inputs is now better commented.
PiperOrigin-RevId: 204496777
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
-rw-r--r-- | tensorflow/contrib/lite/interpreter.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc index 0641a08636..d103786694 100644 --- a/tensorflow/contrib/lite/interpreter.cc +++ b/tensorflow/contrib/lite/interpreter.cc @@ -593,7 +593,7 @@ TfLiteStatus Interpreter::PrepareOpsAndTensors() { if (!memory_planner_) { memory_planner_.reset(new ArenaPlanner( &context_, std::unique_ptr<GraphInfo>(new InterpreterInfo(this)), - /*preserve_inputs=*/true)); + /*preserve_inputs=*/true, /*preserve_intermediates*/ false)); memory_planner_->PlanAllocations(); } |