experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow access to the module BufferLiveness applies to.

Browse Source 
PiperOrigin-RevId: 162690202
This commit is contained in:
A. Unique TensorFlower 2017-07-20 17:43:18 -07:00 committed by TensorFlower Gardener
parent 41ae3a8c0f
commit 9cf6fe1ad3
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/xla/service/buffer_liveness.h
View File

@ -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() {