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 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, 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());