Microprofiler can now output logs for benchmarks in the csv format.

BUG=240626703

Author: tanmaydas82

tensorflow/lite/micro/BUILD

@@ -317,6 +317,7 @@ cc_library(
     ],
     copts = micro_copts(),
     deps = [
+        ":micro_compatibility",
         ":micro_error_reporter",
         ":micro_time",
         "//tensorflow/lite/kernels/internal:compatibility",

tensorflow/lite/micro/micro_profiler.cc

@@ -16,6 +16,7 @@ limitations under the License.
 
 #include <cinttypes>
 #include <cstdint>
+#include <cstring>
 
 #include "tensorflow/lite/kernels/internal/compatibility.h"
 #include "tensorflow/lite/micro/micro_error_reporter.h"
@@ -67,4 +68,48 @@ void MicroProfiler::LogCsv() const {
 #endif
 }
 
+void MicroProfiler::LogTicksPerTagCsv() {
+#if !defined(TF_LITE_STRIP_ERROR_STRINGS)
+  MicroPrintf(
+      "\"Unique Tag\",\"Total ticks across all events with that tag.\"");
+  int total_ticks = 0;
+  for (int i = 0; i < num_events_; ++i) {
+    uint32_t ticks = end_ticks_[i] - start_ticks_[i];
+    TFLITE_DCHECK(tags_[i] != nullptr);
+    int position = FindExistingOrNextPosition(tags_[i]);
+    TFLITE_DCHECK(position >= 0);
+    total_ticks_per_tag[position].tag = tags_[i];
+    total_ticks_per_tag[position].ticks =
+        total_ticks_per_tag[position].ticks + ticks;
+    total_ticks += ticks;
+  }
+
+  for (int i = 0; i < num_events_; ++i) {
+    TicksPerTag each_tag_entry = total_ticks_per_tag[i];
+    if (each_tag_entry.tag == nullptr) {
+      break;
+    }
+    MicroPrintf("%s, %d", each_tag_entry.tag, each_tag_entry.ticks);
+  }
+  MicroPrintf("total number of ticks, %d", total_ticks);
+#endif
+}
+
+// This method finds a particular array element in the total_ticks_per_tag array
+// with the matching tag_name passed in the method. If it can find a
+// matching array element that has the same tag_name, then it will return the
+// position of the matching element. But if it unable to find a matching element
+// with the given tag_name, it will return the next available empty position
+// from the array.
+int MicroProfiler::FindExistingOrNextPosition(const char* tag_name) {
+  int pos = 0;
+  for (; pos < num_events_; pos++) {
+    TicksPerTag each_tag_entry = total_ticks_per_tag[pos];
+    if (each_tag_entry.tag == nullptr ||
+        strcmp(each_tag_entry.tag, tag_name) == 0) {
+      return pos;
+    }
+  }
+  return pos < num_events_ ? pos : -1;
+}
 }  // namespace tflite

tensorflow/lite/micro/micro_profiler.h

@@ -61,6 +61,11 @@ class MicroProfiler {
   // Separated Value) form.
   void LogCsv() const;
 
+  // Prints  total ticks for each unique tag in CSV format.
+  // Output will have one row for each unique tag along with the
+  // total ticks summed across all events with that particular tag.
+  void LogTicksPerTagCsv();
+
  private:
   // Maximum number of events that this class can keep track of. If we call
   // AddEvent more than kMaxEvents number of times, then the oldest event's
@@ -72,6 +77,17 @@ class MicroProfiler {
   uint32_t end_ticks_[kMaxEvents];
   int num_events_ = 0;
 
+  struct TicksPerTag {
+    const char* tag;
+    uint32_t ticks;
+  };
+  // In practice, the number of tags will be much lower than the number of
+  // events. But it is theoretically possible that each event to be unique and
+  // hence we allow total_ticks_per_tag to have kMaxEvents entries.
+  TicksPerTag total_ticks_per_tag[kMaxEvents];
+
+  int FindExistingOrNextPosition(const char* tag_name);
+
   TF_LITE_REMOVE_VIRTUAL_DELETE;
 };