Home
last modified time | relevance | path

Searched refs:kernel_stats_db (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Dkernel_stats_utils.cc233 void SortAndKeepTopKDurationKernelReportsInDb(KernelStatsDb* kernel_stats_db) { in SortAndKeepTopKDurationKernelReportsInDb() argument
241 if (kernel_stats_db->reports_size() > kMaxNumOfKernels) { in SortAndKeepTopKDurationKernelReportsInDb()
243 kernel_stats_db->mutable_reports()->begin(), in SortAndKeepTopKDurationKernelReportsInDb()
244 kernel_stats_db->mutable_reports()->begin() + kMaxNumOfKernels, in SortAndKeepTopKDurationKernelReportsInDb()
245 kernel_stats_db->mutable_reports()->end(), comp); in SortAndKeepTopKDurationKernelReportsInDb()
246 kernel_stats_db->mutable_reports()->erase( in SortAndKeepTopKDurationKernelReportsInDb()
247 kernel_stats_db->mutable_reports()->begin() + kMaxNumOfKernels, in SortAndKeepTopKDurationKernelReportsInDb()
248 kernel_stats_db->mutable_reports()->end()); in SortAndKeepTopKDurationKernelReportsInDb()
250 std::sort(kernel_stats_db->mutable_reports()->begin(), in SortAndKeepTopKDurationKernelReportsInDb()
251 kernel_stats_db->mutable_reports()->end(), comp); in SortAndKeepTopKDurationKernelReportsInDb()
[all …]
Dkernel_stats_utils_test.cc27 KernelStatsDb kernel_stats_db; in TEST() local
28 KernelReport* kernel_report_1 = kernel_stats_db.add_reports(); in TEST()
35 KernelReport* kernel_report_2 = kernel_stats_db.add_reports(); in TEST()
42 KernelReport* kernel_report_3 = kernel_stats_db.add_reports(); in TEST()
50 GroupKernelReportsByOpName(kernel_stats_db); in TEST()
Dkernel_stats_utils.h55 void SortAndKeepTopKDurationKernelReportsInDb(KernelStatsDb* kernel_stats_db);
130 const KernelStatsDb& kernel_stats_db);
/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stats_to_tf_stats.cc107 GroupKernelReportsByOpName(op_stats.kernel_stats_db()); in ConvertOpStatsToTfStats()
Dxplane_to_profile_response.cc126 AddToolData(ToolName(kKernelStats), op_stats.kernel_stats_db(), response); in ConvertXSpaceToProfileResponse()
Dxplane_to_tools_data.cc117 return combined_op_stats.kernel_stats_db().SerializeAsString(); in ConvertMultiXSpacesToKernelStats()
Dop_stats_combiner.cc154 src.kernel_stats_db().reports()); in CombineOpStats()
Dop_stats_to_overview_page.cc181 GroupKernelReportsByOpName(op_stats.kernel_stats_db()); in ComputeAnalysisResult()
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dop_stats.proto126 KernelStatsDb kernel_stats_db = 6; field