code: def _get_state_size(alg):
      if alg == random_ops_util.Algorithm.PHILOX.value:
          return PHILOX_STATE_SIZE
      elif alg == random_ops_util.Algorithm.THREEFRY.value:
          return THREEFRY_STATE_SIZE
      elif alg == random_ops_util.Algorithm.AUTO_SELECT.value:
          # For AUTO_SELECT, we'll manage the state as if it's for Philox.
          return PHILOX_STATE_SIZE
      else:
          raise ValueError(
              stateless_random_ops.unsupported_alg_error_msg(alg)
          )
  
imports: from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import sharded_variable
  from tensorflow.python.distribute import values_util
  from tensorflow.python.eager import context
  from tensorflow.python.framework import config
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import gen_stateful_random_ops
  from tensorflow.python.ops import gen_stateless_random_ops_v2
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.trackable import autotrackable
  from tensorflow.python.util import nest
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 5 for calling a global function with example usage: _get_state_size(alg) and returns: PHILOX_STATE_SIZE, THREEFRY_STATE_SIZE, PHILOX_STATE_SIZE