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.cc229 const OffsetSize os = FindOrDie(assigned_buffers_, &buffer); in GetSlice()
236 CHECK(!assigned_buffers_.contains(&buffer)) in AddAssignment()
250 assigned_buffers_.emplace(&buffer, offset_size); in AddAssignment()
269 for (const auto& buffer_offset_size : assigned_buffers_) { in ToProto()
309 for (const auto& buffer_offset_size : assigned_buffers_) { in ToString()
317 const OffsetSize& offset_size = FindOrDie(assigned_buffers_, buffer); in ToString()
572 for (const auto& buffer_offset_size : temp_allocation.assigned_buffers_) { in CombineTempAllocations()
602 for (const auto& buffer_offset_size : allocation->assigned_buffers_) { in CombineTempAllocations()
Dbuffer_assignment.h157 return assigned_buffers_; in assigned_buffers()
336 absl::flat_hash_map<const LogicalBuffer*, OffsetSize> assigned_buffers_; variable