diff options
-rw-r--r-- | tensorflow/core/common_runtime/shape_refiner.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/shape_refiner.cc b/tensorflow/core/common_runtime/shape_refiner.cc index 7288ecb143..2f65abde0a 100644 --- a/tensorflow/core/common_runtime/shape_refiner.cc +++ b/tensorflow/core/common_runtime/shape_refiner.cc @@ -203,6 +203,9 @@ Status ShapeRefiner::EvaluateConstantTensorForEdge(const Node* node, bool is_constant_graph = false; Graph subgraph(ops_registry_); + auto versions = subgraph.versions(); + versions.set_producer(graph_def_version_); + subgraph.set_versions(versions); // We identify the possibly constant subgraph to evaluate by // recursively iterating backwards through the inputs to 'node' |