From 3ab90242469b9691a1fed7994fe6ba523ed1cada Mon Sep 17 00:00:00 2001
From: Clive Verghese <cliveverghese@google.com>
Date: Fri, 6 Dec 2024 18:53:42 -0800
Subject: [PATCH] Frontend for Inference Profile.

PiperOrigin-RevId: 703683365
---
 tensorflow/core/profiler/convert/xplane_to_tool_names.cc      | 1 +
 tensorflow/core/profiler/convert/xplane_to_tool_names_test.cc | 1 +
 2 files changed, 2 insertions(+)

diff --git a/tensorflow/core/profiler/convert/xplane_to_tool_names.cc b/tensorflow/core/profiler/convert/xplane_to_tool_names.cc
index 03fa8cf063ba57..e01a932fdfaa91 100644
--- 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
index 923ecf05de130f..73a79240343c78 100644
--- 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());