diff options
| author |  Jeremy Lau <lauj@google.com> | 2018-05-02 17:25:10 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-02 17:28:24 -0700 |
| commit | a1ef905926d12b0362c0dcf6d669e1c3d2ffcf70 (patch) | |
| tree | b201ca5878acec8c9ba2306d4393d7a26e23a058 /tensorflow/compiler/xla/service/compiler.h | |
| parent | dde83d4bee2c524cb5bd0adc4f702c9fc5ac6f3f (diff) | |
BufferValue is a new base class for LogicalBuffer and HloValue. This makes it
easier to migrate from TuplePointsToAnalysis/LogicalBuffer to
HloDataflowAnalysis/HloValue. No functional changes.
PiperOrigin-RevId: 195179676
Diffstat (limited to 'tensorflow/compiler/xla/service/compiler.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/compiler.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/compiler.h b/tensorflow/compiler/xla/service/compiler.h index 5c14591d93..a4b59d1ba9 100644 --- a/tensorflow/compiler/xla/service/compiler.h +++ b/tensorflow/compiler/xla/service/compiler.h @@ -25,6 +25,7 @@ limitations under the License. #include <memory> #include <string> +#include "tensorflow/compiler/xla/service/buffer_value.h" #include "tensorflow/compiler/xla/service/executable.h" #include "tensorflow/compiler/xla/service/hlo_module.h" #include "tensorflow/compiler/xla/service/hlo_module_config.h" @@ -181,9 +182,9 @@ class Compiler { // Returns a function that computes the size in bytes of a given // logical buffer. - std::function<int64(const LogicalBuffer&)> BufferSizeBytesFunction() { + std::function<int64(const BufferValue&)> BufferSizeBytesFunction() { HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); - return [shape_size](const LogicalBuffer& buffer) { + return [shape_size](const BufferValue& buffer) { return shape_size(buffer.shape()); }; } |