Method SetBuffer

tensorflow/compiler/xla/service/gpu/buffer_allocations.cc:172–177

}

void BufferAllocations::SetBuffer(BufferAllocation::Index buffer_index,
                                  se::DeviceMemoryBase buffer) {
  CHECK_GE(buffer_index, 0);
  CHECK_LT(buffer_index, buffers_.size());
  buffers_[buffer_index] = buffer;
}

BufferAllocations::KeyType BufferAllocations::Key() const {
  KeyType key;