Searched refs:chunk_map (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator_test.cc | 240 const int64_t offset = result_.chunk_map.size(); in Alloc()
241 result_.chunk_map.emplace(buffer, Chunk{offset, size}); in Alloc()
250 const int64_t offset = result_.chunk_map[shared].offset; in ShareWith()
251 result_.chunk_map.emplace(buffer, Chunk{offset, size}); in ShareWith()
342 return result_.heap_results.at(0).chunk_map.at(buffer).offset; in OffsetAt()
1058 EXPECT_EQ(0, result.heap_results.at(0).chunk_map.size()); in TEST_F()
1090 EXPECT_EQ(10, result.chunk_map.at(buffer_a_).size); in TEST_F()
1091 EXPECT_EQ(30, result.chunk_map.at(buffer_b_).size); in TEST_F()
1092 EXPECT_EQ(20, result.chunk_map.at(buffer_c_).size); in TEST_F()
1093 EXPECT_EQ(40, result.chunk_map.at(buffer_d_).size); in TEST_F()
[all …]
|
| D | memory_space_assignment_best_fit_repacker.cc | 64 auto chunk_it = result_.chunk_map.find(block); in Repack()
65 if (chunk_it != result_.chunk_map.end()) { in Repack()
|
| D | heap_simulator.cc | 418 return lhs + rhs.chunk_map.size(); in Finish()
550 result_.chunk_map.emplace(buffer, Chunk{0, 0}); in Alloc()
565 result_.chunk_map.emplace(buffer, Chunk{0, 0}); in ShareWith()
976 const auto emplace_result = result_.chunk_map.emplace(buffer, chunk); in AddToChunkMap()
|
| D | heap_simulator.h | 79 absl::flat_hash_map<const BufferType*, Chunk> chunk_map; member
|
| D | buffer_assignment.cc | 1699 for (const auto& buffer_chunk : heap_result.chunk_map) { in AssignBuffersFromHeapSimulator()
|