diff options
Diffstat (limited to 'tensorflow/python/keras/engine/input_layer.py')
-rw-r--r-- | tensorflow/python/keras/engine/input_layer.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/keras/engine/input_layer.py b/tensorflow/python/keras/engine/input_layer.py index 8a4018a0df..6a69d0ed90 100644 --- a/tensorflow/python/keras/engine/input_layer.py +++ b/tensorflow/python/keras/engine/input_layer.py @@ -82,6 +82,7 @@ class InputLayer(base_layer.Layer): self.built = True self.sparse = sparse self.batch_size = batch_size + self.supports_masking = True if isinstance(input_shape, tensor_shape.TensorShape): input_shape = tuple(input_shape.as_list()) |