pm.addPass(mlir::tf_saved_model::CreateAddFunctionsForExportedNamesPass());
  pm.addPass(mlir::tf_saved_model::CreateStripSavedModuleMetadataPass());

  pm.addPass(mlir::TF::CreateRemoveUnusedArgumentsPass());
  pm.addNestedPass<mlir::func::FuncOp>(
      mlir::TF::CreateRemoveUnusedWhileResultsPass());

  pm.addPass(mlir::createInlinerPass());
  pm.addPass(mlir::createCanonicalizerPass());

    op->eraseOperands(operands_to_erase);

    EraseResults(op, results_to_erase.lookup(func));
  });
}

}  // namespace

std::unique_ptr<OperationPass<ModuleOp>> CreateRemoveUnusedArgumentsPass() {
  return std::make_unique<RemoveUnusedArgumentsPass>();
}

}  // namespace TF

// Groups ops into functions that only contain one dialect.
std::unique_ptr<Pass> CreateGroupByDialectPass();

// Removes unused parameters from functions & their callers.
std::unique_ptr<OperationPass<ModuleOp>> CreateRemoveUnusedArgumentsPass();

// Removes unused results from WhileRegion ops.
std::unique_ptr<OperationPass<func::FuncOp>>
CreateRemoveUnusedWhileResultsPass();

// Hoists loop invariant ops to the outside of the loop.

        ...
        call @x_1(x)

      Note that, in the above example, both args would be removed if there
      wasn't the "SomeOpThatUsesArg0(%arg0)" line.
  }];
  let constructor = "TF::CreateRemoveUnusedArgumentsPass()";
}

def RemoveUnusedWhileResultsPass : Pass<"tf-remove-unused-while-results", "mlir::func::FuncOp"> {
  let summary = "Removes unused results from tf.WhileRegion ops";
   let description = [{