diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-07-18 10:16:16 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-18 10:22:51 -0700 |
| commit | 9cc29a75ce8131db67b48e92dac3c16a255b92ed (patch) | |
| tree | 73bf7a7483d8f7ae3872437609b6943218938ff4 /tensorflow/contrib/checkpoint | |
| parent | 491b2d61156333c44e6bf06e2ac0a7ac02c4d310 (diff) | |
Allows constructing resource variables from tf.Variable.
Also adds arguments to control distributed aggregation to the tf.Variable constructor.
Removes tfe.Variable from examples as it's now unnecessary.
PiperOrigin-RevId: 205096552
Diffstat (limited to 'tensorflow/contrib/checkpoint')
| -rw-r--r-- | tensorflow/contrib/checkpoint/python/containers.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/checkpoint/python/containers.py b/tensorflow/contrib/checkpoint/python/containers.py index 4d3d531299..242c1e8ba4 100644 --- a/tensorflow/contrib/checkpoint/python/containers.py +++ b/tensorflow/contrib/checkpoint/python/containers.py @@ -35,9 +35,9 @@ class UniqueNameTracker(data_structures.CheckpointableDataStructure): self.slotdeps = tf.contrib.checkpoint.UniqueNameTracker() slotdeps = self.slotdeps slots = [] - slots.append(slotdeps.track(tfe.Variable(3.), "x")) # Named "x" - slots.append(slotdeps.track(tfe.Variable(4.), "y")) - slots.append(slotdeps.track(tfe.Variable(5.), "x")) # Named "x_1" + slots.append(slotdeps.track(tf.Variable(3.), "x")) # Named "x" + slots.append(slotdeps.track(tf.Variable(4.), "y")) + slots.append(slotdeps.track(tf.Variable(5.), "x")) # Named "x_1" ``` """ |