code: def test_nest_rms_norm_on_tensor(self, dtype):
      test_values = np.array([1.4 - 1j, 2.7 + 0.23j, 7.3 + 9.4j])
      test_values = test_values.astype(dtype_util.as_numpy_dtype(dtype))
      input_values = tf.cast(test_values, dtype=dtype)
      actual_norm = rk_util.nest_rms_norm(input_values)
      expected_norm = np.linalg.norm(test_values) / np.sqrt(
          test_values.size
      )
      self.assertAllClose(actual_norm, expected_norm)
  
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 7 for calling the RungeKuttaUtilTest obj's underlying member method code with example usage: obj.test_nest_rms_norm_on_tensor(dtype) without return types