Automated g4 rollback of changelist 175304150
PiperOrigin-RevId: 175571632
This commit is contained in:
parent
e8da93bde5
commit
333bdea952
tensorflow/compiler/xla/service
@ -63,14 +63,6 @@ void ShapedBuffer::clear() {
|
||||
}
|
||||
}
|
||||
|
||||
void ShapedBuffer::AddBufferAtIndex(
|
||||
const perftools::gputools::DeviceMemoryBase& buffer,
|
||||
const ShapeIndex& shape_index) {
|
||||
*mutable_shape_index_to_buffer_entry()->mutable_element(shape_index) =
|
||||
buffers().size();
|
||||
mutable_buffers()->push_back(buffer);
|
||||
}
|
||||
|
||||
const se::DeviceMemoryBase& ShapedBuffer::buffer(
|
||||
const ShapeIndex& index) const {
|
||||
return buffers_[shape_index_to_buffer_entry_.element(index)];
|
||||
|
@ -75,10 +75,6 @@ class ShapedBuffer {
|
||||
// Set all device memory pointers in the object to null.
|
||||
void clear();
|
||||
|
||||
// Adds a new buffer at the given shape index.
|
||||
void AddBufferAtIndex(const perftools::gputools::DeviceMemoryBase& buffer,
|
||||
const ShapeIndex& shape_index);
|
||||
|
||||
protected:
|
||||
// The shape of the device buffer with layout.
|
||||
const Shape shape_;
|
||||
|
Loading…
Reference in New Issue
Block a user