Searched refs:chunk_map (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator_test.cc | 241 const int64 offset = result_.chunk_map.size(); in Alloc()
242 result_.chunk_map.emplace(buffer, Chunk{offset, size}); in Alloc()
251 const int64 offset = result_.chunk_map[shared].offset; in ShareWith()
252 result_.chunk_map.emplace(buffer, Chunk{offset, size}); in ShareWith()
342 return result_.heap_results.at(0).chunk_map.at(buffer).offset; in OffsetAt()
1059 EXPECT_EQ(0, result.heap_results.at(0).chunk_map.size()); in TEST_F()
1091 EXPECT_EQ(10, result.chunk_map.at(buffer_a_).size); in TEST_F()
1092 EXPECT_EQ(30, result.chunk_map.at(buffer_b_).size); in TEST_F()
1093 EXPECT_EQ(20, result.chunk_map.at(buffer_c_).size); in TEST_F()
1094 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 | 415 return lhs + rhs.chunk_map.size(); in Finish()
564 result_.chunk_map.emplace(buffer, Chunk{0, 0}); in Alloc()
579 result_.chunk_map.emplace(buffer, Chunk{0, 0}); in ShareWith()
975 const auto emplace_result = result_.chunk_map.emplace(buffer, chunk); in AddToChunkMap()
|
| D | heap_simulator.h | 73 absl::flat_hash_map<const BufferType*, Chunk> chunk_map; member
|
| D | buffer_assignment.cc | 1611 for (const auto& buffer_chunk : heap_result.chunk_map) { in AssignBuffersFromHeapSimulator()
|