diff options
Diffstat (limited to 'tensorflow/python/keras/engine/base_layer.py')
-rw-r--r-- | tensorflow/python/keras/engine/base_layer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/keras/engine/base_layer.py b/tensorflow/python/keras/engine/base_layer.py index b6b05c0311..cb19a412a2 100644 --- a/tensorflow/python/keras/engine/base_layer.py +++ b/tensorflow/python/keras/engine/base_layer.py @@ -1001,7 +1001,7 @@ class Layer(checkpointable.CheckpointableBase): self.build(input_shape) with context.graph_mode(): - graph = eager_function.CapturingGraph() + graph = eager_function.FuncGraph('graph') with graph.as_default(): if isinstance(input_shape, list): inputs = [generate_placeholders_from_shape(shape) |