diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-02-01 21:43:12 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-02 10:32:17 -0800 |
| commit | 0612bc8fb566d813d35050a33a7109d67bb6136d (patch) | |
| tree | 5b7df0b75e9ff59bdcf51adcd644bf3355551d85 /tensorflow/contrib/lite/arena_planner_test.cc | |
| parent | 4e1863b76676afb863131f09ce8f9ac44f04e816 (diff) | |
Fix tolerance too tight for Wasserstein distance test.
PiperOrigin-RevId: 184240222
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 e10611e6d4..c27c327abc 100644 --- a/tensorflow/contrib/lite/arena_planner_test.cc +++ b/tensorflow/contrib/lite/arena_planner_test.cc @@ -18,8 +18,6 @@ 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 { @@ -466,7 +464,9 @@ TEST_F(ArenaPlannerTest, LargerGraphAndStepwiseAllocation) { } // namespace tflite int main(int argc, char** argv) { - ::tflite::LogToStderr(); + // ::tflite::LogToStderr(); + FLAGS_logtostderr = true; + ::testing::InitGoogleTest(&argc, argv); return RUN_ALL_TESTS(); } |