diff options
| author |  Shashi Shekhar <shashishekhar@google.com> | 2018-06-07 12:23:10 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-07 12:28:53 -0700 |
| commit | 2857228ba6c7b357185e7a0af346f4fc93a10f74 (patch) | |
| tree | 99a92601dde4aca84cd817e25c0208134024b002 /tensorflow/contrib/lite/profiling | |
| parent | 6f20926fb7a181c44cca6191eec8961040d83cd1 (diff) | |
Misc fixes to benchmarks.
PiperOrigin-RevId: 199676652
Diffstat (limited to 'tensorflow/contrib/lite/profiling')
| -rw-r--r-- | tensorflow/contrib/lite/profiling/profile_summarizer.cc | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/profiling/profile_summarizer.cc b/tensorflow/contrib/lite/profiling/profile_summarizer.cc index 6f2c9cd2b3..45388b500c 100644 --- a/tensorflow/contrib/lite/profiling/profile_summarizer.cc +++ b/tensorflow/contrib/lite/profiling/profile_summarizer.cc @@ -85,11 +85,18 @@ OperatorDetails GetOperatorDetails(const tflite::Interpreter& interpreter, return details; } +tensorflow::StatSummarizerOptions GetProfileSummarizerOptions() { + auto options = tensorflow::StatSummarizerOptions(); + options.show_summary = true; + options.show_memory = false; + return options; +} + } // namespace ProfileSummarizer::ProfileSummarizer() - : stats_calculator_(new ::tensorflow::StatsCalculator( - tensorflow::StatSummarizerOptions())) {} + : stats_calculator_( + new ::tensorflow::StatsCalculator(GetProfileSummarizerOptions())) {} void ProfileSummarizer::ProcessProfiles( const std::vector<const ProfileEvent*>& profile_stats, |