diff options
| author |  Geoffrey Irving <geoffreyi@google.com> | 2017-05-17 09:23:54 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-17 09:27:36 -0700 |
| commit | 73882f257ffb1bc9e1a828571c085d080b1d9266 (patch) | |
| tree | 8adcefa226f95d6c6ce067ee45528d76794e55fb /tensorflow/compiler/jit/mark_for_compilation_pass_test.cc | |
| parent | 9a47c258c9c2286ae2c14a0da6458055f3b691d3 (diff) | |
Automated g4 rollback of changelist 156251356
PiperOrigin-RevId: 156315860
Diffstat (limited to 'tensorflow/compiler/jit/mark_for_compilation_pass_test.cc')
| -rw-r--r-- | tensorflow/compiler/jit/mark_for_compilation_pass_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc b/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc index 91e4a2b41c..9f30e12e0e 100644 --- a/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc +++ b/tensorflow/compiler/jit/mark_for_compilation_pass_test.cc @@ -57,7 +57,7 @@ std::unordered_map<string, string> GetClusters(const Graph& graph) { std::unordered_map<string, string> ids; for (Node* node : graph.nodes()) { string cluster; - if (GetNodeAttr(node->def(), kXlaClusterAttr, &cluster).ok()) { + if (GetNodeAttr(node->attrs(), kXlaClusterAttr, &cluster).ok()) { CHECK(!cluster.empty()); ids[node->name()] = cluster; } |