Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc549 const BufferAllocation& temp_allocation = *it; in CombineTempAllocations() local
550 LogicalBuffer::Color color = temp_allocation.color(); in CombineTempAllocations()
556 << " is: " << temp_allocation; in CombineTempAllocations()
557 combined_allocation_map.emplace(color, temp_allocation); in CombineTempAllocations()
563 << temp_allocation; in CombineTempAllocations()
571 combined_allocation->set_size(base + temp_allocation.size()); in CombineTempAllocations()
572 for (const auto& buffer_offset_size : temp_allocation.assigned_buffers_) { in CombineTempAllocations()
578 if (!temp_allocation.HeapTraces().empty()) { in CombineTempAllocations()
579 CHECK_EQ(temp_allocation.HeapTraces().size(), 1); in CombineTempAllocations()
580 combined_allocation->AddHeapTrace(temp_allocation.HeapTraces().front()); in CombineTempAllocations()
[all …]