experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Cleanup some unneeded code from frozen keras base layer.

Browse Source 
PiperOrigin-RevId: 300595321
Change-Id: Ic9ff4b6c402d09cee01c1dea5aee4a2678fe96e6
This commit is contained in:
Scott Zhu 2020-03-12 12:02:47 -07:00 committed by TensorFlower Gardener
parent dbf532e329
commit 6625cee185
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/python/frozen_keras/engine/legacy_base_layer.py
View File

@ -62,8 +62,8 @@ from tensorflow.python.frozen_keras.utils import generic_utils
from tensorflow.python.frozen_keras.utils import layer_utils
from tensorflow.python.frozen_keras.utils import tf_utils
# A module that only depends on `keras.layers` import these from here.
from tensorflow.python.keras.utils.generic_utils import to_snake_case # pylint: disable=unused-import
from tensorflow.python.keras.utils.tf_utils import is_tensor_or_tensor_list # pylint: disable=unused-import
from tensorflow.python.frozen_keras.utils.generic_utils import to_snake_case # pylint: disable=unused-import
from tensorflow.python.frozen_keras.utils.tf_utils import is_tensor_or_tensor_list # pylint: disable=unused-import
from tensorflow.python.module import module
from tensorflow.python.ops import array_ops
from tensorflow.python.ops import math_ops
@ -273,7 +273,6 @@ class LegacyBaseLayer(module.Module):
'batch_size',
'weights',
'activity_regularizer',
'autocast'
}
# Validate optional keyword arguments.
generic_utils.validate_kwargs(kwargs, allowed_kwargs)
@ -453,7 +452,7 @@ class LegacyBaseLayer(module.Module):
Accepted values are constants defined in the class
`tf.VariableAggregation`.
**kwargs: Additional keyword arguments. Accepted values are `getter`,
`collections`, `experimental_autocast` and `caching_device`.
`collections` and `caching_device`.
Returns:
The created variable. Usually either a `Variable` or `ResourceVariable`
@ -1011,7 +1010,7 @@ class LegacyBaseLayer(module.Module):
@trackable.no_automatic_dependency_tracking
def input_spec(self, value):
for v in nest.flatten(value):
if v is not None and not isinstance(v, InputSpec):
if v is not None and not isinstance(v, input_spec.InputSpec):
raise TypeError('Layer input_spec must be an instance of InputSpec. '
'Got: {}'.format(v))
self._input_spec = value
@ -2783,8 +2782,3 @@ class KerasHistory(
# Added to maintain memory and performance characteristics of `namedtuple`
# while subclassing.
__slots__ = ()
# Avoid breaking users who directly import this symbol from this file.
# TODO(fchollet): remove this.
InputSpec = input_spec.InputSpec # pylint:disable=invalid-name