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.cc561 auto* combined_allocation = &combined_it->second; in CombineTempAllocations() local
570 RoundUpToNearest(combined_allocation->size(), alignment); in CombineTempAllocations()
571 combined_allocation->set_size(base + temp_allocation.size()); in CombineTempAllocations()
576 combined_allocation->AddAssignment(*buffer, base + offset, size); in CombineTempAllocations()
580 combined_allocation->AddHeapTrace(temp_allocation.HeapTraces().front()); in CombineTempAllocations()
582 combined_allocation->peak_buffers_.insert( in CombineTempAllocations()
583 combined_allocation->peak_buffers_.end(), in CombineTempAllocations()