diff options
| author |  Igor Ganichev <iga@google.com> | 2018-05-07 15:15:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-07 17:20:38 -0700 |
| commit | 860452f3346e3a450782046081dc4d3263544344 (patch) | |
| tree | 90157a446d582bbda004ec7f61ad1eb49a97d19b | |
| parent | 80c7ebc32324d689ae3feb17a0cb4cf32d736f19 (diff) | |
Replace references to TensorInfo with XlaTensor
PiperOrigin-RevId: 195730139
| -rw-r--r-- | tensorflow/compiler/jit/xla_tensor.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/jit/xla_tensor.h b/tensorflow/compiler/jit/xla_tensor.h index 922a918973..6b29c82ec1 100644 --- a/tensorflow/compiler/jit/xla_tensor.h +++ b/tensorflow/compiler/jit/xla_tensor.h @@ -54,7 +54,7 @@ class XlaTensor { // Some Tensors can have complex on-device shapes, including tuple shapes. To // manage the memory for these tensors a ShapedBuffer may be required. - // Return true if this TensorInfo contains a ShapedBuffer. + // Return true if this XlaTensor contains a ShapedBuffer. bool has_shaped_buffer() const { return shaped_buffer_ != nullptr; } // Return the contained ShapedBuffer. // REQUIRES: has_shaped_buffer() @@ -62,7 +62,7 @@ class XlaTensor { CHECK(has_shaped_buffer()); return *shaped_buffer_; } - // Mutates the TensorInfo to set the ShapedBuffer. + // Mutates the XlaTensor to set the ShapedBuffer. void set_shaped_buffer(xla::ScopedShapedBuffer shaped_buffer) { shaped_buffer_ = xla::MakeUnique<xla::ScopedShapedBuffer>(std::move(shaped_buffer)); @@ -72,7 +72,7 @@ class XlaTensor { // in on-demand mode to avoid re-copying values from the device if we know the // host value already. - // Return true if this TensorInfo contains a host tensor. + // Return true if this XlaTensor contains a host tensor. bool has_host_tensor() const { return host_tensor_ != nullptr; } // Return the contained host tensor. // REQUIRES: has_host_tensor() |