Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc652 BufferAllocation* combined_allocation = combined_it->second; in CombineTempAllocations() local
661 RoundUpToNearest(combined_allocation->size(), alignment); in CombineTempAllocations()
662 combined_allocation->set_size(base + temp_allocation.size()); in CombineTempAllocations()
667 combined_allocation->AddAssignment(*value, base + offset, size); in CombineTempAllocations()
671 combined_allocation->AddHeapTrace(temp_allocation.HeapTraces().front()); in CombineTempAllocations()
674 combined_allocation->peak_buffers_.insert( in CombineTempAllocations()
675 combined_allocation->peak_buffers_.end(), in CombineTempAllocations()