HomeSort by relevance Sort by last modified time
    Searched refs:assigned_buffers_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.cc 229 const OffsetSize os = FindOrDie(assigned_buffers_, &buffer);
236 CHECK(!assigned_buffers_.contains(&buffer))
250 assigned_buffers_.emplace(&buffer, offset_size);
269 for (const auto& buffer_offset_size : assigned_buffers_) {
309 for (const auto& buffer_offset_size : assigned_buffers_) {
317 const OffsetSize& offset_size = FindOrDie(assigned_buffers_, buffer);
572 for (const auto& buffer_offset_size : temp_allocation.assigned_buffers_) {
602 for (const auto& buffer_offset_size : allocation->assigned_buffers_) {
    [all...]
buffer_assignment.h 157 return assigned_buffers_;
336 absl::flat_hash_map<const LogicalBuffer*, OffsetSize> assigned_buffers_; member in class:xla::BufferAllocation

Completed in 936 milliseconds