Set wrapped layer built in keras Wrappers.

PiperOrigin-RevId: 293407052 Change-Id: I58ec07f651f0d8777f840e366fde739b5049bcc0
2020-02-05 11:05:35 -08:00 · 2020-02-05 11:05:35 -08:00 · b3eb3455ed
commit b3eb3455ed
parent b7f8c17030
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/python/keras/layers/wrappers.py
+++ b/tensorflow/python/keras/layers/wrappers.py
@ -56,6 +56,7 @@ class Wrapper(Layer):
  def build(self, input_shape=None):
    if not self.layer.built:
      self.layer.build(input_shape)
+      self.layer.built = True
    self.built = True

  @property