Searched refs:reallocation_graph (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
| D | types_test.cc | 31 UsageGraph reallocation_graph = ReallocationGraph(deps_graph); in TEST() local
32 EXPECT_TRUE(reallocation_graph.empty()); in TEST()
37 UsageGraph reallocation_graph = ReallocationGraph(deps_graph); in TEST() local
38 ASSERT_EQ(reallocation_graph.size(), 2); in TEST()
39 EXPECT_TRUE(reallocation_graph[0].empty()); in TEST()
40 EXPECT_TRUE(reallocation_graph[1].empty()); in TEST()
45 UsageGraph reallocation_graph = ReallocationGraph(deps_graph); in TEST() local
46 ASSERT_EQ(reallocation_graph.size(), 5); in TEST()
47 EXPECT_THAT(reallocation_graph[0], ElementsAre(2, 3, 4)); in TEST()
48 EXPECT_THAT(reallocation_graph[1], ElementsAre(3, 4)); in TEST()
[all …]
|
| D | types.cc | 52 UsageGraph reallocation_graph(num_vertices); in ReallocationGraph() local
65 reallocation_graph[root].push_back(vertex); in ReallocationGraph()
66 reallocation_graph[vertex].push_back(root); in ReallocationGraph()
71 std::sort(reallocation_graph[vertex].begin(), in ReallocationGraph()
72 reallocation_graph[vertex].end()); in ReallocationGraph()
74 return reallocation_graph; in ReallocationGraph()
|
| D | greedy_in_order_assignment.h | 56 const UsageGraph* reallocation_graph = nullptr) {
80 if (reallocation_graph) {
85 (*reallocation_graph)[last_assigned_tensor[pool_it->object_id]];
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | memory_management.h | 93 const UsageGraph* reallocation_graph = nullptr) {
110 const UsageGraph* reallocation_graph);
116 const UsageGraph* reallocation_graph);
122 const UsageGraph* reallocation_graph);
128 const UsageGraph* reallocation_graph);
135 const UsageGraph* reallocation_graph = nullptr);
|
| D | memory_management.cc | 79 const UsageGraph* reallocation_graph) { in AssignObjectsToTensors() argument
87 reallocation_graph); in AssignObjectsToTensors()
107 const UsageGraph* reallocation_graph) { in AssignObjectsToTensors() argument
124 const UsageGraph* reallocation_graph) { in AssignObjectsToTensors() argument
143 const UsageGraph* reallocation_graph) { in AssignObjectsToTensors() argument
161 const UsageGraph* reallocation_graph) { in AssignOffsetsToTensors() argument
167 usage_records, strategy, &objects_assignment, reallocation_graph)); in AssignOffsetsToTensors()
|