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/contrib/data | |
| parent | c898e63d07fc63315be98f0772736e5d7f2fb44c (diff) | |
Updating the V2 variables API.
PiperOrigin-RevId: 214824023
Diffstat (limited to 'tensorflow/contrib/data')
| -rw-r--r-- | tensorflow/contrib/data/python/kernel_tests/iterator_ops_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/data/python/kernel_tests/iterator_ops_test.py b/tensorflow/contrib/data/python/kernel_tests/iterator_ops_test.py index 704c0d1eb2..7e2326bd17 100644 --- a/tensorflow/contrib/data/python/kernel_tests/iterator_ops_test.py +++ b/tensorflow/contrib/data/python/kernel_tests/iterator_ops_test.py @@ -42,7 +42,7 @@ class CheckpointInputPipelineHookTest(test.TestCase): del config global_step = training_util.get_or_create_global_step() update_global_step_op = global_step.assign_add(1) - latest_feature = variables.Variable( + latest_feature = variables.VariableV1( 0, name='latest_feature', dtype=dtypes.int64) store_latest_feature_op = latest_feature.assign(features) ops.add_to_collection('my_vars', global_step) |