Internal change

PiperOrigin-RevId: 448762469
nxp-upstream · May 15, 2022 · cc78e6b · cc78e6b
1 parent 2d05c8b
commit cc78e6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc
@@ -1243,7 +1243,7 @@ Status MetaOptimizer::OptimizeConsumeItem(Cluster* cluster, GrapplerItem&& item,
   // opportunities for other optimizers; they could, but it's unclear whether
   // re-running all the other optimizers is worthwhile.
 #ifndef __Fuchsia__
-  {
+  if (!is_tpu_graph) {
     // Create a Grappler optimization pipeline with only the TFG optimizer.
     std::vector<std::unique_ptr<GraphOptimizer>> optimizers;
     optimizers.push_back(std::make_unique<mlir::tfg::TFGGrapplerOptimizer>(