Rename tensorflow_stats to framework_op_stats

PiperOrigin-RevId: 596735653
tensorflow · Jan 19, 2024 · b9c624c · b9c624c
1 parent 5dbb485
commit b9c624c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/frontend/app/components/main_page/main_page_module.ts b/frontend/app/components/main_page/main_page_module.ts
@@ -53,8 +53,8 @@ export const routes: Routes = [
   {path: 'pod_viewer', component: PodViewer},
   {path: 'pod_viewer^', component: PodViewer},
   {path: 'tf_data_bottleneck_analysis^', component: TfDataBottleneckAnalysis},
-  {path: 'tensorflow_stats', component: TensorflowStatsAdapter},
-  {path: 'tensorflow_stats^', component: TensorflowStatsAdapter},
+  {path: 'framework_op_stats', component: TensorflowStatsAdapter},
+  {path: 'framework_op_stats^', component: TensorflowStatsAdapter},
   {path: 'trace_viewer', component: TraceViewer},
   {path: 'trace_viewer#', component: TraceViewer},
   {path: 'trace_viewer@^', component: TraceViewer},

diff --git a/plugin/tensorboard_plugin_profile/convert/raw_to_tool_data.py b/plugin/tensorboard_plugin_profile/convert/raw_to_tool_data.py
@@ -128,7 +128,7 @@ def xspace_to_tool_data(
     raw_data, success = xspace_wrapper_func(xspace_paths, tool)
     if success:
       data = input_pipeline_proto_to_gviz.to_json(raw_data)
-  elif tool == 'tensorflow_stats':
+  elif tool == 'framework_op_stats':
     raw_data, success = xspace_wrapper_func(xspace_paths, tool)
     if success:
       if tqx == 'out:csv':
@@ -197,7 +197,7 @@ def tool_proto_to_tool_data(tool_proto, tool, params):
   tqx = params.get('tqx', '')
   if tool == 'trace_viewer':
     data = process_raw_trace(tool_proto)
-  elif tool == 'tensorflow_stats':
+  elif tool == 'framework_stats':
     if tqx == 'out:csv;':
       data = tf_stats_proto_to_gviz.to_csv(tool_proto)
     else:

diff --git a/plugin/tensorboard_plugin_profile/profile_plugin.py b/plugin/tensorboard_plugin_profile/profile_plugin.py
@@ -73,7 +73,7 @@
     'overview_page@': 'overview_page.pb',
     'memory_viewer': 'memory_viewer.json',
     'pod_viewer': 'pod_viewer.json',
-    'tensorflow_stats': 'tensorflow_stats.pb',
+    'framework_op_stats': 'tensorflow_stats.pb',
     'kernel_stats': 'kernel_stats.pb',
     'memory_profile#': 'memory_profile.json.gz',
     'xplane': 'xplane.pb',