aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/arena_planner_test.cc
diff options
context:
space:
mode:
authorGravatar Gunhan Gulsoy <gunan@google.com>2018-02-01 23:47:33 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-02-02 10:35:37 -0800
commitffa37312ab71d338adfb035a5ab71beb0da842ef (patch)
tree265fc5ea59e1045c18b6ecb10094635494a049e2 /tensorflow/contrib/lite/arena_planner_test.cc
parent0612bc8fb566d813d35050a33a7109d67bb6136d (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.cc6
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();
}