code: def _call_init_fn(init_fn, seed):
      """Calls `init_fn` with `seed` as a named or positional arg."""
      if not callable(init_fn):
          return tf.convert_to_tensor(
              init_fn
          )  # Non-callable initial value.
      try:
          return init_fn(seed=seed)  # Try passing seed as named arg.
      except TypeError:
          pass
      try:
          return init_fn()  # Maybe init_fn is deterministic?
      except (TypeError, ValueError):
          pass
      return init_fn(seed)
  
file_path: probability/tensorflow_probability/python/internal/trainable_state_util.py
imports: import collections
  import functools
  import inspect
  import re
  import types
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import structural_tuple
label_desc: Use image node_id 1 for calling a global function with example usage: _call_init_fn(init_fn, seed) and returns: init_fn, tf, init_fn, init_fn