module.walk([&](func::FuncOp f) {
    if (f->hasAttr(kFromXlaCallModuleAttrName)) {
      f->erase();
    }
  });
}

class XlaCallModuleSerializationPass
    : public impl::XlaCallModuleSerializationPassBase<
          XlaCallModuleSerializationPass> {
 public:
  void runOnOperation() override {
    mlir::ModuleOp module = getOperation();
    mlir::SymbolTableCollection symbol_table;

    "stablehlo::StablehloDialect",
    "vhlo::VhloDialect",
    "quant::QuantizationDialect",
  ];

  let constructor = "TF::CreateXlaCallModuleDeserializationPass()";
}

def XlaCallModuleSerializationPass
    : Pass<"tf-xla-call-module-serialization", "ModuleOp"> {
  let summary = "Serializes StableHLO functions from top-level module into `tf.XlaCallModule`'s `module` attribute";

  let description = [{