Enable Model subclassing, both in eager-mode and symbolic-mode.
authorFrancois Chollet <fchollet@google.com>
Tue, 13 Feb 2018 01:44:01 +0000 (17:44 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 13 Feb 2018 01:48:15 +0000 (17:48 -0800)
commit03cebfcc80169b867ff87f700bdfd27e28e1c7bc
tree252a941a52048943873f17bda5370251bd9d81b0
parent4b9ef6c8e07dea7d18f552fa4955c3176646f95d
Enable Model subclassing, both in eager-mode and symbolic-mode.

PiperOrigin-RevId: 185464334
15 files changed:
tensorflow/python/keras/BUILD
tensorflow/python/keras/_impl/keras/backend.py
tensorflow/python/keras/_impl/keras/engine/topology.py
tensorflow/python/keras/_impl/keras/engine/training.py
tensorflow/python/keras/_impl/keras/engine/training_eager.py
tensorflow/python/keras/_impl/keras/model_subclassing_test.py [new file with mode: 0644]
tensorflow/python/keras/_impl/keras/models.py
tensorflow/python/keras/_impl/keras/utils/layer_utils.py
tensorflow/python/layers/base.py
tensorflow/python/layers/network.py
tensorflow/python/layers/network_test.py
tensorflow/tools/api/golden/tensorflow.keras.-model.pbtxt
tensorflow/tools/api/golden/tensorflow.keras.-sequential.pbtxt
tensorflow/tools/api/golden/tensorflow.keras.models.-model.pbtxt
tensorflow/tools/api/golden/tensorflow.keras.models.-sequential.pbtxt