Home
last modified time | relevance | path

Searched refs:tf_function_db (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_tf_functions_test.cc50 TfFunctionDb tf_function_db = ConvertHostThreadsXLineToTfFunctionDb(line); in ConvertXSpaceToTfFunctionDb() local
51 CombineTfFunctionDb(tf_function_db, &result); in ConvertXSpaceToTfFunctionDb()
80 TfFunctionDb tf_function_db = ConvertXSpaceToTfFunctionDb(space); in TEST() local
81 EXPECT_EQ(tf_function_db.tf_functions().size(), 1); in TEST()
82 EXPECT_EQ(tf_function_db.tf_functions().count(kFunctionName), 1); in TEST()
84 tf_function_db.tf_functions().at(kFunctionName); in TEST()
114 TfFunctionDb tf_function_db = ConvertXSpaceToTfFunctionDb(space); in TEST() local
115 EXPECT_EQ(tf_function_db.tf_functions().size(), 2); in TEST()
116 EXPECT_EQ(tf_function_db.tf_functions().count(kOuterFunctionName), 1); in TEST()
117 EXPECT_EQ(tf_function_db.tf_functions().count(kInnerFunctionName), 1); in TEST()
[all …]
Dxplane_to_tf_functions.h31 std::string DebugString(const TfFunctionDb tf_function_db);
Dop_stats_to_overview_page.h65 std::string TfFunctionRecommendationHtml(const TfFunctionDb& tf_function_db);
Dop_stats_to_overview_page.cc311 std::string TfFunctionRecommendationHtml(const TfFunctionDb& tf_function_db) { in TfFunctionRecommendationHtml() argument
313 for (const auto& name_fun : tf_function_db.tf_functions()) { in TfFunctionRecommendationHtml()
382 TfFunctionRecommendationHtml(op_stats.tf_function_db()), in ConvertOpStatsToOverviewPage()
Dxplane_to_tf_functions.cc276 std::string DebugString(const TfFunctionDb& tf_function_db) { in DebugString() argument
278 protobuf::TextFormat::PrintToString(tf_function_db, &str); in DebugString()
Dop_stats_combiner.cc161 CombineTfFunctionDb(src.tf_function_db(), dst->mutable_tf_function_db()); in CombineOpStats()
/external/tensorflow/tensorflow/core/profiler/protobuf/
Dop_stats.proto123 TfFunctionDb tf_function_db = 8; field