Home
last modified time | relevance | path

Searched refs:task_profiles (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dinternal_test.cc32 std::vector<TaskProfile> task_profiles = CalculateTaskProfiles({}); in TEST() local
33 EXPECT_TRUE(task_profiles.empty()); in TEST()
42 std::vector<TaskProfile> task_profiles = CalculateTaskProfiles(usage_records); in TEST() local
43 ASSERT_EQ(task_profiles.size(), correct_idx.size()); in TEST()
44 for (size_t i = 0; i < task_profiles.size(); ++i) { in TEST()
45 ASSERT_EQ(task_profiles[i].size(), correct_idx[i].size()); in TEST()
46 for (size_t j = 0; j < task_profiles[i].size(); ++j) { in TEST()
47 ASSERT_EQ(task_profiles[i][j].usage_record, in TEST()
49 ASSERT_EQ(task_profiles[i][j].idx, correct_idx[i][j]); in TEST()
67 std::vector<TaskProfile> task_profiles = CalculateTaskProfiles(usage_records); in TEST() local
[all …]
Dinternal.cc70 std::vector<TaskProfile> task_profiles(num_tasks); in CalculateTaskProfiles() local
76 task_profiles[task_id].emplace_back(&usage_records[rec_id], rec_id); in CalculateTaskProfiles()
81 for (auto& task_profile : task_profiles) { in CalculateTaskProfiles()
84 return task_profiles; in CalculateTaskProfiles()
89 std::vector<TaskProfile> task_profiles = CalculateTaskProfiles(usage_records); in CalculatePositionalMaximums() local
91 for (const auto& task_profile : task_profiles) { in CalculatePositionalMaximums()
Dgreedy_by_breadth_assignment.cc53 std::vector<TaskProfile> task_profiles = CalculateTaskProfiles(usage_records); in GreedyByBreadthAssignment() local
57 for (size_t task_id = 0; task_id < task_profiles.size(); ++task_id) { in GreedyByBreadthAssignment()
59 for (const auto& tensor_info : task_profiles[task_id]) { in GreedyByBreadthAssignment()
74 for (const auto& tensor_info : task_profiles[task.task_id]) { in GreedyByBreadthAssignment()