diff options
author | Gunhan Gulsoy <gunan@google.com> | 2018-02-01 23:47:33 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-02 10:35:37 -0800 |
commit | ffa37312ab71d338adfb035a5ab71beb0da842ef (patch) | |
tree | 265fc5ea59e1045c18b6ecb10094635494a049e2 /tensorflow/contrib/lite/arena_planner_test.cc | |
parent | 0612bc8fb566d813d35050a33a7109d67bb6136d (diff) |
Fix some tf-lite tests
PiperOrigin-RevId: 184247187
Diffstat (limited to 'tensorflow/contrib/lite/arena_planner_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/arena_planner_test.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/arena_planner_test.cc b/tensorflow/contrib/lite/arena_planner_test.cc index c27c327abc..e10611e6d4 100644 --- a/tensorflow/contrib/lite/arena_planner_test.cc +++ b/tensorflow/contrib/lite/arena_planner_test.cc @@ -18,6 +18,8 @@ limitations under the License. #include <gmock/gmock.h> #include <gtest/gtest.h> +#include "tensorflow/contrib/lite/testing/util.h" +#include "tensorflow/core/platform/logging.h" namespace tflite { namespace { @@ -464,9 +466,7 @@ TEST_F(ArenaPlannerTest, LargerGraphAndStepwiseAllocation) { } // namespace tflite int main(int argc, char** argv) { - // ::tflite::LogToStderr(); - FLAGS_logtostderr = true; - + ::tflite::LogToStderr(); ::testing::InitGoogleTest(&argc, argv); return RUN_ALL_TESTS(); } |