diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-13 16:08:12 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 16:12:39 -0700 |
| commit | 29b56bde1e28e558111b917fd44b973e2aea7fcf (patch) | |
| tree | 5de81939bd5226b22deb304d4e3f6914fdb04c0c /tensorflow/core/common_runtime/graph_execution_state.cc | |
| parent | 4292b8107175b3c3223f65c75b3ca091bd0604ec (diff) | |
Automated rollback of commit ac60b46e2c5962fd8099a4406c1788d826ad3c0d
PiperOrigin-RevId: 212896336
Diffstat (limited to 'tensorflow/core/common_runtime/graph_execution_state.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/graph_execution_state.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/graph_execution_state.cc b/tensorflow/core/common_runtime/graph_execution_state.cc index 7f260b3139..4475fa979e 100644 --- a/tensorflow/core/common_runtime/graph_execution_state.cc +++ b/tensorflow/core/common_runtime/graph_execution_state.cc @@ -561,6 +561,10 @@ Status GraphExecutionState::OptimizeGraph( grappler::GrapplerItem item; item.id = "tf_graph"; graph_->ToGraphDef(&item.graph); + // TODO(b/114748242): Add a unit test to test this bug fix. + if (flib_def_) { + *item.graph.mutable_library() = flib_def_->ToProto(); + } item.fetch.insert(item.fetch.end(), options.callable_options.fetch().begin(), |