Home
last modified time | relevance | path

Searched refs:rec_id (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_by_size_assignment.cc168 for (size_t rec_id = 0; rec_id < usage_records.size(); ++rec_id) { in GreedyBySizeDistPriorityAssignment() local
169 priority_info[rec_id].tensor_usage_id = rec_id; in GreedyBySizeDistPriorityAssignment()
170 priority_info[rec_id].tensor_size = usage_records[rec_id].tensor_size; in GreedyBySizeDistPriorityAssignment()
173 priority_info[rec_id].best_dist = kNotAssigned; in GreedyBySizeDistPriorityAssignment()
174 priority_info[rec_id].best_object = kNotAssigned; in GreedyBySizeDistPriorityAssignment()
179 positional_max[pos] >= priority_info[rec_id].tensor_size) { in GreedyBySizeDistPriorityAssignment()
185 priority_info[rec_id].position = pos - 1; in GreedyBySizeDistPriorityAssignment()
231 size_t rec_id = priority_info[info_id].tensor_usage_id; in GreedyBySizeDistPriorityAssignment() local
233 if (assignment->object_ids[rec_id] != kNotAssigned) { in GreedyBySizeDistPriorityAssignment()
245 if (usage_records[rec_id].last_task < in GreedyBySizeDistPriorityAssignment()
[all …]
Dinternal.cc71 for (size_t rec_id = 0; rec_id < usage_records.size(); ++rec_id) { in CalculateTaskProfiles() local
74 for (TaskId task_id = usage_records[rec_id].first_task; in CalculateTaskProfiles()
75 task_id <= usage_records[rec_id].last_task; ++task_id) { in CalculateTaskProfiles()
76 task_profiles[task_id].emplace_back(&usage_records[rec_id], rec_id); in CalculateTaskProfiles()