code: def test_nest_rms_norm_on_nest(self, dtype):
      del dtype  # not used in this test case.
      a = np.array([1.4, 2.7, 7.3])
      b = 0.3 * np.eye(3, dtype=np.float32) + 0.64 * np.ones((3, 3))
      input_nest = (tf.convert_to_tensor(a), tf.convert_to_tensor(b))
      actual_norm_nest = rk_util.nest_rms_norm(input_nest)
      full_state = np.concatenate([np.expand_dims(a, 0), b])
      expected_norm_nest = np.linalg.norm(full_state) / np.sqrt(
          full_state.size
      )
      self.assertAllClose(expected_norm_nest, actual_norm_nest)
  
file_path: probability/tensorflow_probability/python/math/ode/runge_kutta_util_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.ode import runge_kutta_util
label_desc: Use image node_id 8 for calling the RungeKuttaUtilTest obj's underlying member method code with example usage: obj.test_nest_rms_norm_on_nest(dtype) without return types