diff options
-rw-r--r-- | tensorflow/core/grappler/optimizers/layout_optimizer.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/layout_optimizer.cc b/tensorflow/core/grappler/optimizers/layout_optimizer.cc index 0cc68632bf..0a802fa0f5 100644 --- a/tensorflow/core/grappler/optimizers/layout_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/layout_optimizer.cc @@ -1170,6 +1170,7 @@ Status LayoutOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, } if (num_gpus_ < 1) { // LayoutOptimizer is currently only tuned for GPU. + *output = item.graph; return Status::OK(); } |