Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.cc235 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()
Dbuffer_assignment.h162 return assigned_buffers_; in assigned_buffers()
340 absl::flat_hash_map<const HloValue*, OffsetSize> assigned_buffers_; variable