add accessor to stats byte vlaue.

add a ForEachEventMetadata method to iterate through [plane's] event metadata. PiperOrigin-RevId: 450572175
cnuSODLearning · May 24, 2022 · 800c8af · 800c8af
1 parent c72bc08
commit 800c8af
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tensorflow/core/profiler/utils/xplane_visitor.h b/tensorflow/core/profiler/utils/xplane_visitor.h
@@ -58,6 +58,8 @@ class XStatVisitor {
 
   uint64 UintValue() const { return stat_->uint64_value(); }
 
+  absl::string_view BytesValue() const { return stat_->bytes_value(); }
+
   uint64 IntOrUintValue() const {
     return ValueCase() == XStat::kUint64Value ? UintValue()
                                               : static_cast<uint64>(IntValue());
@@ -275,6 +277,15 @@ class XPlaneVisitor : public XStatsOwner<XPlane> {
     bundle.JoinAll();
   }
 
+  template <typename ForEachEventMetadataFunc>
+  void ForEachEventMetadata(
+      ForEachEventMetadataFunc&& for_each_event_metadata) {
+    for (const auto& event : plane_->event_metadata()) {
+      for_each_event_metadata(event.first,
+                              XEventMetadataVisitor(this, &event.second));
+    }
+  }
+
   // Returns event metadata given its id. Returns a default value if not found.
   const XEventMetadata* GetEventMetadata(int64_t event_metadata_id) const;