Searched refs:assigned_buffers_ (Results 1 – 2 of 2) sorted by relevance
235 const OffsetSize os = FindOrDie(assigned_buffers_, &buffer); in GetSlice()244 CHECK(!assigned_buffers_.contains(&buffer)) in AddAssignment()258 assigned_buffers_.emplace(&buffer, offset_size); in AddAssignment()286 for (const auto& buffer_offset_size : assigned_buffers_) { in ToProto()369 for (const auto& buffer_offset_size : assigned_buffers_) { in ToString()374 const OffsetSize& offset_size = FindOrDie(assigned_buffers_, buffer); in ToString()663 for (const auto& buffer_offset_size : temp_allocation.assigned_buffers_) { in CombineTempAllocations()694 for (const auto& buffer_offset_size : allocation->assigned_buffers_) { in CombineTempAllocations()
162 return assigned_buffers_; in assigned_buffers()340 absl::flat_hash_map<const HloValue*, OffsetSize> assigned_buffers_; variable