diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-11-10 10:12:35 -0800 |
|---|---|---|
| committer |  Andrew Selle <aselle@andyselle.com> | 2017-11-10 16:14:42 -0800 |
| commit | d82ab02a926c3564c44a19deaf298afc10326bf6 (patch) | |
| tree | 6678f1e67ffb56cefb9abdac147eb1531cef650f /tensorflow/compiler/xla/service/shaped_buffer.h | |
| parent | 813451826892869acec35643bc3ca9800437a00b (diff) | |
[TF:XLA] Add helper to append buffers to ShapedBuffer.
PiperOrigin-RevId: 175304150
Diffstat (limited to 'tensorflow/compiler/xla/service/shaped_buffer.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/shaped_buffer.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shaped_buffer.h b/tensorflow/compiler/xla/service/shaped_buffer.h index e5ea06fb13..b440948700 100644 --- a/tensorflow/compiler/xla/service/shaped_buffer.h +++ b/tensorflow/compiler/xla/service/shaped_buffer.h @@ -75,6 +75,10 @@ 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_; |