494cf0b3bbf5b4288face03be974364f25a2370c,official/transformer/v2/transformer.py,EncoderStack,build,#EncoderStack#Any#,440

Before Change

      ])

    // Create final layer normalization layer.
    self.output_normalization = LayerNormalization(params["hidden_size"])
    super(EncoderStack, self).build(input_shape)

  def get_config(self):
    return {

After Change

      ])

    // Create final layer normalization layer.
    self.output_normalization = tf.keras.layers.LayerNormalization(
        epsilon=1e-6, dtype="float32")
    super(EncoderStack, self).build(input_shape)

  def get_config(self):
    return {

Instances