Home
last modified time | relevance | path

Searched refs:allocation_block (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_best_fit_repacker.cc39 for (AllocationBlock* allocation_block : allocations) { in ImportAllocationBlocks()
43 allocation_block->colocations, [&](AllocationBlock* search) { in ImportAllocationBlocks()
46 if (aliased_it != allocation_block->colocations.end()) { in ImportAllocationBlocks()
47 buffer_intervals_[*aliased_it].colocations.push_back(allocation_block); in ImportAllocationBlocks()
50 buffer_intervals_[allocation_block] = {allocation_block, in ImportAllocationBlocks()
51 allocation_block->size, in ImportAllocationBlocks()
52 allocation_block->start_time, in ImportAllocationBlocks()
53 allocation_block->end_time, in ImportAllocationBlocks()
Dmemory_space_assignment.h1310 RepackAllocationBlock allocation_block; in MakeRepackAllocationBlock() local
1311 allocation_block.start_time = start_time; in MakeRepackAllocationBlock()
1312 allocation_block.end_time = end_time; in MakeRepackAllocationBlock()
1313 allocation_block.size = size; in MakeRepackAllocationBlock()
1314 allocation_block.offset = -1; in MakeRepackAllocationBlock()
1315 allocation_block.initial_offset = initial_offset; in MakeRepackAllocationBlock()
1316 allocation_block.id = id; in MakeRepackAllocationBlock()
1317 allocation_block.colocations = {}; in MakeRepackAllocationBlock()
1318 allocation_block.allocation = allocation; in MakeRepackAllocationBlock()
1319 return allocation_block; in MakeRepackAllocationBlock()
Dmemory_space_assignment.cc1809 for (RepackAllocationBlock& allocation_block : repack_allocation_blocks_) { in ExportAllocationsForRepacking()
1810 allocations.push_back(&allocation_block); in ExportAllocationsForRepacking()
1816 for (RepackAllocationBlock& allocation_block : repack_allocation_blocks_) { in ImportRepackedAllocations()
1817 MemorySpaceAssignment::Allocation* allocation = allocation_block.allocation; in ImportRepackedAllocations()
1819 << allocation->chunk().size << ", (" << allocation_block.start_time in ImportRepackedAllocations()
1820 << ", " << allocation_block.end_time << ") from " in ImportRepackedAllocations()
1821 << allocation_block.initial_offset << " to " in ImportRepackedAllocations()
1822 << allocation_block.offset; in ImportRepackedAllocations()
1823 allocation_block.allocation->mutable_chunk()->offset = in ImportRepackedAllocations()
1824 allocation_block.offset; in ImportRepackedAllocations()
[all …]