class_docstr: A safety check should be skipped on (de)serialization.
  
  Most of these checks are performed on serialization, but some are deferred to
  deserialization. The list of disabled checks is attached to the serialization,
  e.g., as a sequence of string attributes to `jax_export.Exported` or of
  `tf.XlaCallModuleOp`.
  
  You can disable more deserialization safety checks by passing
  `TF_XLA_FLAGS=--tf_xla_call_module_disabled_checks=platform`.
class_docstr_tok: ["A","safety","check","should","be","skipped","on","(","de",")","serialization",".","Most","of","these","checks","are","performed","on","serialization",",","but","some","are","deferred","to","deserialization",".","The","list","of","disabled","checks","is","attached","to","the","serialization",",","e.g.",",","as","a","sequence","of","string","attributes","to","`","jax_export.Exported","`","or","of","`","tf.XlaCallModuleOp","`",".","You","can","disable","more","deserialization","safety","checks","by","passing","`","TF_XLA_FLAGS=","--","tf_xla_call_module_disabled_checks=platform","`","."]
docstr: Allows the serialization of a call target not known to be stable.
  
  Has effect only on serialization.
  Args:
    target_name: the name of the custom call target to allow.
docstr_tok: ["Allows","the","serialization","of","a","call","target","not","known","to","be","stable",".","Has","effect","only","on","serialization",".","Args",":","target_name",":","the","name","of","the","custom","call","target","to","allow","."]
code: def custom_call(cls, target_name: str) -> DisabledSafetyCheck:
      """Allows the serialization of a call target not known to be stable.
  
      Has effect only on serialization.
      Args:
        target_name: the name of the custom call target to allow.
      """
      return DisabledSafetyCheck(f"custom_call:{target_name}")
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import copy
  import dataclasses
  import functools
  import itertools
  import re
  from typing import Any, Callable, Union
  import warnings
  from absl import logging
  import numpy
  import jax
  from jax import sharding
  from jax._src import ad_util
  from jax._src import config
  from jax._src import core
  from jax._src import dispatch
  from jax._src import dtypes
  from jax._src import effects
  from jax._src.interpreters import mlir
  from jax._src.interpreters import pxla
  from jax._src.lib import xla_client
  from jax._src.lib.mlir import ir
  from jax._src.lib.mlir.dialects import hlo
  from jax._src.lib.mlir.dialects import func
  from jax._src import pjit
  from jax._src import sharding_impls
  from jax._src import source_info_util
  from jax._src import tree_util
  from jax._src import util
  from jax._src import xla_bridge
  from jax.experimental.export import _shape_poly
label_desc: Use image node_id 2 for calling the DisabledSafetyCheck obj's underlying member method code with example usage: obj.custom_call(cls, target_name) and returns: DisabledSafetyCheck