Searched refs:peak_buffers_ (Results 1 – 2 of 2) sorted by relevance
259 return peak_buffers_; in PeakMemoryLogicalBuffers()346 std::vector<const HloValue*> peak_buffers_; variable
568 allocation->peak_buffers_.push_back(buffer.values()[0]); in NewAllocation()674 combined_allocation->peak_buffers_.insert( in CombineTempAllocations()675 combined_allocation->peak_buffers_.end(), in CombineTempAllocations()676 temp_allocation.peak_buffers_.begin(), in CombineTempAllocations()677 temp_allocation.peak_buffers_.end()); in CombineTempAllocations()1616 allocation->peak_buffers_ = in AssignBuffersFromHeapSimulator()