diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-09-27 13:18:33 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 13:23:04 -0700 |
| commit | 4cedc8b6e738b7a188c9c091cf667bacafae44b7 (patch) | |
| tree | 56de35940e5f9daedd5f39a82d2cd90cf374e4e4 /tensorflow/python/keras | |
| parent | c898e63d07fc63315be98f0772736e5d7f2fb44c (diff) | |
Updating the V2 variables API.
PiperOrigin-RevId: 214824023
Diffstat (limited to 'tensorflow/python/keras')
| -rw-r--r-- | tensorflow/python/keras/engine/base_layer.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/keras/engine/base_layer.py b/tensorflow/python/keras/engine/base_layer.py index cb19a412a2..e98b131ae6 100644 --- a/tensorflow/python/keras/engine/base_layer.py +++ b/tensorflow/python/keras/engine/base_layer.py @@ -1972,7 +1972,9 @@ def make_variable(name, if use_resource is None: use_resource = True - v = tf_variables.Variable( + # TODO(apassos,rohanj) figure out how to remove collections from here so we + # can remove the V1. + v = tf_variables.VariableV1( initial_value=init_val, name=name, trainable=trainable, |