diff options
| author |  Martin Wicke <wicke@google.com> | 2017-03-23 12:31:16 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-23 13:44:29 -0700 |
| commit | bc456e361d49d1d89a74b80060c70efb51fd7d87 (patch) | |
| tree | 825e04287f1e2d2ac098ca3f0fdd4e361aefd68c /tensorflow/tools/benchmark | |
| parent | 8ca071456537e6c96ae8896c2a20b1f08b0e59d3 (diff) | |
Merge changes from github.
Change: 151046259
Diffstat (limited to 'tensorflow/tools/benchmark')
| -rw-r--r-- | tensorflow/tools/benchmark/BUILD | 1 | ||||
| -rw-r--r-- | tensorflow/tools/benchmark/benchmark_model.cc | 5 |
2 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/tools/benchmark/BUILD b/tensorflow/tools/benchmark/BUILD index 3b1901fd56..a2ffca97ec 100644 --- a/tensorflow/tools/benchmark/BUILD +++ b/tensorflow/tools/benchmark/BUILD @@ -34,6 +34,7 @@ cc_library( "//tensorflow/core:lib", "//tensorflow/core:framework", "//tensorflow/core:framework_internal", + "//tensorflow/core:framework_lite", "//tensorflow/core:protos_all_cc", "//tensorflow/core:tensorflow", "//tensorflow/core:test", diff --git a/tensorflow/tools/benchmark/benchmark_model.cc b/tensorflow/tools/benchmark/benchmark_model.cc index 0fcfaf747b..db2ac31baf 100644 --- a/tensorflow/tools/benchmark/benchmark_model.cc +++ b/tensorflow/tools/benchmark/benchmark_model.cc @@ -36,6 +36,7 @@ limitations under the License. #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/init_main.h" #include "tensorflow/core/platform/logging.h" +#include "tensorflow/core/platform/platform.h" #include "tensorflow/core/platform/types.h" #include "tensorflow/core/public/session.h" #include "tensorflow/core/util/command_line_flags.h" @@ -272,7 +273,11 @@ Status TimeMultipleRuns(double sleep_seconds, int num_runs, // This can be helpful to determine the effect of mobile processor // scaling and thermal throttling. if (sleep_seconds > 0.0) { +#ifdef PLATFORM_WINDOWS + Sleep(sleep_seconds * 1000); +#else nanosleep(&req, nullptr); +#endif } } std::stringstream stream; |