Add node occurrences to detail section for easier debugging

PiperOrigin-RevId: 624314662
tensorflow · Apr 12, 2024 · 55ed9c7 · 55ed9c7
1 parent 9218164
commit 55ed9c7
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/frontend/app/common/interfaces/op_profile.jsonpb_decls.d.ts b/frontend/app/common/interfaces/op_profile.jsonpb_decls.d.ts
@@ -47,11 +47,11 @@ export namespace Node {
  *  Metrics are always "total" rather than "self".
  */
 export interface Metrics {
-  time?: /* double */ number;
   flops?: /* double */ number;
   bandwidthUtils?: /* double */ number[];
   rawTime?: /* double */ number;
   rawFlops?: /* double */ number;
   rawBytesAccessedArray?: /* double */ number[];
+  occurrences?: /* uint32 */ number;
   avgTimePs?: /* double */ number;
 }
diff --git a/frontend/app/components/op_profile/op_details/op_details.ng.html b/frontend/app/components/op_profile/op_details/op_details.ng.html
@@ -97,6 +97,10 @@
           <div class="title">Total Time Sum:</div>
           <code class="expression">{{rawTimeMs}}</code>
         </div>
+        <div class="info" [hidden]="!occurrences">
+          <div class="title">Occurrences:</div>
+          <code class="expression">{{occurrences}}</code>
+        </div>
         <div class="info" [hidden]="!avgTimeMs">
           <div class="title">Total Time Avg:</div>
           <code class="expression">{{avgTimeMs}}</code>

diff --git a/frontend/app/components/op_profile/op_details/op_details.ts b/frontend/app/components/op_profile/op_details/op_details.ts
@@ -50,6 +50,7 @@ export class OpDetails {
   expression: string = '';
   provenance: string = '';
   rawTimeMs = '';
+  occurrences = 0;
   avgTimeMs = '';
   fused: boolean = false;
   hasCategory: boolean = false;
@@ -229,6 +230,7 @@ export class OpDetails {
     } else {
       this.rawTimeMs = '';
     }
+    this.occurrences = this.node.metrics?.occurrences || 0;
 
     if (this.node.metrics && this.node.metrics.avgTimePs) {
       this.avgTimeMs = utils.humanReadableText(