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/cc/framework/scope.cc | |
| parent | 9a47c258c9c2286ae2c14a0da6458055f3b691d3 (diff) | |
Automated g4 rollback of changelist 156251356
PiperOrigin-RevId: 156315860
Diffstat (limited to 'tensorflow/cc/framework/scope.cc')
| -rw-r--r-- | tensorflow/cc/framework/scope.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/cc/framework/scope.cc b/tensorflow/cc/framework/scope.cc index 8b7fc1406f..32c0822de6 100644 --- a/tensorflow/cc/framework/scope.cc +++ b/tensorflow/cc/framework/scope.cc @@ -271,9 +271,9 @@ Scope::Impl::Impl(const Scope& other, Tags::Colocate, std::unordered_set<string> Scope::Impl::GetColocationConstraints( const Operation& colocate_with_op) const { std::unordered_set<string> current_constraints(colocation_constraints_); - const NodeDef& node_def = colocate_with_op.node()->def(); + const AttrSlice attrs = colocate_with_op.node()->attrs(); std::vector<string> node_constraints; - if (GetNodeAttr(node_def, kColocationAttrName, &node_constraints).ok()) { + if (GetNodeAttr(attrs, kColocationAttrName, &node_constraints).ok()) { for (const string& entry : node_constraints) { StringPiece s(entry); if (s.Consume(kColocationGroupPrefix)) { |