Exposed GenerateHostResult() and GenerateRecommendation() to header f…

…ile. PiperOrigin-RevId: 288705912 Change-Id: Iae647aa415c7b011a8b9ced3c729d06c94a919c8
frreiss · Jan 8, 2020 · 2886154 · 2886154
1 parent f65d625
commit 2886154
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/tensorflow/core/profiler/convert/op_stats_to_input_pipeline_analysis.cc b/tensorflow/core/profiler/convert/op_stats_to_input_pipeline_analysis.cc
@@ -256,6 +256,12 @@ InputOpDetails ConvertOpMetricsToInputOpDetails(const OpMetrics& op_metrics,
   return details;
 }
 
+string AnchorElement(absl::string_view url, absl::string_view text) {
+  return absl::StrCat("<a href=\"", url, "\" target=\"_blank\">", text, "</a>");
+}
+
+}  // namespace
+
 void GenerateHostResult(const OpMetricsDb& host_tf_metrics_db,
                         InputPipelineAnalysisResult* result) {
   InputOpMetrics input_op_metrics = SelectInputOpMetrics(host_tf_metrics_db);
@@ -320,10 +326,6 @@ void GenerateHostResult(const OpMetricsDb& host_tf_metrics_db,
       unclassified_non_enqueue_time_us);
 }
 
-string AnchorElement(absl::string_view url, absl::string_view text) {
-  return absl::StrCat("<a href=\"", url, "\" target=\"_blank\">", text, "</a>");
-}
-
 InputPipelineAnalysisRecommendation GenerateRecommendation() {
   const absl::string_view kDatasetIntro =
       "https://www.tensorflow.org/programmers_guide/datasets";
@@ -365,8 +367,6 @@ InputPipelineAnalysisRecommendation GenerateRecommendation() {
   return recommendation;
 }
 
-}  // namespace
-
 StepSummary ComputeStepTimeSummaryInMs(
     const protobuf::RepeatedPtrField<PerCoreStepInfo>& grouped_by_step) {
   Stat<double> total_step_stats_in_ms;

diff --git a/tensorflow/core/profiler/convert/op_stats_to_input_pipeline_analysis.h b/tensorflow/core/profiler/convert/op_stats_to_input_pipeline_analysis.h
@@ -33,6 +33,11 @@ StepSummary ComputeStepTimeSummaryInMs(
     const ::tensorflow::protobuf::RepeatedPtrField<PerCoreStepInfo>&
         grouped_by_step);
 
+void GenerateHostResult(const OpMetricsDb& host_tf_metrics_db,
+                        InputPipelineAnalysisResult* result);
+
+InputPipelineAnalysisRecommendation GenerateRecommendation();
+
 }  // namespace profiler
 }  // namespace tensorflow