Internal Changes

PiperOrigin-RevId: 622017286
tensorflow · Apr 5, 2024 · 30fc866 · 30fc866
1 parent b01455b
commit 30fc866
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/frontend/app/components/graph_viewer/graph_config/graph_config.ng.html b/frontend/app/components/graph_viewer/graph_config/graph_config.ng.html
@@ -11,6 +11,7 @@
         panelClass="panel-override"
         [(ngModel)]="params.selectedModule"
         [disabled]="useProgramId"
+        (selectionChange)="onModuleSelectionChange($event)"
       >
         <mat-option *ngFor="let module of getModuleList()" [value]="module">
           {{ module }}

diff --git a/frontend/app/components/graph_viewer/graph_config/graph_config.ts b/frontend/app/components/graph_viewer/graph_config/graph_config.ts
@@ -1,4 +1,5 @@
 import {Component, EventEmitter, Input, OnChanges, OnDestroy, Output, SimpleChanges} from '@angular/core';
+import {MatSelectChange} from '@angular/material/select';
 import {type GraphConfigInput} from 'org_xprof/frontend/app/common/interfaces/graph_viewer';
 import {ReplaySubject} from 'rxjs';
 
@@ -13,6 +14,7 @@ export class GraphConfig implements OnDestroy, OnChanges {
   private readonly destroyed = new ReplaySubject<void>(1);
 
   @Output() readonly plot = new EventEmitter<Partial<GraphConfigInput>>();
+  @Output() readonly updateSelectedModule = new EventEmitter<string>();
 
   /** Form inputs properties */
   @Input() initialInputs: GraphConfigInput|undefined = undefined;
@@ -67,6 +69,10 @@ export class GraphConfig implements OnDestroy, OnChanges {
     this.plot.emit(this.params);
   }
 
+  onModuleSelectionChange(e: MatSelectChange) {
+    this.updateSelectedModule.emit(e.value);
+  }
+
   ngOnDestroy() {
     // Unsubscribes all pending subscriptions.
     this.destroyed.next();