Return the original graph is optimization is not applied.

PiperOrigin-RevId: 1585830
TeeeJay · Jun 10, 2017 · ef8b5fd · ef8b5fd
1 parent f77cd10
commit ef8b5fd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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();
  }