code: def reduce_rms(x, **kwargs):
      return mtf.sqrt(mtf.reduce_mean(mtf.square(x), **kwargs))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  import re
  import gin
  from mesh_tensorflow import layers
  from mesh_tensorflow import ops_with_redefined_builtins
  import tensorflow.compat.v1
label_desc: Use image node_id 5 for calling a global function with example usage: reduce_rms(x) and returns: mtf