Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_in_order_assignment.h81 for (auto pool_it = pool.begin(); pool_it != pool.end(); ++pool_it) { variable
82 size_t size_diff = AbsDiffInElements(pool_it->object_size, tensor_size);
85 (*reallocation_graph)[last_assigned_tensor[pool_it->object_id]];
93 best_it = pool_it;
100 auto pool_it = pool.lower_bound({tensor_size, 0}); variable
102 if (pool_it != pool.end()) {
104 size_diff = pool_it->object_size - tensor_size;
105 best_it = pool_it;
107 if (pool_it != pool.begin()) {
109 pool_it--;
[all …]
Dequality_assignment.h58 auto pool_it = pool.find(tensor_size); in EqualityAssignmentWithHash() local
59 if (pool_it == pool.end() || pool_it->second.empty()) { in EqualityAssignmentWithHash()
69 assignment->object_ids[i] = pool_it->second.back(); in EqualityAssignmentWithHash()
70 pool_it->second.pop_back(); in EqualityAssignmentWithHash()