Home
last modified time | relevance | path

Searched refs:repack_allocation_blocks_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc1612 CHECK_EQ(repack_allocation_blocks_.size(), 0); in AllocateCrossProgramPrefetchBuffer()
1615 repack_allocation_blocks_.push_back(MakeRepackAllocationBlock( in AllocateCrossProgramPrefetchBuffer()
1618 static_cast<int64>(repack_allocation_blocks_.size()), in AllocateCrossProgramPrefetchBuffer()
1620 RepackAllocationBlock* inserted = &repack_allocation_blocks_.back(); in AllocateCrossProgramPrefetchBuffer()
1621 for (RepackAllocationBlock& colocation : repack_allocation_blocks_) { in AllocateCrossProgramPrefetchBuffer()
1809 for (RepackAllocationBlock& allocation_block : repack_allocation_blocks_) { in ExportAllocationsForRepacking()
1816 for (RepackAllocationBlock& allocation_block : repack_allocation_blocks_) { in ImportRepackedAllocations()
1905 repack_allocation_blocks_.push_back(MakeRepackAllocationBlock( in FinalizeAllocations()
1909 static_cast<int64>(repack_allocation_blocks_.size()), in FinalizeAllocations()
1911 colocations.push_back(&repack_allocation_blocks_.back()); in FinalizeAllocations()
Dmemory_space_assignment.h1334 std::list<RepackAllocationBlock> repack_allocation_blocks_; variable