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/arena_planner_test.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/arena_planner_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/arena_planner_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/arena_planner_test.cc b/tensorflow/contrib/lite/arena_planner_test.cc index f5bd1932f9..1adb426d58 100644 --- a/tensorflow/contrib/lite/arena_planner_test.cc +++ b/tensorflow/contrib/lite/arena_planner_test.cc @@ -156,7 +156,7 @@ class ArenaPlannerTest : public ::testing::Test { context_.ReportError = ReportError; planner_.reset(new ArenaPlanner( &context_, std::unique_ptr<GraphInfo>(new TestGraphInfo(graph)), - preserve_inputs)); + preserve_inputs, /*preserve intermediates*/ false)); CHECK(planner_->ResetAllocations() == kTfLiteOk); CHECK(planner_->PlanAllocations() == kTfLiteOk); } |