Home
last modified time | relevance | path

Searched refs:chunk_map (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator_test.cc238 const int64 offset = result_.chunk_map.size(); in Alloc()
239 result_.chunk_map.emplace(buffer, Chunk{offset, size}); in Alloc()
329 return result_.chunk_map.at(buffer).offset; in OffsetAt()
927 EXPECT_EQ(0, result.chunk_map.size()); in TEST_F()
943 EXPECT_EQ(10, result.chunk_map.at(buffer_a_).size); in TEST_F()
944 EXPECT_EQ(20, result.chunk_map.at(buffer_b_).size); in TEST_F()
945 EXPECT_EQ(30, result.chunk_map.at(buffer_c_).size); in TEST_F()
946 EXPECT_EQ(30, result.chunk_map.at(buffer_d_).size); in TEST_F()
948 EXPECT_EQ(0, result.chunk_map.at(buffer_a_).offset); in TEST_F()
949 EXPECT_EQ(10, result.chunk_map.at(buffer_b_).offset); in TEST_F()
[all …]
Dheap_simulator.cc474 if (!result.chunk_map.empty()) { in Finish()
482 Chunk chunk = FindOrDie(result.chunk_map, group->canonical); in Finish()
484 result.chunk_map.emplace(buffer, chunk); in Finish()
490 CHECK_EQ(options_.buffers_to_assign->size(), result.chunk_map.size()); in Finish()
617 result_.chunk_map.emplace(buffer, Chunk{0, 0}); in Alloc()
634 result_.chunk_map.emplace(buffer, Chunk{new_offset, size}); in Alloc()
666 result_.chunk_map.emplace(buffer, Chunk{new_offset, size}); in Alloc()
679 result_.chunk_map.emplace(buffer, Chunk{kLazyAllocOffset, size}); in Alloc()
683 auto alloc_it = result_.chunk_map.find(buffer); in Free()
684 CHECK(alloc_it != result_.chunk_map.end()) in Free()
[all …]
Dheap_simulator.h61 absl::flat_hash_map<const BufferValue*, Chunk> chunk_map; member
Dbuffer_assignment.cc1335 for (const auto& buffer_chunk : result.chunk_map) { in AssignBuffersFromHeapSimulator()