From 9cf6fe1ad3843383b527ee718203763036c6e396 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 20 Jul 2017 17:43:18 -0700 Subject: [PATCH] Allow access to the module BufferLiveness applies to. PiperOrigin-RevId: 162690202 --- tensorflow/compiler/xla/service/buffer_liveness.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/compiler/xla/service/buffer_liveness.h b/tensorflow/compiler/xla/service/buffer_liveness.h index 24807551f1e..70d642b40c8 100644 --- a/tensorflow/compiler/xla/service/buffer_liveness.h +++ b/tensorflow/compiler/xla/service/buffer_liveness.h @@ -67,6 +67,8 @@ class BufferLiveness { // Returns the underlying hlo ordering used for this liveness analysis. const HloOrdering& hlo_ordering() const { return *hlo_ordering_; } + const HloModule& module() const { return *module_; } + string ToString() const; static Colorer DefaultColorer() {