experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Test both V1 and V2 variables.

Browse Source 
PiperOrigin-RevId: 253063488
This commit is contained in:
Peter Buchlovsky 2019-06-13 10:55:31 -07:00 committed by TensorFlower Gardener
parent c22bc79896
commit 824fab41c0
1 changed files with 11 additions and 10 deletions

21
tensorflow/python/distribute/strategy_test_lib.py
View File

@ -427,18 +427,19 @@ class DistributionTestBase(test.TestCase):
run_and_concatenate(strategy, i)
def _test_trainable_variable(self, strategy):
with strategy.scope():
v1 = variables.Variable(1.0)
self.assertEqual(True, v1.trainable)
for cls in [variables.VariableV1, variables.Variable]:
with strategy.scope():
v1 = cls(1.0)
self.assertEqual(True, v1.trainable)
v2 = variables.Variable(
1.0, synchronization=variables.VariableSynchronization.ON_READ)
self.assertEqual(False, v2.trainable)
v2 = cls(
1.0, synchronization=variables.VariableSynchronization.ON_READ)
self.assertEqual(False, v2.trainable)
v3 = variables.Variable(
1.0, synchronization=variables.VariableSynchronization.ON_READ,
trainable=True)
self.assertEqual(True, v3.trainable)
v3 = cls(
1.0, synchronization=variables.VariableSynchronization.ON_READ,
trainable=True)
self.assertEqual(True, v3.trainable)
class OneDeviceDistributionTestBase(test.TestCase):