class_docstr: A stack of layers for Universal Transformer.
  
  This implementation is largely adapted from t2t universal transformer
  implementation. Reference:
  third_party/py/tensor2tensor/models/research
class_docstr_tok: ["A","stack","of","layers","for","Universal","Transformer",".","This","implementation","is","largely","adapted","from","t2t","universal","transformer","implementation",".","Reference",":","third_party\/py\/tensor2tensor\/models\/research"]
docstr: Layer normalization.
  
  Deprecated - can we remove this?
  
  Args:
    context: a Context
    x: a Tensor
    name: an optional string
  
  Returns:
    a Tensor
docstr_tok: ["Layer","normalization",".","Deprecated","-","can","we","remove","this","?","Args",":","context",":","a","Context","x",":","a","Tensor","name",":","an","optional","string","Returns",":","a","Tensor"]
code: def _layer_norm(self, context, x, name=None):
      """Layer normalization.
  
      Deprecated - can we remove this?
  
      Args:
        context: a Context
        x: a Tensor
        name: an optional string
  
      Returns:
        a Tensor
      """
      return mtf.layers.layer_norm(
          x, context.model.model_dim, self._norm_epsilon
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import math
  import gin
  import mesh_tensorflow
  from mesh_tensorflow.transformer import transformer
  import tensorflow.compat.v1
label_desc: Use image node_id 15 for calling the UTLayerStack obj's underlying member method code with example usage: obj._layer_norm(context, x, name) and returns: mtf