auto result = func_op->walk([&](TF::XlaCallModuleOp op) {
    // Converts the serialized StableHLO module to bfloat16.
    auto result =
        ConvertSerializedStableHloModuleToBfloat16(op.getModuleAttr());
    if (!result.ok()) {
      llvm::errs() << "Failed to convert StableHLO module to bfloat16: "
                   << result.status().message();
      return WalkResult::interrupt();
    }

                                PatternRewriter &rewriter) const override {
    OwningOpRef<ModuleOp> stablehlo_module_op =
        stablehlo::deserializePortableArtifact(op.getModuleAttr(),
                                               getContext());
    if (stablehlo_module_op.get() == nullptr) {
      return failure();
    }
    SymbolTable parent_module_symbol_table(module_op_);

  // quantized.
  auto new_call_op = rewriter.create<TF::XlaCallModuleOp>(
      call_op.getLoc(), call_op.getResultTypes(), call_op.getOperands(),
      call_op.getVersionAttr(), call_op.getModuleAttr(), call_op.getSoutAttr());
  new_call_op->setAttr(kEntryFuncAttrName,
                       rewriter.getStringAttr(new_ref_func_name.getValue()));
  new_call_op->setAttrs(call_op->getAttrs());