#include "tensorflow/lite/tools/benchmark/profiling_listener.h"
#include <fstream>
#include <string>
#include "tensorflow/lite/interpreter.h"
#include "tensorflow/lite/profiling/profile_summarizer.h"
#include "tensorflow/lite/profiling/profile_summary_formatter.h"
#include "tensorflow/lite/tools/benchmark/benchmark_model.h"
#include "tensorflow/lite/tools/benchmark/benchmark_params.h"
#include "tensorflow/lite/tools/logging.h"
namespace tflite {
namespace benchmark {
ProfilingListener::ProfilingListener(
Interpreter* interpreter, uint32_t max_num_initial_entries,
bool allow_dynamic_buffer_increase, const std::string& output_file_path,
std::shared_ptr<profiling::ProfileSummaryFormatter> summarizer_formatter)
: … { … }
void ProfilingListener::OnBenchmarkStart(const BenchmarkParams& params) { … }
void ProfilingListener::OnSingleRunStart(RunType run_type) { … }
void ProfilingListener::OnSingleRunEnd() { … }
void ProfilingListener::OnBenchmarkEnd(const BenchmarkResults& results) { … }
}
}