Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dinternal_test.cc34 std::vector<size_t> positional_max = CalculatePositionalMaximums({}); in TEST() local
35 EXPECT_TRUE(positional_max.empty()); in TEST()
52 std::vector<size_t> positional_max = in TEST() local
54 EXPECT_THAT(positional_max, ElementsAre(16)); in TEST()
77 std::vector<size_t> positional_max = in TEST() local
79 EXPECT_THAT(positional_max, ElementsAre(64, 32)); in TEST()
106 std::vector<size_t> positional_max = in TEST() local
108 EXPECT_THAT(positional_max, ElementsAre(64, 32, 8, 8)); in TEST()
Dinternal.cc90 std::vector<size_t> positional_max; in CalculatePositionalMaximums() local
94 for (; i < task_profile.size() && i < positional_max.size(); ++i) { in CalculatePositionalMaximums()
95 positional_max[i] = std::max(positional_max[i], in CalculatePositionalMaximums()
101 positional_max.push_back(task_profile[i].usage_record->tensor_size); in CalculatePositionalMaximums()
104 return positional_max; in CalculatePositionalMaximums()
Dgreedy_by_size_assignment.cc163 std::vector<size_t> positional_max = in GreedyBySizeDistPriorityAssignment() local
178 while (pos < positional_max.size() && in GreedyBySizeDistPriorityAssignment()
179 positional_max[pos] >= priority_info[rec_id].tensor_size) { in GreedyBySizeDistPriorityAssignment()