code: def InnerSRUCell():
      """The inner (non-parallel) computation of an SRU."""
  
      def f(
          cur_x_times_one_minus_f, cur_f, cur_state
      ):  # pylint: disable=invalid-name
          res = cur_f * cur_state + cur_x_times_one_minus_f
          return res, res
  
      return base.Fn("InnerSRUCell", f, n_out=2)
  
imports: from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import activation_fns
  from trax.layers import base
  from trax.layers import combinators
  from trax.layers import convolution
  from trax.layers import core
  from trax.layers import initializers
label_desc: Use image node_id 6 for calling a global function with example usage: InnerSRUCell() and returns: base, res,  res