diff options
Diffstat (limited to 'tensorflow/core/grappler/optimizers/meta_optimizer.cc')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/meta_optimizer.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc index e778b7879d..5fd34efeb1 100644 --- a/tensorflow/core/grappler/optimizers/meta_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/meta_optimizer.cc @@ -361,7 +361,8 @@ Status MetaOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, // Make a GrapplerItem from a FunctionDef. GrapplerFunctionItem func_item; - TF_RETURN_IF_ERROR(MakeGrapplerFunctionItem(func, flib, &func_item)); + TF_RETURN_IF_ERROR(MakeGrapplerFunctionItem( + func, flib, item.graph.versions().producer(), &func_item)); // Optimize function body graph. GraphDef optimized_func_graph; |