message: Small fix so that GDN can run on TPU
diff: @@ -1890,6 +1890,7 @@ class GDN(base.Layer):
  def beta_initializer(shape, dtype=None, partition_info=None):
  del partition_info # unused
  + pedestal = array_ops.constant(self._reparam_offset**2, dtype=self.dtype)
  return math_ops.sqrt(array_ops.ones(shape, dtype=dtype) + pedestal)
  def gamma_initializer(shape, dtype=None, partition_info=None):
  @@ -1897,6 +1898,7 @@ class GDN(base.Layer):
  assert len(shape) == 2
  assert shape[0] == shape[1]
  eye = linalg_ops.eye(shape[0], dtype=dtype)
  + pedestal = array_ops.constant(self._reparam_offset**2, dtype=self.dtype)
  return math_ops.sqrt(self._gamma_init * eye + pedestal)
  beta = self.add_variable(
  