experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add operator less to TensorUsageRecord; Small fix in tests.

Browse Source 
PiperOrigin-RevId: 248616927
This commit is contained in:
A. Unique TensorFlower 2019-05-16 15:53:42 -07:00 committed by TensorFlower Gardener
parent bb15fff6b9
commit bd0dbb9b54
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/delegates/gpu/common/memory_management.h
View File

@ -40,6 +40,11 @@ struct TensorUsageRecord {
TensorUsageRecord(uint32_t size, TaskId first, TaskId last)
: tensor_size(size), first_task(first), last_task(last) {}
// Default order of tensor usage records is increasing order of first_task.
bool operator<(const TensorUsageRecord& other) const {
return first_task < other.first_task;
}
};
// Information about assignment of tensors to shared objects

12
tensorflow/lite/delegates/gpu/common/memory_management_test.cc
View File

@ -28,17 +28,17 @@ TEST(Model, EmptyRecords) {
ObjectsAssignment assignment;
ASSERT_TRUE(
AssignObjectsToTensors({}, MemoryStrategy::NAIVE, &assignment).ok());
ASSERT_TRUE(assignment.object_ids.empty());
ASSERT_TRUE(assignment.object_sizes.empty());
EXPECT_TRUE(assignment.object_ids.empty());
EXPECT_TRUE(assignment.object_sizes.empty());
ASSERT_TRUE(
AssignObjectsToTensors({}, MemoryStrategy::GREEDY, &assignment).ok());
ASSERT_TRUE(assignment.object_ids.empty());
ASSERT_TRUE(assignment.object_sizes.empty());
EXPECT_TRUE(assignment.object_ids.empty());
EXPECT_TRUE(assignment.object_sizes.empty());
ASSERT_TRUE(
AssignObjectsToTensors({}, MemoryStrategy::MINCOSTFLOW, &assignment)
.ok());
ASSERT_TRUE(assignment.object_ids.empty());
ASSERT_TRUE(assignment.object_sizes.empty());
EXPECT_TRUE(assignment.object_ids.empty());
EXPECT_TRUE(assignment.object_sizes.empty());
}
TEST(Model, OneRecord) {