Frontend for Inference Profile.

PiperOrigin-RevId: 703683365
Intel-tensorflow · Dec 7, 2024 · 3ab9024 · 3ab9024
1 parent af46a3d
commit 3ab9024
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tensorflow/core/profiler/convert/xplane_to_tool_names.cc b/tensorflow/core/profiler/convert/xplane_to_tool_names.cc
@@ -44,6 +44,7 @@ absl::StatusOr<std::string> GetAvailableToolNames(
     tools.push_back("pod_viewer");
     tools.push_back("tf_data_bottleneck_analysis");
     tools.push_back("op_profile");
+    tools.push_back("inference_profile");
     tools.push_back("hlo_stats");
 
     TF_ASSIGN_OR_RETURN(std::unique_ptr<XSpace> xspace,

diff --git a/tensorflow/core/profiler/convert/xplane_to_tool_names_test.cc b/tensorflow/core/profiler/convert/xplane_to_tool_names_test.cc
@@ -123,6 +123,7 @@ TEST_P(XPlaneToToolsTest, ToolsList) {
       "tf_data_bottleneck_analysis",
       "op_profile",
       "hlo_stats",
+      "inference_profile",
   };
   expected_tools.insert(expected_tools.end(), test_case.expected_tools.begin(),
                         test_case.expected_tools.end());