diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-24 14:59:05 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-24 15:03:39 -0700 |
commit | e918dabba67c3eb53e6f852d299dffc12a2da6e9 (patch) | |
tree | 6b1150a5857b7b3376a3a61b400a8fb7554916a2 /tensorflow/contrib/layers | |
parent | b42bc6ee8448496f11e03e5c9a881d3655caf945 (diff) |
Small fix so that GDN can run on TPU
PiperOrigin-RevId: 197959536
Diffstat (limited to 'tensorflow/contrib/layers')
-rw-r--r-- | tensorflow/contrib/layers/python/layers/layers.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/layers/python/layers/layers.py b/tensorflow/contrib/layers/python/layers/layers.py index 25c3b1e7ea..f708da6693 100644 --- a/tensorflow/contrib/layers/python/layers/layers.py +++ b/tensorflow/contrib/layers/python/layers/layers.py @@ -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( |