op->setAttr(kStablehloEntryFunctionAttrName, SymbolRefAttr::get(*main_func));

  return success();
}

class XlaCallModuleDeserializationPass
    : public impl::XlaCallModuleDeserializationPassBase<
          XlaCallModuleDeserializationPass> {
 public:
  void getDependentDialects(mlir::DialectRegistry &registry) const override {

    const StringRef serialized_stablehlo_module) {
  // StableHLO module is empty often because the XlaCallModuleOp is already
  // deserialized, e.g. after invoking XlaCallModuleDeserializationPass. We
  // don't handle this situation.
  if (serialized_stablehlo_module.empty()) {
    return absl::InvalidArgumentError("StableHLO module is empty.");
  }

  MLIRContext context;

}

//===----------------------------------------------------------------------===//
// XlaCallModule
//===----------------------------------------------------------------------===//

def XlaCallModuleDeserializationPass
    : Pass<"tf-xla-call-module-deserialization", "ModuleOp"> {
  let summary = "Deserializes StableHLO functions embedded in `tf.XlaCallModule` to top level module";

  let description = [{