code: def _prepend_batch(batch_size, shape):
      """Prepends the batch dimension to the shape.
  
      If the batch_size value is known statically, this function returns a
      TensorShape, otherwise a Tensor.
      """
      if isinstance(batch_size, tf.Tensor):
          static_batch_size = tf.get_static_value(batch_size)
      else:
          static_batch_size = batch_size
      if static_batch_size is None:
          return tf.concat(([batch_size], shape), axis=0)
      return [static_batch_size] + shape
  
imports: import abc
  import tensorflow
  from tensorflow_addons.utils.types import TensorLike
  from typeguard import typechecked
  from typing import Any, Optional, Tuple, Union
  from tensorflow.python.ops import control_flow_util
label_desc: Use image node_id 2 for calling a global function with example usage: _prepend_batch(batch_size, shape) and returns: unknown, tf