Reverts 6a9740c

PiperOrigin-RevId: 931763149

Author: tensorflower-gardener

tensorflow/compiler/mlir/tfrt/transforms/mlrt/import_model.cc

@@ -120,11 +120,11 @@ absl::StatusOr<mlrt::bc::Buffer> ConvertTfMlirToBytecode(
 
         mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext());
 
-        tensorflow::CreateTFInvariantOptimizationPipelineHelper(pm, options);
-
         pm.addPass(mlrt_compiler::CreateWhileToMapFnPass());
         // Remove unreachable private functions after map_fn conversion.
         pm.addPass(mlir::createSymbolDCEPass());
+
+        tensorflow::CreateTFInvariantOptimizationPipelineHelper(pm, options);
         // TODO(b/283481729): Add test to cover unused constants that do not
         // cause op_key discontinuity
         pm.addNestedPass<mlir::func::FuncOp>(mlir::createCanonicalizerPass());