aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/interpreter.cc
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-29 16:19:37 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-29 16:25:43 -0700
commit8648bd52264116760c54de16ffbce6c98d7397e8 (patch)
treeb26123fca57aab79b86c45c99a1b7724ceceec44 /tensorflow/contrib/lite/interpreter.cc
parentd7642767d24464127aae8c118caad597dea9e017 (diff)
Do not overwrite inputs.
PiperOrigin-RevId: 202724720
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
-rw-r--r--tensorflow/contrib/lite/interpreter.cc3
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc
index 4b3ba5df10..dcb4ef593e 100644
--- a/tensorflow/contrib/lite/interpreter.cc
+++ b/tensorflow/contrib/lite/interpreter.cc
@@ -538,7 +538,8 @@ TfLiteStatus Interpreter::PrepareOpsStartingAt(
TfLiteStatus Interpreter::PrepareOpsAndTensors() {
if (!memory_planner_) {
memory_planner_.reset(new ArenaPlanner(
- &context_, std::unique_ptr<GraphInfo>(new InterpreterInfo(this))));
+ &context_, std::unique_ptr<GraphInfo>(new InterpreterInfo(this)),
+ /*preserve_inputs=*/true));
memory_planner_->PlanAllocations();
}